|
|
@@ -12,14 +12,14 @@ type Connection interface {
|
|
|
|
|
|
type StatCouterConnection struct {
|
|
|
Connection
|
|
|
- Uplink stats.Counter
|
|
|
- Downlink stats.Counter
|
|
|
+ ReadCounter stats.Counter
|
|
|
+ WriteCounter stats.Counter
|
|
|
}
|
|
|
|
|
|
func (c *StatCouterConnection) Read(b []byte) (int, error) {
|
|
|
nBytes, err := c.Connection.Read(b)
|
|
|
- if c.Uplink != nil {
|
|
|
- c.Uplink.Add(int64(nBytes))
|
|
|
+ if c.ReadCounter != nil {
|
|
|
+ c.ReadCounter.Add(int64(nBytes))
|
|
|
}
|
|
|
|
|
|
return nBytes, err
|
|
|
@@ -27,8 +27,8 @@ func (c *StatCouterConnection) Read(b []byte) (int, error) {
|
|
|
|
|
|
func (c *StatCouterConnection) Write(b []byte) (int, error) {
|
|
|
nBytes, err := c.Connection.Write(b)
|
|
|
- if c.Downlink != nil {
|
|
|
- c.Downlink.Add(int64(nBytes))
|
|
|
+ if c.WriteCounter != nil {
|
|
|
+ c.WriteCounter.Add(int64(nBytes))
|
|
|
}
|
|
|
return nBytes, err
|
|
|
}
|