Xray-core/transport/internet/stat/connection.go

41 lines
670 B
Go
Raw Normal View History

package stat
2020-11-25 11:01:53 +00:00
import (
"net"
2020-12-04 01:36:16 +00:00
"github.com/xtls/xray-core/features/stats"
2020-11-25 11:01:53 +00:00
)
type UserIpRestriction struct {
User string
IpAddress net.IP
Time int64
}
2020-11-25 11:01:53 +00:00
type Connection interface {
net.Conn
}
type CounterConnection struct {
2020-11-25 11:01:53 +00:00
Connection
ReadCounter stats.Counter
WriteCounter stats.Counter
}
func (c *CounterConnection) Read(b []byte) (int, error) {
2020-11-25 11:01:53 +00:00
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) {
2020-11-25 11:01:53 +00:00
nBytes, err := c.Connection.Write(b)
if c.WriteCounter != nil {
c.WriteCounter.Add(int64(nBytes))
}
return nBytes, err
}