429247f407
* [COMPLIANCE] Update Copyright and License Headers * chore: make gen to generate files with copyright headers * chore: update missed copyright headers --------- Co-authored-by: hashicorp-copywrite[bot] <110428419+hashicorp-copywrite[bot]@users.noreply.github.com> Co-authored-by: Justin Nguyen <justin.nguyen@hashicorp.com>
30 lines
773 B
Go
30 lines
773 B
Go
// Copyright IBM Corp. 2020, 2025
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package ratelimit
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/hashicorp/go-rate"
|
|
)
|
|
|
|
// Limiter is used to provide rate limiting. This is the interface implemented
|
|
// by both rate.Limiter and rate.NopLimiter.
|
|
type Limiter interface {
|
|
SetPolicyHeader(string, string, http.Header) error
|
|
SetUsageHeader(*rate.Quota, http.Header)
|
|
Allow(string, string, string, string) (bool, *rate.Quota, error)
|
|
Shutdown() error
|
|
}
|
|
|
|
// NewLimiter creates a rate.Limiter.
|
|
func NewLimiter(limits []rate.Limit, maxQuotas int) (*rate.Limiter, error) {
|
|
return rate.NewLimiter(
|
|
limits,
|
|
maxQuotas,
|
|
rate.WithQuotaStorageUsageMetric(rateLimitQuotaUsage),
|
|
rate.WithQuotaStorageCapacityMetric(rateLimitQuotaStorageCapacity),
|
|
)
|
|
}
|