From f0d2cd494a8d06dfdaa2a31c944e0f7bd7f38177 Mon Sep 17 00:00:00 2001 From: 9seconds Date: Mon, 23 Mar 2020 09:05:18 +0300 Subject: [PATCH] Use stack allocation instead of heap one --- proxy/direct.go | 14 ++------------ 1 file changed, 2 insertions(+), 12 deletions(-) diff --git a/proxy/direct.go b/proxy/direct.go index 6d01c0d..2d13a13 100644 --- a/proxy/direct.go +++ b/proxy/direct.go @@ -13,15 +13,6 @@ import ( const directPipeBufferSize = 1024 -var ( - directPipePool = sync.Pool{ - New: func() interface{} { - rv := make([]byte, directPipeBufferSize) - return &rv - }, - } -) - func directConnection(request *protocol.TelegramRequest) error { telegramConnRaw, err := obfuscated2.TelegramProtocol(request) if err != nil { @@ -51,10 +42,9 @@ func directPipe(dst io.WriteCloser, src io.ReadCloser, wg *sync.WaitGroup, logge wg.Done() }() - buf := directPipePool.Get().(*[]byte) - defer directPipePool.Put(buf) + buf := [directPipeBufferSize]byte{} - if _, err := io.CopyBuffer(dst, src, *buf); err != nil { + if _, err := io.CopyBuffer(dst, src, buf[:]); err != nil { logger.Debugw("Cannot pump sockets", "error", err) } }