From 3731c8b39613edabb071edee4d374fc7a21dd3e1 Mon Sep 17 00:00:00 2001
From: GoEdgeLab <goedgecloud@gmail.com>
Date: Thu, 14 Jan 2021 16:43:59 +0800
Subject: [PATCH] =?UTF-8?q?TCP=E4=BB=A3=E7=90=86=EF=BC=9A=E4=BD=BF?=
 =?UTF-8?q?=E7=94=A8pool=E7=AE=A1=E7=90=86buffer?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/nodes/listener_tcp.go | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/internal/nodes/listener_tcp.go b/internal/nodes/listener_tcp.go
index 1946a40..18bb062 100644
--- a/internal/nodes/listener_tcp.go
+++ b/internal/nodes/listener_tcp.go
@@ -65,7 +65,10 @@ func (this *TCPListener) handleConn(conn net.Conn) error {
 	}
 
 	go func() {
-		originBuffer := make([]byte, 4*1024) // TODO 需要可以设置，并可以使用Pool
+		originBuffer := bytePool32k.Get()
+		defer func() {
+			bytePool32k.Put(originBuffer)
+		}()
 		for {
 			n, err := originConn.Read(originBuffer)
 			if n > 0 {
@@ -85,7 +88,10 @@ func (this *TCPListener) handleConn(conn net.Conn) error {
 		}
 	}()
 
-	clientBuffer := make([]byte, 4*1024) // TODO 需要可以设置，并可以使用Pool
+	clientBuffer := bytePool32k.Get()
+	defer func() {
+		bytePool32k.Put(clientBuffer)
+	}()
 	for {
 		n, err := conn.Read(clientBuffer)
 		if n > 0 {