ratelimit

package

v0.12.0 Latest Latest Go to latest Published: Jun 23, 2026 License: MIT Imports: 4 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

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

APIKeyRateLimiter manages rate limits per API key

func NewAPIKeyRateLimiter() *APIKeyRateLimiter

NewAPIKeyRateLimiter creates a new API key-based rate limiter

func (rl *APIKeyRateLimiter) Allow(apiKey string) (bool, error)

Allow checks if a request with an API key is allowed

func (rl *APIKeyRateLimiter) SetLimit(apiKey string, config RateLimitConfig)

SetLimit sets the rate limit for an API key

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

IPRateLimiter manages rate limits per IP address

func NewIPRateLimiter(maxTokens, refillRate float64) *IPRateLimiter

NewIPRateLimiter creates a new IP-based rate limiter

func (rl *IPRateLimiter) Allow(ip string) bool

Allow checks if a request from an IP is allowed

type RateLimitConfig struct {
	MaxTokens  float64
	RefillRate float64
	BurstSize  int
}

RateLimitConfig defines rate limit configuration

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

RateLimiter implements a token bucket rate limiter

func NewRateLimiter(maxTokens float64, refillRate float64) *RateLimiter

NewRateLimiter creates a new rate limiter

func (rl *RateLimiter) Allow() bool

Allow checks if a request is allowed

func (rl *RateLimiter) AllowN(n float64) bool

AllowN checks if n requests are allowed

func (rl *RateLimiter) Wait(ctx context.Context) error

Wait blocks until a token is available