init

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

@@ -0,0 +1,42 @@
+package conn
+
+import (
+	limiter "github.com/go-gost/core/limiter/conn"
+)
+
+type ConnLimitGenerator interface {
+	Limiter() limiter.Limiter
+}
+
+type connLimitGenerator struct {
+	n int
+}
+
+func NewConnLimitGenerator(n int) ConnLimitGenerator {
+	return &connLimitGenerator{
+		n: n,
+	}
+}
+
+func (p *connLimitGenerator) Limiter() limiter.Limiter {
+	if p == nil || p.n <= 0 {
+		return nil
+	}
+	return NewLimiter(p.n)
+}
+
+type connLimitSingleGenerator struct {
+	limiter limiter.Limiter
+}
+
+func NewConnLimitSingleGenerator(n int) ConnLimitGenerator {
+	p := &connLimitSingleGenerator{}
+	if n > 0 {
+		p.limiter = NewLimiter(n)
+	}
+	return p
+}
+
+func (p *connLimitSingleGenerator) Limiter() limiter.Limiter {
+	return p.limiter
+}