Merge pull request #392 from xxx7xxxx/fix-ratelimiter-fallback

Return resultFallback when it happened
easegress-io · Sep 6, 2019 · bcf3769 · bcf3769
2 parents 5c84e71 + e901680
commit bcf3769
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pkg/plugin/ratelimiter/ratelimiter.go b/pkg/plugin/ratelimiter/ratelimiter.go
@@ -18,7 +18,8 @@ const (
  // Kind is the kind of RateLimiter.
  Kind = "RateLimiter"
 
- resultTimeout = "timeout"
+ resultTimeout = "timeout"
+ resultFallback = "fallback"
 )
 
 func init() {
@@ -121,6 +122,7 @@ func (rl *RateLimiter) Handle(ctx context.HTTPContext) string {
  if err != nil {
  if rl.fallback != nil {
  rl.fallback.Fallback(ctx)
+ return resultFallback
  }
  ctx.Response().SetStatusCode(http.StatusTooManyRequests)
  return resultTimeout