From 9e11b045e251488ff8f4bce9cb49ab73cfb28ac7 Mon Sep 17 00:00:00 2001
From: lifeiyi <736926938@qq.com>
Date: Mon, 14 Dec 2020 17:56:31 +0800
Subject: [PATCH] =?UTF-8?q?tcp=E6=B6=88=E6=81=AF=E5=86=85=E5=AD=98?=
 =?UTF-8?q?=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 network/slicepool.go | 11 ++++++++++-
 network/tcp_msg.go   |  8 ++++----
 2 files changed, 14 insertions(+), 5 deletions(-)

diff --git a/network/slicepool.go b/network/slicepool.go
index b7aa2f3..1e11cf2 100644
--- a/network/slicepool.go
+++ b/network/slicepool.go
@@ -5,7 +5,8 @@ import (
 )
 
 type INetMempool interface {
-	MakeByteSlice(size int) []byte
+	MakeReadByteSlice(size int) []byte
+	MakeWriteByteSlice(size int) []byte
 	ReleaseByteSlice(byteBuff []byte) bool
 }
 
@@ -80,6 +81,14 @@ func (areaPool *memAreaPool) MakeByteSlice(size int) []byte{
 	return nil
 }
 
+func (areaPool *memAreaPool) MakeReadByteSlice(size int) []byte{
+	return areaPool.MakeByteSlice(size)
+}
+
+func (areaPool *memAreaPool) MakeWriteByteSlice(size int) []byte{
+	return areaPool.MakeByteSlice(size)
+}
+
 func (areaPool *memAreaPool) ReleaseByteSlice(byteBuff []byte) bool {
 	for i:=0;i<len(memAreaPoolList);i++{
 		if cap(byteBuff) <= memAreaPoolList[i].maxAreaValue {
diff --git a/network/tcp_msg.go b/network/tcp_msg.go
index 71f53fd..a22bcdf 100644
--- a/network/tcp_msg.go
+++ b/network/tcp_msg.go
@@ -101,13 +101,13 @@ func (p *MsgParser) Read(conn *TCPConn) ([]byte, error) {
 	
 	// data
 	//msgData := make([]byte, msgLen)
-	msgData := p.MakeByteSlice(int(msgLen))
-	if _, err := io.ReadFull(conn, msgData); err != nil {
+	msgData := p.MakeReadByteSlice(int(msgLen))
+	if _, err := io.ReadFull(conn, msgData[:msgLen]); err != nil {
 		p.ReleaseByteSlice(msgData)
 		return nil, err
 	}
 
-	return msgData, nil
+	return msgData[:msgLen], nil
 }
 
 // goroutine safe
@@ -126,7 +126,7 @@ func (p *MsgParser) Write(conn *TCPConn, args ...[]byte) error {
 	}
 
 	//msg := make([]byte, uint32(p.lenMsgLen)+msgLen)
-	msg := p.MakeByteSlice(p.lenMsgLen+int(msgLen))
+	msg := p.MakeWriteByteSlice(p.lenMsgLen+int(msgLen))
 	// write len
 	switch p.lenMsgLen {
 	case 1: