mirror of https://github.com/v2ray/v2ray-core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
37 lines
571 B
37 lines
571 B
package internet
|
|
|
|
import (
|
|
"net"
|
|
)
|
|
|
|
type Connection interface {
|
|
net.Conn
|
|
}
|
|
|
|
type addInt64 interface {
|
|
Add(int64) int64
|
|
}
|
|
|
|
type StatCouterConnection struct {
|
|
Connection
|
|
Uplink addInt64
|
|
Downlink addInt64
|
|
}
|
|
|
|
func (c *StatCouterConnection) Read(b []byte) (int, error) {
|
|
nBytes, err := c.Connection.Read(b)
|
|
if c.Uplink != nil {
|
|
c.Uplink.Add(int64(nBytes))
|
|
}
|
|
|
|
return nBytes, err
|
|
}
|
|
|
|
func (c *StatCouterConnection) Write(b []byte) (int, error) {
|
|
nBytes, err := c.Connection.Write(b)
|
|
if c.Downlink != nil {
|
|
c.Downlink.Add(int64(nBytes))
|
|
}
|
|
return nBytes, err
|
|
}
|