mirror of
https://github.com/hanzoai/gateway.git
synced 2026-10-30 15:04:44 +00:00
zeekayandHanzo Dev
a5c5cc019e
Modules declare go 1.27.1 and the build images follow. go fix applied the modernizers; omitzero is not applied, since it changes what the JSON says. Co-authored-by: Hanzo Dev <[email protected]>
41 lines
1.0 KiB
Go
41 lines
1.0 KiB
Go
// krakendrate contains a collection of curated rate limit adaptors for the KrakenD framework
|
|
package krakendrate
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
// ErrLimited is the error returned when the rate limit has been exceded
|
|
ErrLimited = errors.New("rate limit exceded")
|
|
|
|
// DataTTL is the default eviction time
|
|
DataTTL = 10 * time.Minute
|
|
|
|
// DefaultShards are the number of shards to create by default
|
|
DefaultShards uint64 = 2048
|
|
|
|
now = time.Now
|
|
)
|
|
|
|
// Limiter defines a simple interface for a rate limiter
|
|
type Limiter interface {
|
|
Allow() bool
|
|
}
|
|
|
|
// LimiterStore defines the interface for a limiter lookup function
|
|
type LimiterStore func(string) Limiter
|
|
|
|
// Backend is the interface of the persistence layer
|
|
type Backend interface {
|
|
Load(string, func() any) any
|
|
Store(string, any) error
|
|
}
|
|
|
|
// DefaultShardedMemoryBackend is a 2048 sharded ShardedMemoryBackend
|
|
func DefaultShardedMemoryBackend(ctx context.Context) *ShardedMemoryBackend {
|
|
return NewShardedMemoryBackend(ctx, DefaultShards, DataTTL, PseudoFNV64a)
|
|
}
|