Skip to main content

KeyedRatePolicy

Struct KeyedRatePolicy 

Source
pub struct KeyedRatePolicy<X, K> { /* private fields */ }
Available on crate feature std only.
Expand description

A limit Policy that rate limits inputs per key: every key gets its own token bucket, lazily created on first use and stored in a bounded, idle-evicting cache.

The typical use is per-client fairness, keying on the client IP with ClientIpRateKey; any InputToRateKey extractor (including plain closures) works.

Modes mirror RatePolicy: KeyedRatePolicy::abort rejects over-budget inputs with RateLimitReached (a 429 path), KeyedRatePolicy::wait paces them. Inputs without a derivable key are allowed through by default. This is convenient for stacks where the key is genuinely optional, but is fail-open when the extractor depends on missing metadata; security limits should set KeyedRatePolicy::set_missing_key_allowed to false and abort them with MissingRateKey instead.

Memory is bounded: at most KeyedRatePolicy::set_max_keys buckets are kept, and buckets idle longer than KeyedRatePolicy::set_idle_timeout are evicted. The idle timeout is clamped to the time it takes to refill the configured burst from empty, so a bucket evicted for idleness and recreated full cannot regain budget any faster than one that stayed cached.

A new key is rejected with RateKeyCapacityReached while max_keys non-idle buckets are live. Live buckets are never evicted to admit another key, because recreating an exhausted bucket full would let callers bypass the rate limit by cycling keys at the memory bound.

Size max_keys for the number of simultaneously active keys after aggregation. The default IPv6 /64 aggregation means one routed /48 can still fill the default 65 536-key capacity; deployments serving larger IPv6 populations can aggregate more broadly with ClientIpRateKey::set_ipv6_prefix and/or raise this bound.

Implementations§

Source§

impl<X, K> KeyedRatePolicy<X, K>
where K: RateKey,

Source

pub fn wait(extractor: X, rate: Rate) -> Self

Create a new KeyedRatePolicy that paces inputs beyond the given per-key Rate. A known key waits rather than failing when its bucket is empty; a new key can still fail closed when the configured key capacity is exhausted.

Source

pub fn abort(extractor: X, rate: Rate) -> Self

Create a new KeyedRatePolicy that aborts inputs beyond the given per-key Rate with RateLimitReached.

Source

pub fn with_burst(self, burst: u64) -> Self

Override the per-key burst capacity (default: one period worth of units).

§Panics

Panics if burst is zero.

Source

pub fn set_burst(&mut self, burst: u64) -> &mut Self

Override the per-key burst capacity (default: one period worth of units).

§Panics

Panics if burst is zero.

Source

pub fn with_missing_key_allowed(self, allowed: bool) -> Self

Allow (default) or abort — with MissingRateKey — inputs for which no key can be derived.

Source

pub fn set_missing_key_allowed(&mut self, allowed: bool) -> &mut Self

Allow (default) or abort — with MissingRateKey — inputs for which no key can be derived.

Source

pub fn with_max_keys(self, max_keys: u64) -> Self

Bound the number of tracked keys (default: 65 536). When all tracked buckets are still active, a new key is rejected with RateKeyCapacityReached instead of evicting a live bucket and resetting its budget.

This is an availability bound as well as a memory bound. With the default IPv6 /64 keys, one /48 contains 65 536 distinct keys. Aggregate more broadly or raise this value when that is a realistic share of the expected active client population.

§Panics

Panics if max_keys is zero.

Source

pub fn set_max_keys(&mut self, max_keys: u64) -> &mut Self

Bound the number of tracked keys (default: 65 536). When all tracked buckets are still active, a new key is rejected with RateKeyCapacityReached instead of evicting a live bucket and resetting its budget.

This is an availability bound as well as a memory bound. With the default IPv6 /64 keys, one /48 contains 65 536 distinct keys. Aggregate more broadly or raise this value when that is a realistic share of the expected active client population.

§Panics

Panics if max_keys is zero.

Source

pub fn with_idle_timeout(self, idle_timeout: Duration) -> Self

Evict buckets idle for this long (default: 1 minute), clamped to at least the time required to refill the burst from empty.

Source

pub fn set_idle_timeout(&mut self, idle_timeout: Duration) -> &mut Self

Evict buckets idle for this long (default: 1 minute), clamped to at least the time required to refill the burst from empty.

Trait Implementations§

Source§

impl<X: Debug, K> Debug for KeyedRatePolicy<X, K>

Source§

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter. Read more
Source§

impl<X, K, Input> Policy<Input> for KeyedRatePolicy<X, K>
where X: InputToRateKey<Input, Key = K>, K: RateKey, Input: Send + 'static,

Source§

type Guard = ()

The guard type that is returned when the input is allowed to proceed. Read more
Source§

type Error = Box<dyn Error + Sync + Send>

The error type that is returned when the input is not allowed to proceed, and should be aborted. Read more
Source§

async fn check( &self, input: Input, ) -> PolicyResult<Input, Self::Guard, Self::Error>

Check whether the input is allowed to proceed. Read more

Auto Trait Implementations§

§

impl<X, K> !Freeze for KeyedRatePolicy<X, K>

§

impl<X, K> !RefUnwindSafe for KeyedRatePolicy<X, K>

