limiter

README

限流器

var limiter = NewIPRateLimiter(1, 5, 0)

rateLimiter := limiter.GetLimiter(r.RemoteAddr)
if !rateLimiter.Allow() {
http.Error(w, http.StatusText(http.StatusTooManyRequests), http.StatusTooManyRequests)
return
}

Documentation

Index

• func Allow() bool
• type KeyRateLimiter
  • func NewKeyRateLimiter(r float64, b int, t int) *KeyRateLimiter
  • func (k *KeyRateLimiter) GetLimiter(key string) *Limiter
• type Limiter
  • func NewLimiter(r float64, cap int, timeout int) *Limiter
  • func (l *Limiter) Allow() bool

Functions

func Allow

func Allow() bool

Types

type KeyRateLimiter

type KeyRateLimiter struct {
	// contains filtered or unexported fields
}

func NewKeyRateLimiter

func NewKeyRateLimiter(r float64, b int, t int) *KeyRateLimiter

func (*KeyRateLimiter) GetLimiter

func (k *KeyRateLimiter) GetLimiter(key string) *Limiter

type Limiter

type Limiter struct {
	// contains filtered or unexported fields
}

func NewLimiter

func NewLimiter(r float64, cap int, timeout int) *Limiter

NewLimiter r为放入桶中的速率/秒 cap位桶容量

r 每秒流入桶内的数量

cap 桶容量

timeout 等待时长[毫秒]

func (*Limiter) Allow

func (l *Limiter) Allow() bool

Allow 不等待 直接返回是否允许通过