mtg/internal/proxyprotocol/conn.go

package proxyprotocol

import (
	"errors"
	"syscall"

	"github.com/pires/go-proxyproto"
)

type connWrapper struct {
	*proxyproto.Conn
}

func (c connWrapper) CloseRead() error {
	tcpConn, ok := c.TCPConn()
	if !ok {
		panic("we support only tcp connections")
	}

	return tcpConn.CloseRead()
}

func (c connWrapper) CloseWrite() error {
	tcpConn, ok := c.TCPConn()
	if !ok {
		panic("we support only tcp connections")
	}

	return tcpConn.CloseWrite()
}

func (c connWrapper) SyscallConn() (syscall.RawConn, error) {
	tcpConn, ok := c.TCPConn()
	if !ok {
		return nil, errors.New("proxy protocol connection is not TCP")
	}

	return tcpConn.SyscallConn()
}