add traffic reset for stats

observer/stats/stats.go

@@ -0,0 +1,102 @@
+package stats
+
+import (
+	"sync/atomic"
+
+	"github.com/go-gost/core/observer"
+	"github.com/go-gost/core/observer/stats"
+)
+
+type Stats struct {
+	updated      atomic.Bool
+	totalConns   atomic.Uint64
+	currentConns atomic.Uint64
+	inputBytes   atomic.Uint64
+	outputBytes  atomic.Uint64
+	totalErrs    atomic.Uint64
+	resetTraffic bool
+}
+
+func NewStats(resetTraffic bool) stats.Stats {
+	return &Stats{
+		resetTraffic: resetTraffic,
+	}
+}
+
+func (s *Stats) Add(kind stats.Kind, n int64) {
+	if s == nil {
+		return
+	}
+	switch kind {
+	case stats.KindTotalConns:
+		if n > 0 {
+			s.totalConns.Add(uint64(n))
+		}
+	case stats.KindCurrentConns:
+		s.currentConns.Add(uint64(n))
+	case stats.KindInputBytes:
+		s.inputBytes.Add(uint64(n))
+	case stats.KindOutputBytes:
+		s.outputBytes.Add(uint64(n))
+	case stats.KindTotalErrs:
+		if n > 0 {
+			s.totalErrs.Add(uint64(n))
+		}
+	}
+	s.updated.Store(true)
+}
+
+func (s *Stats) Get(kind stats.Kind) uint64 {
+	if s == nil {
+		return 0
+	}
+
+	switch kind {
+	case stats.KindTotalConns:
+		return s.totalConns.Load()
+	case stats.KindCurrentConns:
+		return s.currentConns.Load()
+	case stats.KindInputBytes:
+		if s.resetTraffic {
+			return s.inputBytes.Swap(0)
+		}
+		return s.inputBytes.Load()
+	case stats.KindOutputBytes:
+		if s.resetTraffic {
+			return s.outputBytes.Swap(0)
+		}
+		return s.outputBytes.Load()
+	case stats.KindTotalErrs:
+		return s.totalErrs.Load()
+	}
+	return 0
+}
+
+func (s *Stats) Reset() {
+	s.updated.Store(false)
+	s.totalConns.Store(0)
+	s.currentConns.Store(0)
+	s.inputBytes.Store(0)
+	s.outputBytes.Store(0)
+	s.totalErrs.Store(0)
+}
+
+func (s *Stats) IsUpdated() bool {
+	return s.updated.Swap(false)
+}
+
+type StatsEvent struct {
+	Kind    string
+	Service string
+	Client  string
+
+	TotalConns   uint64
+	CurrentConns uint64
+	InputBytes   uint64
+	OutputBytes  uint64
+	TotalErrs    uint64
+}
+
+func (StatsEvent) Type() observer.EventType {
+	return observer.EventStats
+}

observer/stats/wrapper/conn.go

@@ -19,12 +19,12 @@ var (
 
 type conn struct {
 	net.Conn
-	stats  *stats.Stats
+	stats  stats.Stats
 	closed chan struct{}
 	mu     sync.Mutex
 }
 
-func WrapConn(c net.Conn, pStats *stats.Stats) net.Conn {
+func WrapConn(c net.Conn, pStats stats.Stats) net.Conn {
 	if pStats == nil {
 		return c
 	}
@@ -84,10 +84,10 @@ func (c *conn) Metadata() metadata.Metadata {
 
 type packetConn struct {
 	net.PacketConn
-	stats *stats.Stats
+	stats stats.Stats
 }
 
-func WrapPacketConn(pc net.PacketConn, stats *stats.Stats) net.PacketConn {
+func WrapPacketConn(pc net.PacketConn, stats stats.Stats) net.PacketConn {
 	if stats == nil {
 		return pc
 	}
@@ -118,10 +118,10 @@ func (c *packetConn) Metadata() metadata.Metadata {
 
 type udpConn struct {
 	net.PacketConn
-	stats *stats.Stats
+	stats stats.Stats
 }
 
-func WrapUDPConn(pc net.PacketConn, stats *stats.Stats) udp.Conn {
+func WrapUDPConn(pc net.PacketConn, stats stats.Stats) udp.Conn {
 	return &udpConn{
 		PacketConn: pc,
 		stats:      stats,

observer/stats/wrapper/io.go

@@ -9,10 +9,10 @@ import (
 // readWriter is an io.ReadWriter with Stats.
 type readWriter struct {
 	io.ReadWriter
-	stats *stats.Stats
+	stats stats.Stats
 }
 
-func WrapReadWriter(rw io.ReadWriter, stats *stats.Stats) io.ReadWriter {
+func WrapReadWriter(rw io.ReadWriter, stats stats.Stats) io.ReadWriter {
 	if stats == nil {
 		return rw
 	}

observer/stats/wrapper/listener.go

@@ -7,11 +7,11 @@ import (
 )
 
 type listener struct {
-	stats *stats.Stats
+	stats stats.Stats
 	net.Listener
 }
 
-func WrapListener(ln net.Listener, stats *stats.Stats) net.Listener {
+func WrapListener(ln net.Listener, stats stats.Stats) net.Listener {
 	if stats == nil {
 		return ln
 	}