feat 接入Prometheus

Server/gateway/app/app.go

@@ -21,6 +21,7 @@ type Module interface {
 func (p *Program) Init(_ svc.Environment) error {
 	p.moduleList = append(p.moduleList, &ModuleBase{})
 	p.moduleList = append(p.moduleList, &ModuleDB{})
+	p.moduleList = append(p.moduleList, &ModulePrometheus{})
 	p.moduleList = append(p.moduleList, &ModuleWebServer{})
 	p.moduleList = append(p.moduleList, &ModuleWebsocketServer{})
 	p.moduleList = append(p.moduleList, &ModuleGrpcServer{})

Server/gateway/app/prometheus.go

@@ -0,0 +1,47 @@
+package app
+
+import (
+	"common/log"
+	"context"
+	"errors"
+	"fmt"
+	"gateway/config"
+	"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{}
+	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().Monitor.Prometheus.Address, config.Get().Monitor.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
+}

Server/gateway/app/web.go

@@ -2,6 +2,7 @@ package app
 
 import (
 	"common/log"
+	"context"
 	"errors"
 	"fmt"
 	"gateway/config"
@@ -41,7 +42,7 @@ func (m *ModuleWebServer) Start() error {
 }
 
 func (m *ModuleWebServer) Stop() error {
-	if err := m.server.Close(); err != nil {
+	if err := m.server.Shutdown(context.Background()); err != nil {
 		log.Errorf("stop http server failed: %v", err)
 	}
 	m.wg.Wait()