75 lines
1.9 KiB
Go
75 lines
1.9 KiB
Go
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
|
|
}
|