Add possibility to scale middleproxy buffers

This commit is contained in:
9seconds
2020-03-30 16:53:14 +03:00
parent 35f0fca3c5
commit b5a8d8b804
5 changed files with 52 additions and 9 deletions
+3 -5
View File
@@ -4,24 +4,22 @@ import (
"fmt"
"net"
"time"
"github.com/9seconds/mtg/config"
)
const tcpKeepAlivePingPeriod = 2 * time.Second
func InitTCP(conn net.Conn) error {
func InitTCP(conn net.Conn, readBufferSize int, writeBufferSize int) error {
tcpConn := conn.(*net.TCPConn)
if err := tcpConn.SetNoDelay(true); err != nil {
return fmt.Errorf("cannot set TCP_NO_DELAY: %w", err)
}
if err := tcpConn.SetReadBuffer(config.C.ReadBuffer); err != nil {
if err := tcpConn.SetReadBuffer(readBufferSize); err != nil {
return fmt.Errorf("cannot set read buffer size: %w", err)
}
if err := tcpConn.SetWriteBuffer(config.C.WriteBuffer); err != nil {
if err := tcpConn.SetWriteBuffer(writeBufferSize); err != nil {
return fmt.Errorf("cannot set write buffer size: %w", err)
}