ratelimit

package

v0.1.1 Latest Latest Go to latest Published: Mar 26, 2025 License: MIT Imports: 5 Imported by: 0

Details

This section is empty.

This section is empty.

func RateLimit(opts ...Option) middleware.Middleware

RateLimit 返回一个限流中间件

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

IPRateLimiter 是基于IP的限流器

func NewIPRateLimiter(rate, capacity float64) *IPRateLimiter

NewIPRateLimiter 创建一个新的基于IP的限流器

func (l *IPRateLimiter) Allow(ctx context.Context) (bool, error)

Allow 实现Limiter接口

type Limiter interface {
	// Allow 判断请求是否被允许
	Allow(ctx context.Context) (bool, error)
}

Limiter 是限流器接口

type Option func(*options)

Option 是限流中间件的选项

func WithLimiter(limiter Limiter) Option

WithLimiter 设置限流器

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

TokenBucket 是令牌桶限流器

func NewTokenBucket(rate, capacity float64) *TokenBucket

NewTokenBucket 创建一个新的令牌桶限流器

func (l *TokenBucket) Allow(ctx context.Context) (bool, error)

Allow 实现Limiter接口