TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-11-19 20:40:26 +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() { go func() {
originBuffer := make([]byte, 4*1024) // TODO 需要可以设置，并可以使用Pool originBuffer := bytePool32k.Get()
defer func() {
bytePool32k.Put(originBuffer)
}()
for { for {
n, err := originConn.Read(originBuffer) n, err := originConn.Read(originBuffer)
if n > 0 { 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 { for {
n, err := conn.Read(clientBuffer) n, err := conn.Read(clientBuffer)
if n > 0 { if n > 0 {