Browse Source

Rename fields in StatCouterConnection

The fields will be used by Outbounds as well in the next commit
yuhan6665 5 years ago
parent
commit
f34c092a06
2 changed files with 9 additions and 9 deletions
  1. 3 3
      app/proxyman/inbound/worker.go
  2. 6 6
      transport/internet/connection.go

+ 3 - 3
app/proxyman/inbound/worker.go

@@ -91,9 +91,9 @@ func (w *tcpWorker) callback(conn internet.Connection) {
 	ctx = session.ContextWithContent(ctx, content)
 	if w.uplinkCounter != nil || w.downlinkCounter != nil {
 		conn = &internet.StatCouterConnection{
-			Connection: conn,
-			Uplink:     w.uplinkCounter,
-			Downlink:   w.downlinkCounter,
+			Connection:   conn,
+			ReadCounter:  w.uplinkCounter,
+			WriteCounter: w.downlinkCounter,
 		}
 	}
 	if err := w.proxy.Process(ctx, net.Network_TCP, conn, w.dispatcher); err != nil {

+ 6 - 6
transport/internet/connection.go

@@ -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
 }