1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54
|
package server
import (
"crypto/tls"
"fmt"
"net"
"github.com/prometheus/client_golang/prometheus"
"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/module/modserver"
"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/module/modshared"
"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/module/observability"
"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/tool/tlstool"
)
type Factory struct {
Gatherer prometheus.Gatherer
}
func (f *Factory) New(config *modserver.Config) (modserver.Module, error) {
listenCfg := config.Config.Observability.Listen
var listener func() (net.Listener, error)
tlsConfig, err := tlstool.MaybeDefaultServerTLSConfig(listenCfg.GetCertificateFile(), listenCfg.GetKeyFile())
if err != nil {
return nil, err
}
if tlsConfig != nil {
listener = func() (net.Listener, error) {
return tls.Listen(*listenCfg.Network, listenCfg.Address, tlsConfig)
}
} else {
listener = func() (net.Listener, error) {
return net.Listen(*listenCfg.Network, listenCfg.Address)
}
}
return &module{
log: config.Log,
api: config.Api,
cfg: config.Config.Observability,
listener: listener,
gatherer: f.Gatherer,
registerer: config.Registerer,
serverName: fmt.Sprintf("%s/%s/%s", config.KasName, config.Version, config.CommitId),
probeRegistry: config.ProbeRegistry,
}, nil
}
func (f *Factory) Name() string {
return observability.ModuleName
}
func (f *Factory) StartStopPhase() modshared.ModuleStartStopPhase {
return modshared.ModuleStartBeforeServers
}
|