Skip to content

Commit 66b2085

Browse files
committed
fixed ratelimiter bucket
1 parent 33673c1 commit 66b2085

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

pkg/middlewares/canary/canary.go

+1-1
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ const (
2323
headerAuth = "Authorization"
2424
headerUA = "User-Agent"
2525
headerXCanary = "X-Canary"
26-
headerXRequestID = "X-Request-ID"
26+
headerXRequestID = "X-Request-Id"
2727
defaultCacheSize = 100000
2828
defaultExpiration = time.Minute * 10
2929
defaultCacheCleanDuration = time.Minute * 20

pkg/middlewares/ratelimiter/rate_limiter.go

+1-1
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ func (rl *rateLimiter) ServeHTTP(w http.ResponseWriter, r *http.Request) {
121121
bucket = rlSource.(*rate.Limiter)
122122
} else {
123123
bucket = rate.NewLimiter(rl.rate, int(rl.burst))
124-
if err := rl.buckets.Set(source, bucket, int(rl.maxDelay)*10+1); err != nil {
124+
if err := rl.buckets.Set(source, bucket, int(rl.maxDelay/time.Second)*10+1); err != nil {
125125
logger.Errorf("could not insert bucket: %v", err)
126126
http.Error(w, "could not insert bucket", http.StatusInternalServerError)
127127
return

0 commit comments

Comments
 (0)