feat Prometheus

app/app.go

@@ -21,11 +21,11 @@ func (p *Program) Init(_ svc.Environment) error {
 	}
 	p.moduleList = append(p.moduleList, base)
 	p.moduleList = append(p.moduleList, (&module.DB{}).Bind(config.Get().DB))
-	p.moduleList = append(p.moduleList, &ModulePrometheus{})
 	p.moduleList = append(p.moduleList, &ModuleWebServer{})
 	p.moduleList = append(p.moduleList, &ModuleWebsocketServer{})
 	p.moduleList = append(p.moduleList, &ModuleGrpcServer{})
 	p.moduleList = append(p.moduleList, &ModuleLoginQueue{})
+	p.moduleList = append(p.moduleList, (&module.Prometheus{}).Bind(config.Get().Metric))
 	p.moduleList = append(p.moduleList, (&module.Tracer{}).Bind(config.Get().Metric, common.KeyDiscoverServiceNameGateway))
 
 	for i, m := range p.moduleList {

app/prometheus.go

@@ -1,61 +0,0 @@
-package app
-
-import (
-	"context"
-	"errors"
-	"fmt"
-	"git.hlsq.asia/mmorpg/service-common/log"
-	"git.hlsq.asia/mmorpg/service-common/module"
-	"git.hlsq.asia/mmorpg/service-gateway/config"
-	"git.hlsq.asia/mmorpg/service-gateway/internal/global"
-	"github.com/prometheus/client_golang/prometheus"
-	"github.com/prometheus/client_golang/prometheus/promhttp"
-	"net/http"
-	"sync"
-)
-
-// ModulePrometheus 普罗米修斯模块
-type ModulePrometheus struct {
-	wg     *sync.WaitGroup
-	server *http.Server
-}
-
-func (m *ModulePrometheus) Init() error {
-	m.wg = &sync.WaitGroup{}
-
-	global.OnlineUsersGauge = prometheus.NewGauge(prometheus.GaugeOpts{
-		Name: "gateway_online_users_total",
-		Help: "Total number of online users in gateway",
-	})
-	prometheus.MustRegister(global.OnlineUsersGauge)
-
-	return nil
-}
-
-func (m *ModulePrometheus) Start() error {
-	m.wg.Add(1)
-	go func() {
-		defer m.wg.Done()
-		m.server = &http.Server{
-			Addr:    fmt.Sprintf("%v:%v", config.Get().Metric.Prometheus.Address, config.Get().Metric.Prometheus.Port),
-			Handler: promhttp.Handler(),
-		}
-		if err := m.server.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
-			log.Errorf("prometheus server failed: %v", err.Error())
-		}
-		log.Infof("prometheus server stop.")
-	}()
-	return nil
-}
-
-func (m *ModulePrometheus) Stop() error {
-	if err := m.server.Shutdown(context.Background()); err != nil {
-		log.Errorf("stop prometheus server failed: %v", err)
-	}
-	m.wg.Wait()
-	return nil
-}
-
-func (m *ModulePrometheus) Bind(_ ...any) module.Module {
-	return m
-}