feat 初次提交

internal/net/ws_gateway/server.go

@@ -0,0 +1,74 @@
+package ws_gateway
+
+import (
+	"common/log"
+	"common/net/socket"
+	"common/utils"
+	"fmt"
+	"gateway/internal/handler/ws_handler"
+	"go.uber.org/zap"
+	"strconv"
+	"time"
+)
+
+type GatewayWsServer struct {
+	logger *zap.SugaredLogger
+}
+
+func (g *GatewayWsServer) OnOpen(conn socket.ISocketConn) ([]byte, socket.Action) {
+	g.logger = log.GetLogger().Named(fmt.Sprintf("addr:%v", conn.RemoteAddr()))
+	return nil, socket.None
+}
+
+func (g *GatewayWsServer) OnHandShake(conn socket.ISocketConn, bytes []byte, callback func(conn socket.ISocketConn, bytes []byte)) socket.Action {
+	token, ok := conn.GetParam("token").(string)
+	if !ok {
+		g.logger.Warnf("token is not string")
+		return socket.Close
+	}
+	//claims, err := utils.ParseToken(token, config.Get().Auth.Secret)
+	//if err != nil {
+	//	g.logger.Warnf("token is invalid")
+	//	return socket.Close
+	//}
+
+	t, _ := strconv.Atoi(token)
+	claims := utils.Claims{
+		USN: int64(t),
+	}
+	go func(shResp []byte) {
+		if oldClient := ws_handler.UserMgr.GetByUSN(claims.USN); oldClient != nil {
+			oldClient.CloseClient()
+		}
+		client := ws_handler.NewClient(claims.USN, conn)
+		ws_handler.UserMgr.Add(claims.USN, client)
+		conn.SetParam("client", client)
+		callback(conn, shResp)
+	}(bytes)
+	return socket.None
+}
+
+func (g *GatewayWsServer) OnMessage(conn socket.ISocketConn, bytes []byte) socket.Action {
+	client, ok := conn.GetParam("client").(*ws_handler.Client)
+	if !ok || client.USN == 0 {
+		return socket.Close
+	}
+	client.OnEvent(&ws_handler.ClientEvent{Msg: bytes})
+	return socket.None
+}
+
+func (g *GatewayWsServer) OnPong(conn socket.ISocketConn) {
+	client, ok := conn.GetParam("client").(*ws_handler.Client)
+	if !ok || client.USN == 0 {
+		return
+	}
+	client.OnEvent(&ws_handler.PongEvent{})
+}
+
+func (g *GatewayWsServer) OnClose(_ socket.ISocketConn, _ error) socket.Action {
+	return socket.Close
+}
+
+func (g *GatewayWsServer) OnTick() (time.Duration, socket.Action) {
+	return 5 * time.Second, socket.None
+}