流量统计

go-gost/x/limiter/traffic/generator.go

@@ -0,0 +1,31 @@
+package traffic
+
+import (
+	limiter "github.com/go-gost/core/limiter/traffic"
+)
+
+type limitGenerator struct {
+	in  int
+	out int
+}
+
+func newLimitGenerator(in, out int) *limitGenerator {
+	return &limitGenerator{
+		in:  in,
+		out: out,
+	}
+}
+
+func (p *limitGenerator) In() limiter.Limiter {
+	if p == nil || p.in <= 0 {
+		return nil
+	}
+	return NewLimiter(p.in)
+}
+
+func (p *limitGenerator) Out() limiter.Limiter {
+	if p == nil || p.out <= 0 {
+		return nil
+	}
+	return NewLimiter(p.out)
+}