fix(socket): 修复连接处理逻辑并优化数据解码流程

- 修复 `OnOpen` 中网络类型判断位置不正确的问题，提前过滤非 TCP 连接
- 移除 `OnTraffic` 中重复的网络类型判断
- 优化 `TomeeSocketCodec` 的解码逻辑，使用 `InboundBuffered` 和 `Next` 提高效率
- 调整 `ByteArray` 创建方法参数，避免可变参数带来的性能损耗
- 在 `ClientData` 中将 `IsCrossDomain` 改为 `sync.Once` 避免重复处理
- 使用 `AsyncWrite` 替代 `Write` 提升写入异步性
- 修复连接关闭流程，使用

common/rpc/rpc.go

@@ -56,7 +56,7 @@ func (h *ServerHandler) RegisterLogic(ctx context.Context, id, port uint16) erro
 	}
 	t, _ := blservice.NewLoginServiceService().GetServerID(id)
 	aa, ok := clientmap[t]
-	if ok { //如果已经存在且这个端口已经被存过
+	if ok && aa != nil { //如果已经存在且这个端口已经被存过
 		aa.QuitSelf(0)
 	}
 	clientmap[port] = &revClient

common/socket/ServerEvent.go

@@ -27,9 +27,6 @@ func (s *Server) Boot() error {
 		panic(err)
 	}
 
-	// err := gnet.Run(s, s.network+"://"+s.addr, gnet.WithMulticore(s.multicore))
-	cool.Loger.Debug(context.Background(), "server exits with error: %v", err)
-	// logging.Infof("server exits with error: %v", err)
 	return nil
 }
 
