feat: separate TCP and UDP buffer pools in Client and Server for improved memory management

Author: yosebyte

internal/client.go

@@ -33,10 +33,16 @@ func NewClient(parsedURL *url.URL, logger *logs.Logger) (*Client, error) {
 		Common: Common{
 			logger:     logger,
 			signalChan: make(chan string, semaphoreLimit),
-			bufferPool: &sync.Pool{
+			tcpBufferPool: &sync.Pool{
 				New: func() any {
-					b := make([]byte, 1048576) // 1MB
-					return &b
+					buf := make([]byte, tcpDataBufSize)
+					return &buf
+				},
+			},
+			udpBufferPool: &sync.Pool{
+				New: func() any {
+					buf := make([]byte, udpDataBufSize)
+					return &buf
 				},
 			},
 		},

internal/server.go

@@ -35,10 +35,16 @@ func NewServer(parsedURL *url.URL, tlsCode string, tlsConfig *tls.Config, logger
 			tlsCode:    tlsCode,
 			logger:     logger,
 			signalChan: make(chan string, semaphoreLimit),
-			bufferPool: &sync.Pool{
+			tcpBufferPool: &sync.Pool{
 				New: func() any {
-					b := make([]byte, 1048576) // 1MB
-					return &b
+					buf := make([]byte, tcpDataBufSize)
+					return &buf
+				},
+			},
+			udpBufferPool: &sync.Pool{
+				New: func() any {
+					buf := make([]byte, udpDataBufSize)
+					return &buf
 				},
 			},
 		},