TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-11-03 23:20:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

TCP代理：使用pool管理buffer

Browse Source
This commit is contained in:
GoEdgeLab
2021-01-14 16:43:59 +08:00
parent 2c413ce039
commit 3731c8b396
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
internal/nodes/listener_tcp.go
View File

@@ -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 {