baalajimaestro/traefik
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity Actions

feat: rate-limit ceil Retry-After to superior integer

Browse source
This commit is contained in:
Pierre-Yves Aillet 2021-11-16 16:38:11 +01:00 • committed by GitHub
parent 83a7f10c75
commit 95fabeae73
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
pkg/middlewares/ratelimiter/rate_limiter.go
View file

@ -4,6 +4,7 @@ package ratelimiter
import ( import (
"context" "context"
"fmt" "fmt"
"math"
"net/http" "net/http"
"time" "time"
@ -174,7 +175,7 @@ func (rl *rateLimiter) ServeHTTP(w http.ResponseWriter, r *http.Request) {
} }
func (rl *rateLimiter) serveDelayError(ctx context.Context, w http.ResponseWriter, r *http.Request, delay time.Duration) { func (rl *rateLimiter) serveDelayError(ctx context.Context, w http.ResponseWriter, r *http.Request, delay time.Duration) {
w.Header().Set("Retry-After", fmt.Sprintf("%.0f", delay.Seconds())) w.Header().Set("Retry-After", fmt.Sprintf("%.0f", math.Ceil(delay.Seconds())))
w.Header().Set("X-Retry-In", delay.String()) w.Header().Set("X-Retry-In", delay.String())
w.WriteHeader(http.StatusTooManyRequests) w.WriteHeader(http.StatusTooManyRequests)