Files
tasks/common/quotas/request_rate_limiter_delegator.go
Alex ShtinandGitHub 8e8f1b0f7c Remove license header from every file (#7689)
## What changed?
<!-- Describe what has changed in this PR -->
Remove license header from every file. Because it is really hard to
follow in this PR here is the summary:
1. License header is removed from all `.go` and `.proto` files
:fireworks::fireworks:🎆.
2. `LICENSE` file in the root directory has only Temporal and Uber
copyrights.
3. 5 other `LICENSE` files added to the packages which have copyrights
different from Temporal and Uber: Datadog, Xargin, "Mat Ryer, Tyler
Bunnell and contributors".
4. `license_file` flag is removed from all code generation tools.
5. `copyright_file` flag is removed from `go:generate mockgen`
directive.
6. All copyright related targets are removed from `Makefile`.
7. Updated Temporal copyright year to 2025 everywhere.

## Why?
<!-- Tell your future self why have you made these changes -->
I double checked with legal department that it is not needed to have
license header in every file. One file per repo is enough. I put all
copyrights to the root `LICENSE` file and removed header from all other
files. Also updated tools and `Makefile`.
2025-05-01 18:50:21 -07:00

48 lines
1.8 KiB
Go

package quotas
import (
"context"
"sync/atomic"
"time"
)
// RequestRateLimiterDelegator is a request rate limiter that delegates to another rate limiter. The delegate can be
// changed at runtime by calling SetRateLimiter. This rate limiter is useful for cases where you want to substitute one
// rate limiter implementation for another at runtime. All methods of this type are thread-safe.
type RequestRateLimiterDelegator struct {
// delegate is an atomic.Value so that it can be safely read from and written to concurrently. It stores the rate
// limiter that this rate limiter delegates to as a monomorphicRequestRateLimiter.
delegate atomic.Value
}
// monomorphicRequestRateLimiter is a workaround for the fact that the value stored in an atomic.Value must always be
// the same type, but we want to allow the rate limiter delegate to be any type that implements the RequestRateLimiter
// interface.
type monomorphicRequestRateLimiter struct {
RequestRateLimiter
}
// SetRateLimiter sets the rate limiter to delegate to.
func (d *RequestRateLimiterDelegator) SetRateLimiter(rl RequestRateLimiter) {
d.delegate.Store(monomorphicRequestRateLimiter{rl})
}
// loadDelegate returns the rate limiter that this rate limiter delegates to.
func (d *RequestRateLimiterDelegator) loadDelegate() RequestRateLimiter {
return d.delegate.Load().(RequestRateLimiter)
}
// The following methods just delegate to the underlying rate limiter.
func (d *RequestRateLimiterDelegator) Allow(now time.Time, request Request) bool {
return d.loadDelegate().Allow(now, request)
}
func (d *RequestRateLimiterDelegator) Reserve(now time.Time, request Request) Reservation {
return d.loadDelegate().Reserve(now, request)
}
func (d *RequestRateLimiterDelegator) Wait(ctx context.Context, request Request) error {
return d.loadDelegate().Wait(ctx, request)
}