@@ -53,21 +50,14 @@ func (s *Server) OnClose(c gnet.Conn, _ error) (action gnet.Action) {
 	atomic.AddInt64(&s.connected, -1)
 	//logging.Infof("conn[%v] disconnected", c.RemoteAddr().String())
 	v, _ := c.Context().(*player.ClientData)
-	s.workerPool.Submit(func() { //TODO 这里可能存在顺序执行问题,待修复
+	if v.Player != nil {
+		v.SaveL.Do(func() { //使用保存锁,确保在踢人和掉线的时候只保存一次
+			//cool.Loger.Info(context.TODO(), "准备保存", v.Player.Info.UserID)
+			v.Player.Save() //保存玩家数据
 
-		if v.Player != nil {
-			v.SaveL.Do(func() { //使用保存锁,确保在踢人和掉线的时候只保存一次
-				//cool.Loger.Info(context.TODO(), "准备保存", v.Player.Info.UserID)
-				v.Player.Save() //保存玩家数据
-				//cool.Loger.Info(context.TODO(), "保存完成", v.Player.Info.UserID)
-				if v.CloseChan != nil {
-					close(v.CloseChan)
-				}
-			})
+		})
 
-		}
-
-	})
+	}
 
 	//}
 	//关闭连接
@@ -90,6 +80,10 @@ func (s *Server) OnBoot(eng gnet.Engine) gnet.Action {
 }
 
 func (s *Server) OnOpen(conn gnet.Conn) (out []byte, action gnet.Action) {
+	if s.network != "tcp" {
+		return nil, gnet.Close
+	}
+
 	if conn.Context() == nil {
 		conn.SetContext(player.NewClientData(conn)) //注入data
 	}
@@ -108,9 +102,6 @@ func (s *Server) OnTraffic(c gnet.Conn) (action gnet.Action) {
 
 		}
 	}()
-	if s.network != "tcp" {
-		return gnet.Close
-	}
 
 	ws := c.Context().(*player.ClientData).Wsmsg
 	if ws.Tcp { //升级失败时候防止缓冲区溢出
@@ -136,9 +127,7 @@ func (s *Server) OnTraffic(c gnet.Conn) (action gnet.Action) {
 	}
 	//	fmt.Println(ws.Buf.Bytes())
 	c.Discard(len1)
-	if ws.Buf.Len() <= 0 {
-		return gnet.None
-	}
+
 	messages, err := ws.Decode(c)
 	if err != nil {
 		return gnet.Close
@@ -151,9 +140,8 @@ func (s *Server) OnTraffic(c gnet.Conn) (action gnet.Action) {
 		t := c.Context().(*player.ClientData)
 		//client := conn.RemoteAddr().String()
 		s.workerPool.Submit(func() { //TODO 这里可能存在顺序执行问题,待修复
-			t.Mu.RLock()
+
 			t.OnEvent(msg.Payload)
-			t.Mu.RUnlock()
 
 		})
 
@@ -163,33 +151,28 @@ func (s *Server) OnTraffic(c gnet.Conn) (action gnet.Action) {
 }
 
 func (s *Server) handleTcp(conn gnet.Conn) (action gnet.Action) {
-	if s.discorse && !conn.Context().(*player.ClientData).IsCrossDomain {
+
+	conn.Context().(*player.ClientData).IsCrossDomain.Do(func() { //跨域检测
 		handle(conn)
-	}
-	conn.Context().(*player.ClientData).IsCrossDomain = true
-	for i := 0; i < s.batchRead; i++ {
-
-		data, err := s.codec.Decode(conn)
+	})
 
+	data, err := s.codec.Decode(conn)
+	if err != nil {
 		if err == codec.ErrIncompletePacket {
-			break
+			if err := conn.Wake(nil); err != nil { // wake up the connection manually to avoid missing the leftover data
+				cool.Loger.Errorf(context.Background(), "failed to wake up the connection, %v", err)
+				return gnet.Close
+			}
 		}
-		if err != nil {
-
-			return gnet.Close
-
-		}
-
-		//cool.Loger.Debug(context.Background(), "原始数据", hex.EncodeToString(data))
-		t := conn.Context().(*player.ClientData)
-
-		err = s.workerPool.Submit(func() { //TODO 这里可能存在顺序执行问题,待修复
-			t.OnEvent(data)
-
-		})
+		return gnet.Close
 
 	}
 
+	s.workerPool.Submit(func() { //TODO 这里可能存在顺序执行问题,待修复
+		conn.Context().(*player.ClientData).OnEvent(data)
+
+	})
+
 	if conn.InboundBuffered() > 0 {
 		if err := conn.Wake(nil); err != nil { // wake up the connection manually to avoid missing the leftover data
 			cool.Loger.Errorf(context.Background(), "failed to wake up the connection, %v", err)

common/socket/codec/SocketCodec_Tomee.go

@@ -63,21 +63,11 @@ func (codec TomeeSocketCodec) Decode(c gnet.Conn) ([]byte, error) {
 		return nil, errors.New("packet body exceeds max length")
 	}
 
-	// 检查整个包是否完整
-	buf, err := c.Peek(int(bodyLen))
-	if err != nil {
-		if errors.Is(err, io.ErrShortBuffer) {
-			return nil, ErrIncompletePacket
-		}
-		return nil, err
+	if c.InboundBuffered() < int(bodyLen) {
+		return nil, ErrIncompletePacket
 	}
-
 	// 提取包体
-	body := make([]byte, bodyLen)
-	copy(body, buf)
-
-	// 从缓冲区中丢弃已读取的数据
-	_, _ = c.Discard(int(bodyLen))
+	body, _ := c.Next(int(bodyLen))
 
 	return body, nil
 }

common/utils/bytearray/bytearray.go

@@ -27,7 +27,7 @@ var bufferpool = &sync.Pool{
 }
 
 // CreateByteArray 创建一个新的ByteArray实例，使用指定的字节数组
-func CreateByteArray(bytes ...[]byte) *ByteArray {
+func CreateByteArray(bytes []byte) *ByteArray {
 	var ba *ByteArray
 	if len(bytes) == 0 { //如果是0，则为新创建
 		ba = bufferpool.Get().(*ByteArray)
@@ -35,10 +35,7 @@ func CreateByteArray(bytes ...[]byte) *ByteArray {
 		ba = &ByteArray{endian: defaultEndian}
 	}
 
-	for _, num := range bytes {
-		ba.buf = append(ba.buf, num...)
-	}
-
+	ba.buf = append(ba.buf, bytes...)
 	ba.ResetPos()
 	return ba
 }