mtg/utils/init_tcp.go

package utils

import (
	"fmt"
	"net"
	"time"

	"github.com/9seconds/mtg/config"
)

const tcpKeepAlivePingPeriod = 2 * time.Second

func InitTCP(conn net.Conn) 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 {
		return fmt.Errorf("cannot set read buffer size: %w", err)
	}

	if err := tcpConn.SetWriteBuffer(config.C.WriteBuffer); err != nil {
		return fmt.Errorf("cannot set write buffer size: %w", err)
	}

	if err := tcpConn.SetKeepAlive(true); err != nil {
		return fmt.Errorf("cannot enable keep-alive: %w", err)
	}

	if err := tcpConn.SetKeepAlivePeriod(tcpKeepAlivePingPeriod); err != nil {
		return fmt.Errorf("cannot set keep-alive period: %w", err)
	}

	return nil
}