From b73a796dc5d9404c4cada41ae434d1ed87036599 Mon Sep 17 00:00:00 2001
From: yimin <yimin@yahaha.com>
Date: Thu, 9 May 2024 20:48:56 +0800
Subject: [PATCH] =?UTF-8?q?gate=20tcp=20server=E6=9B=B4=E6=96=B0=E4=BB=A3?=
 =?UTF-8?q?=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 network/tcp/pack.go        | 8 ++++----
 network/tcp/server_conn.go | 3 +++
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/network/tcp/pack.go b/network/tcp/pack.go
index 123330c0..c2fc0b90 100644
--- a/network/tcp/pack.go
+++ b/network/tcp/pack.go
@@ -17,7 +17,7 @@ import (
 )
 
 const (
-	msgLenBytes  uint16 = 2        // 消息长度字节数
+	msgLenBytes  uint16 = 4        // 消息长度字节数
 	msgByteOrder string = "little" // 消息字节排序
 )
 
@@ -30,7 +30,7 @@ func pack(msg []byte) (packet []byte, err error) {
 	var buf bytes.Buffer
 	buf.Grow(len(msg) + int(msgLenBytes))
 
-	if err = binary.Write(&buf, byteOrder(), uint16(len(msg))); err != nil {
+	if err = binary.Write(&buf, byteOrder(), uint32(len(msg))); err != nil {
 		return
 	}
 
@@ -47,7 +47,7 @@ func pack(msg []byte) (packet []byte, err error) {
 func unpack(packet []byte) (msg []byte, err error) {
 	var (
 		buf    = bytes.NewBuffer(packet)
-		msgLen uint16
+		msgLen uint32
 	)
 
 	if err = binary.Read(buf, byteOrder(), &msgLen); err != nil {
@@ -78,7 +78,7 @@ func readMsgFromConn(reader *bufio.Reader, maxMsgLength int) (msg []byte, err er
 
 	var (
 		buf    = bytes.NewBuffer(packet)
-		msgLen uint16
+		msgLen uint32
 	)
 
 	if err = binary.Read(buf, byteOrder(), &msgLen); err != nil {
diff --git a/network/tcp/server_conn.go b/network/tcp/server_conn.go
index c6068582..c68a62ec 100644
--- a/network/tcp/server_conn.go
+++ b/network/tcp/server_conn.go
@@ -414,5 +414,8 @@ func (c *serverConn) doWrite(buf []byte) (err error) {
 		return
 	}
 	_, err = c.conn.Write(buf)
+	if err != nil {
+		log.Errorf("cid:%+v,uid:%+v doWrite error:%+v", c.ID(), c.UID(), err)
+	}
 	return
 }