conntrack

type ConnSuccessRateStats struct {
	CalculatedAt    time.Time
	ConnSuccessRate float64
	TotalConns      int
}

type ConnSuccessRateTracker struct {
	ConnAttempts         *cache.Cache
	ConnSuccessRateStats atomic.Value
}

type InstrumentedConn struct {
	net.Conn
	Role         string
	OutboundHost string

	ConnError error

	Start        time.Time
	LastActivity *int64 // Unix nano

	BytesIn  *uint64
	BytesOut *uint64

	sync.Mutex

	CloseError error
	// contains filtered or unexported fields
}

type InstrumentedConnStats struct {
	Role                     string    `json:"role"`
	Rhost                    string    `json:"rhost"`
	Raddr                    string    `json:"raddr"`
	Created                  time.Time `json:"created"`
	BytesIn                  uint64    `json:"bytesIn"`
	BytesOut                 uint64    `json:"bytesOut"`
	SecondsSinceLastActivity float64   `json:"secondsSinceLastActivity"`
	ProxyType                string    `json:"proxyType"`
}

type Tracker struct {
	*sync.Map
	ShuttingDown atomic.Value

	SuccessRateTracker *ConnSuccessRateTracker

	// A connection is idle if it has been inactive (no bytes in/out) for this
	// many seconds.
	IdleTimeout time.Duration
	// contains filtered or unexported fields
}

type TrackerInterface interface {
	ReportConnectionSuccessRate() *ConnSuccessRateStats
	RecordAttempt(string, bool)
	MaybeIdleIn(time.Duration) time.Duration
	NewInstrumentedConn(net.Conn, *logrus.Entry, string, string, string) *InstrumentedConn
	NewInstrumentedConnWithTimeout(net.Conn, time.Duration, *logrus.Entry, string, string, string) *InstrumentedConn
	Wg() *sync.WaitGroup
	Range(f func(interface{}, interface{}) bool)
}