add rate limiter

limiter/generator.go

@@ -0,0 +1,61 @@
+package limiter
+
+import (
+	"github.com/go-gost/core/limiter"
+)
+
+type RateLimitGenerator interface {
+	In() limiter.Limiter
+	Out() limiter.Limiter
+}
+
+type rateLimitGenerator struct {
+	in  int
+	out int
+}
+
+func NewRateLimitGenerator(in, out int) RateLimitGenerator {
+	return &rateLimitGenerator{
+		in:  in,
+		out: out,
+	}
+}
+
+func (p *rateLimitGenerator) In() limiter.Limiter {
+	if p == nil || p.in <= 0 {
+		return nil
+	}
+	return NewLimiter(p.in)
+}
+
+func (p *rateLimitGenerator) Out() limiter.Limiter {
+	if p == nil || p.out <= 0 {
+		return nil
+	}
+	return NewLimiter(p.out)
+}
+
+type rateLimitSingleGenerator struct {
+	in  limiter.Limiter
+	out limiter.Limiter
+}
+
+func NewRateLimitSingleGenerator(in, out int) RateLimitGenerator {
+	p := &rateLimitSingleGenerator{}
+	if in > 0 {
+		p.in = NewLimiter(in)
+	}
+	if out > 0 {
+		p.out = NewLimiter(out)
+	}
+
+	return p
+}
+
+func (p *rateLimitSingleGenerator) In() limiter.Limiter {
+	return p.in
+}
+
+func (p *rateLimitSingleGenerator) Out() limiter.Limiter {
+	return p.out
+}