§

impl<X, K> !UnwindSafe for KeyedRatePolicy<X, K>

§

impl<X, K> Send for KeyedRatePolicy<X, K>
where X: Send, K: Send,

§

impl<X, K> Sync for KeyedRatePolicy<X, K>
where X: Sync, K: Send,

§

impl<X, K> Unpin for KeyedRatePolicy<X, K>
where X: Unpin, K: Unpin,

§

impl<X, K> UnsafeUnpin for KeyedRatePolicy<X, K>
where X: UnsafeUnpin,

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
where ST: ?Sized, DT: ?Sized,

Source§

impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
where ST: ?Sized, DT: ?Sized,

Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T> FutureExt for T

Source§

fn with_context(self, otel_cx: Context) -> WithContext<Self>

Attaches the provided Context to this type, returning a WithContext wrapper. Read more
Source§

fn with_current_context(self) -> WithContext<Self>

Attaches the current Context to this type, returning a WithContext wrapper. Read more
Source§

impl<T> Instrument for T

Source§

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided Span, returning an Instrumented wrapper. Read more
Source§

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more
Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> IntoEither for T

Source§

fn into_either(self, into_left: bool) -> Either<Self, Self>

Converts self into a Left variant of Either<Self, Self> if into_left is true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more
Source§

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
where F: FnOnce(&Self) -> bool,

Converts self into a Left variant of Either<Self, Self> if into_left(&self) returns true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more
Source§

impl<T> Pointable for T

Source§

const ALIGN: usize

The alignment of pointer.
Source§

type Init = T

The type for initializers.
Source§

unsafe fn init(init: <T as Pointable>::Init) -> usize

Initializes a with the given initializer. Read more
Source§

unsafe fn deref<'a>(ptr: usize) -> &'a T

Dereferences the given pointer. Read more
Source§

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

Mutably dereferences the given pointer. Read more
Source§

unsafe fn drop(ptr: usize)

Drops the object pointed to by the given pointer. Read more
Source§

impl<T, U> RamaFrom<T> for U
where U: From<T>,

Source§

fn rama_from(value: T) -> U

Source§

impl<T, U, CrateMarker> RamaInto<U, CrateMarker> for T
where U: RamaFrom<T, CrateMarker>,

Source§

fn rama_into(self) -> U

Source§

impl<T, U> RamaTryFrom<T> for U
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

Source§

fn rama_try_from(value: T) -> Result<U, <U as RamaTryFrom<T>>::Error>

Source§

impl<T, U, CrateMarker> RamaTryInto<U, CrateMarker> for T
where U: RamaTryFrom<T, CrateMarker>,

Source§

type Error = <U as RamaTryFrom<T, CrateMarker>>::Error

Source§

fn rama_try_into(self) -> Result<U, <U as RamaTryFrom<T, CrateMarker>>::Error>

Source§

impl<T> Read<Exclusive, BecauseExclusive> for T
where T: ?Sized,

Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
Source§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

Source§

fn vzip(self) -> V

Source§

impl<V, F> ValueFormatter<&V> for F
where F: ValueFormatter<V> + ?Sized, V: ?Sized,

Source§

const SHAPE: FieldShape<'static>

Available on non-metrique_require_explicit_impls only.
The shape of values produced by this formatter. Read more
Source§

fn format_value(writer: impl ValueWriter, value: &&V)

Write value to writer
Source§

impl<V, F> ValueFormatter<Arc<V>> for F
where F: ValueFormatter<V> + ?Sized, V: ?Sized,

Source§

const SHAPE: FieldShape<'static>

Available on non-metrique_require_explicit_impls only.
The shape of values produced by this formatter. Read more
Source§

fn format_value(writer: impl ValueWriter, value: &Arc<V>)

Write value to writer
Source§

impl<V, F> ValueFormatter<Box<V>> for F
where F: ValueFormatter<V> + ?Sized, V: ?Sized,

Source§

const SHAPE: FieldShape<'static>

Available on non-metrique_require_explicit_impls only.
The shape of values produced by this formatter. Read more
Source§

fn format_value(writer: impl ValueWriter, value: &Box<V>)

Write value to writer
Source§

impl<V, F> ValueFormatter<Cow<'_, V>> for F
where V: ToOwned + ?Sized, F: ValueFormatter<V> + ?Sized,

Source§

const SHAPE: FieldShape<'static>

Available on non-metrique_require_explicit_impls only.
The shape of values produced by this formatter. Read more
Source§

fn format_value(writer: impl ValueWriter, value: &Cow<'_, V>)

Write value to writer
Source§

impl<V, F> ValueFormatter<Option<V>> for F
where F: ValueFormatter<V> + ?Sized,

Source§

const SHAPE: FieldShape<'static>

Available on non-metrique_require_explicit_impls only.
The shape of values produced by this formatter. Read more
Source§

fn format_value(writer: impl ValueWriter, value: &Option<V>)

Write value to writer
Source§

impl<T> WithSubscriber for T

Source§

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a WithDispatch wrapper. Read more
Source§

fn with_current_subscriber(self) -> WithDispatch<Self>

Attaches the current default Subscriber to this type, returning a WithDispatch wrapper. Read more