package stat import ( "net" "github.com/xtls/xray-core/features/stats" ) type UserIpRestriction struct { User string IpAddress net.IP Time int64 } type Connection interface { net.Conn } type CounterConnection struct { Connection ReadCounter stats.Counter WriteCounter stats.Counter } func (c *CounterConnection) Read(b []byte) (int, error) { nBytes, err := c.Connection.Read(b) if c.ReadCounter != nil { c.ReadCounter.Add(int64(nBytes)) } return nBytes, err } func (c *CounterConnection) Write(b []byte) (int, error) { nBytes, err := c.Connection.Write(b) if c.WriteCounter != nil { c.WriteCounter.Add(int64(nBytes)) } return nBytes, err }