feat 接入Prometheus

Server/gateway/handler/ws_handler/client.go

@@ -3,11 +3,9 @@ package ws_handler
 import (
 	"common/log"
 	"common/net/socket"
-	"common/proto/sc/sc_pb"
 	"context"
 	"fmt"
 	"go.uber.org/zap"
-	"google.golang.org/protobuf/proto"
 	"runtime/debug"
 	"sync"
 	"time"
@@ -84,47 +82,6 @@ func (c *Client) OnEvent(event Event) {
 	}
 }
 
-// WriteMessage 向客户端发送消息
-func (c *Client) WriteMessage(id sc_pb.MessageID, data proto.Message) {
-	if c.conn.IsClose() {
-		return
-	}
-	d, err := proto.Marshal(data)
-	if err != nil {
-		c.logger.Errorf("WriteMessage proto.Marshal err: %v", err)
-		return
-	}
-	m, err := proto.Marshal(&sc_pb.Message{
-		ID:      id,
-		Payload: d,
-	})
-	if err != nil {
-		c.logger.Errorf("WriteMessage proto.Marshal err: %v", err)
-		return
-	}
-	if err = c.conn.Write(m); err != nil {
-		c.logger.Errorf("WriteMessage err: %v", err)
-	}
-}
-
-// WriteBytes 向客户端发送字节数据
-func (c *Client) WriteBytes(id sc_pb.MessageID, data []byte) {
-	if c.conn.IsClose() {
-		return
-	}
-	m, err := proto.Marshal(&sc_pb.Message{
-		ID:      id,
-		Payload: data,
-	})
-	if err != nil {
-		c.logger.Errorf("WriteBytes proto.Marshal err: %v", err)
-		return
-	}
-	if err = c.conn.Write(m); err != nil {
-		c.logger.Errorf("WriteBytes err: %v", err)
-	}
-}
-
 // CloseClient 关闭客户端（同步，会等待onClose执行完成）
 func (c *Client) CloseClient() {
 	if c.cancel != nil {

Server/gateway/handler/ws_handler/client_write.go

@@ -0,0 +1,57 @@
+package ws_handler
+
+import (
+	"common/proto/sc/sc_pb"
+	"google.golang.org/protobuf/proto"
+)
+
+// WriteMessage 向客户端发送消息
+func (c *Client) WriteMessage(id sc_pb.MessageID, data proto.Message) {
+	if c.conn == nil || c.conn.IsClose() {
+		return
+	}
+	d, err := proto.Marshal(data)
+	if err != nil {
+		c.logger.Errorf("WriteMessage proto.Marshal err: %v", err)
+		return
+	}
+	m, err := proto.Marshal(&sc_pb.Message{
+		ID:      id,
+		Payload: d,
+	})
+	if err != nil {
+		c.logger.Errorf("WriteMessage proto.Marshal err: %v", err)
+		return
+	}
+	if err = c.conn.Write(m); err != nil {
+		c.logger.Errorf("WriteMessage err: %v", err)
+	}
+}
+
+// WriteBytes 向客户端发送字节数据
+func (c *Client) WriteBytes(id sc_pb.MessageID, data []byte) {
+	if c.conn == nil || c.conn.IsClose() {
+		return
+	}
+	m, err := proto.Marshal(&sc_pb.Message{
+		ID:      id,
+		Payload: data,
+	})
+	if err != nil {
+		c.logger.Errorf("WriteBytes proto.Marshal err: %v", err)
+		return
+	}
+	if err = c.conn.Write(m); err != nil {
+		c.logger.Errorf("WriteBytes err: %v", err)
+	}
+}
+
+// WriteBytesPreMarshal 向客户端发送字节数据（需要预先打包，适合广播相同数据）
+func (c *Client) WriteBytesPreMarshal(data []byte) {
+	if c.conn == nil || c.conn.IsClose() {
+		return
+	}
+	if err := c.conn.Write(data); err != nil {
+		c.logger.Errorf("WriteBytes err: %v", err)
+	}
+}

Server/gateway/handler/ws_handler/manager.go

@@ -32,7 +32,23 @@ func (m *userManager) Delete(uid int32) {
 func (m *userManager) GetAll() map[int32]*Client {
 	m.RLock()
 	defer m.RUnlock()
-	return m.userMap
+
+	copyMap := make(map[int32]*Client, len(m.userMap))
+	for k, v := range m.userMap {
+		copyMap[k] = v
+	}
+	return copyMap
+}
+
+func (m *userManager) GetAllInterface() []interface{} {
+	m.RLock()
+	defer m.RUnlock()
+
+	r := make([]interface{}, 0)
+	for _, v := range m.userMap {
+		r = append(r, v)
+	}
+	return r
 }
 
 func (m *userManager) GetByUID(uid int32) *Client {