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