Imp: use bytes.Bytes from memory pool

7f03717a · AlexStocks · f40d3daf · 7f03717a
Commit 7f03717a authored Jul 26, 2019 by AlexStocks
Hide whitespace changes
Inline Side-by-side

Showing with 9 additions and 2 deletions

session.go session.go +9 -2

No files found.
--- a/session.go
+++ b/session.go
@@ -621,9 +621,16 @@ func (s *session) handleTCPPackage() error {

 	// buf = make([]byte, maxReadBufLen)
 	bufp = gxbytes.GetBytes(maxReadBufLen)
-	defer gxbytes.PutBytes(bufp)
 	buf = *bufp
-	pktBuf = new(bytes.Buffer)
+
+	// pktBuf = new(bytes.Buffer)
+	pktBuf = gxbytes.GetBytesBuffer()
+
+	defer func() {
+		gxbytes.PutBytes(bufp)
+		gxbytes.PutBytesBuffer(pktBuf)
+	}()
+
 	conn = s.Connection.(*gettyTCPConn)
 	for {
 		if s.IsClosed() {