forked from argoproj/argo-cd
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: use rate limited queue (argoproj#15480)
* feat: use rate limited queue Signed-off-by: Soumya Ghosh Dastidar <gdsoumya@gmail.com> Signed-off-by: jmilic1 <70441727+jmilic1@users.noreply.github.com>
- Loading branch information
Showing
8 changed files
with
247 additions
and
17 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,123 @@ | ||
| package ratelimiter | ||
|
|
||
| import ( | ||
| "math" | ||
| "sync" | ||
| "time" | ||
|
|
||
| "golang.org/x/time/rate" | ||
| "k8s.io/client-go/util/workqueue" | ||
| ) | ||
|
|
||
| type AppControllerRateLimiterConfig struct { | ||
| BucketSize int64 | ||
| BucketQPS int64 | ||
| FailureCoolDown time.Duration | ||
| BaseDelay time.Duration | ||
| MaxDelay time.Duration | ||
| BackoffFactor float64 | ||
| } | ||
|
|
||
| func GetDefaultAppRateLimiterConfig() *AppControllerRateLimiterConfig { | ||
| return &AppControllerRateLimiterConfig{ | ||
| // global queue rate limit config | ||
| 500, | ||
| 50, | ||
| // individual item rate limit config | ||
| // when WORKQUEUE_FAILURE_COOLDOWN is 0 per item rate limiting is disabled(default) | ||
| 0, | ||
| time.Millisecond, | ||
| time.Second, | ||
| 1.5, | ||
| } | ||
| } | ||
|
|
||
| // NewCustomAppControllerRateLimiter is a constructor for the rate limiter for a workqueue used by app controller. It has | ||
| // both overall and per-item rate limiting. The overall is a token bucket and the per-item is exponential(with auto resets) | ||
| func NewCustomAppControllerRateLimiter(cfg *AppControllerRateLimiterConfig) workqueue.RateLimiter { | ||
| return workqueue.NewMaxOfRateLimiter( | ||
| NewItemExponentialRateLimiterWithAutoReset(cfg.BaseDelay, cfg.MaxDelay, cfg.FailureCoolDown, cfg.BackoffFactor), | ||
| &workqueue.BucketRateLimiter{Limiter: rate.NewLimiter(rate.Limit(cfg.BucketQPS), int(cfg.BucketSize))}, | ||
| ) | ||
| } | ||
|
|
||
| type failureData struct { | ||
| failures int | ||
| lastFailure time.Time | ||
| } | ||
|
|
||
| // ItemExponentialRateLimiterWithAutoReset does a simple baseDelay*2^<num-failures> limit | ||
| // dealing with max failures and expiration/resets are up dependent on the cooldown period | ||
| type ItemExponentialRateLimiterWithAutoReset struct { | ||
| failuresLock sync.Mutex | ||
| failures map[interface{}]failureData | ||
|
|
||
| baseDelay time.Duration | ||
| maxDelay time.Duration | ||
| coolDown time.Duration | ||
| backoffFactor float64 | ||
| } | ||
|
|
||
| var _ workqueue.RateLimiter = &ItemExponentialRateLimiterWithAutoReset{} | ||
|
|
||
| func NewItemExponentialRateLimiterWithAutoReset(baseDelay, maxDelay, failureCoolDown time.Duration, backoffFactor float64) workqueue.RateLimiter { | ||
| return &ItemExponentialRateLimiterWithAutoReset{ | ||
| failures: map[interface{}]failureData{}, | ||
| baseDelay: baseDelay, | ||
| maxDelay: maxDelay, | ||
| coolDown: failureCoolDown, | ||
| backoffFactor: backoffFactor, | ||
| } | ||
| } | ||
|
|
||
| func (r *ItemExponentialRateLimiterWithAutoReset) When(item interface{}) time.Duration { | ||
| r.failuresLock.Lock() | ||
| defer r.failuresLock.Unlock() | ||
|
|
||
| if _, ok := r.failures[item]; !ok { | ||
| r.failures[item] = failureData{ | ||
| failures: 0, | ||
| lastFailure: time.Now(), | ||
| } | ||
| } | ||
|
|
||
| exp := r.failures[item] | ||
|
|
||
| // if coolDown period is reached reset failures for item | ||
| if time.Since(exp.lastFailure) >= r.coolDown { | ||
| delete(r.failures, item) | ||
| return r.baseDelay | ||
| } | ||
|
|
||
| r.failures[item] = failureData{ | ||
| failures: exp.failures + 1, | ||
| lastFailure: time.Now(), | ||
| } | ||
|
|
||
| // The backoff is capped such that 'calculated' value never overflows. | ||
| backoff := float64(r.baseDelay.Nanoseconds()) * math.Pow(r.backoffFactor, float64(exp.failures)) | ||
| if backoff > math.MaxInt64 { | ||
| return r.maxDelay | ||
| } | ||
|
|
||
| calculated := time.Duration(backoff) | ||
| if calculated > r.maxDelay { | ||
| return r.maxDelay | ||
| } | ||
|
|
||
| return calculated | ||
| } | ||
|
|
||
| func (r *ItemExponentialRateLimiterWithAutoReset) NumRequeues(item interface{}) int { | ||
| r.failuresLock.Lock() | ||
| defer r.failuresLock.Unlock() | ||
|
|
||
| return r.failures[item].failures | ||
| } | ||
|
|
||
| func (r *ItemExponentialRateLimiterWithAutoReset) Forget(item interface{}) { | ||
| r.failuresLock.Lock() | ||
| defer r.failuresLock.Unlock() | ||
|
|
||
| delete(r.failures, item) | ||
| } |
Oops, something went wrong.