Use TLS with the monitoring / profiling routes

giongto35 · Jul 5, 2023 · 8fef57a · 8fef57a
1 parent cc41488
commit 8fef57a
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/pkg/coordinator/coordinator.go b/pkg/coordinator/coordinator.go
@@ -28,7 +28,7 @@ func New(conf config.CoordinatorConfig, log *logger.Logger) (services service.Gr
 	}
 	services.Add(hub, h)
 	if conf.Coordinator.Monitoring.IsEnabled() {
-		services.Add(monitoring.New(conf.Coordinator.Monitoring, h.GetHost(), log))
+		services.Add(monitoring.New(conf.Coordinator.Monitoring, conf.Coordinator.Server, h.GetHost(), log))
 	}
 	return
 }

diff --git a/pkg/monitoring/monitoring.go b/pkg/monitoring/monitoring.go
@@ -23,7 +23,7 @@ type Monitoring struct {
 
 // New creates new monitoring service.
 // The tag param specifies owner label for logs.
-func New(conf config.Monitoring, baseAddr string, log *logger.Logger) *Monitoring {
+func New(conf config.Monitoring, servConf config.Server, baseAddr string, log *logger.Logger) *Monitoring {
 	serv, err := httpx.NewServer(
 		net.JoinHostPort(baseAddr, strconv.Itoa(conf.Port)),
 		func(s *httpx.Server) httpx.Handler {
@@ -52,6 +52,7 @@ func New(conf config.Monitoring, baseAddr string, log *logger.Logger) *Monitorin
 			return h
 		},
 		httpx.WithPortRoll(true),
+		httpx.WithServerConfig(servConf),
 		httpx.WithLogger(log),
 	)
 	if err != nil {

diff --git a/pkg/worker/worker.go b/pkg/worker/worker.go
@@ -48,7 +48,7 @@ func New(conf config.WorkerConfig, log *logger.Logger, done chan struct{}) (serv
 	}
 	services.Add(h)
 	if conf.Worker.Monitoring.IsEnabled() {
-		services.Add(monitoring.New(conf.Worker.Monitoring, h.GetHost(), log))
+		services.Add(monitoring.New(conf.Worker.Monitoring, conf.Worker.Server, h.GetHost(), log))
 	}
 	st, err := GetCloudStorage(conf.Storage.Provider, conf.Storage.Key)
 	if err != nil {