feat 在线人数埋点

app/prometheus.go

@@ -6,6 +6,8 @@ import (
 	"fmt"
 	"git.hlsq.asia/mmorpg/service-common/log"
 	"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"
@@ -19,6 +21,13 @@ type ModulePrometheus struct {
 
 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
 }
 

internal/global/global.go

@@ -1,5 +1,7 @@
 package global
 
+import "github.com/prometheus/client_golang/prometheus"
+
 const (
 	KeyGatewayAccessToken  = "gateway:token:access:%v"
 	KeyGatewayRefreshToken = "gateway:token:refresh:%v"
@@ -17,3 +19,7 @@ const (
 var PublicPaths = []string{
 	"/user/info",
 }
+
+var (
+	OnlineUsersGauge prometheus.Gauge
+)

internal/handler/ws_handler/client/manager.go

@@ -1,6 +1,7 @@
 package client
 
 import (
+	"git.hlsq.asia/mmorpg/service-gateway/internal/global"
 	"sync"
 )
 
@@ -21,12 +22,14 @@ func (m *userManager) Add(usn int64, client *Client) {
 	m.Lock()
 	defer m.Unlock()
 	m.userMap[usn] = client
+	global.OnlineUsersGauge.Inc()
 }
 
 func (m *userManager) Delete(usn int64) {
 	m.Lock()
 	defer m.Unlock()
 	delete(m.userMap, usn)
+	global.OnlineUsersGauge.Dec()
 }
 
 func (m *userManager) GetAll() map[int64]*Client {