Skip to main content

KeyedRatePolicy

Struct KeyedRatePolicy 

pub struct KeyedRatePolicy<X, K> { /* private fields */ }
Available on crate feature net 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§

§

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

pub fn wait(extractor: X, rate: Rate) -> KeyedRatePolicy<X, K>

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.

pub fn abort(extractor: X, rate: Rate) -> KeyedRatePolicy<X, K>

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

pub fn with_burst(self, burst: u64) -> KeyedRatePolicy<X, K>

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

§Panics

Panics if burst is zero.

pub fn set_burst(&mut self, burst: u64) -> &mut KeyedRatePolicy<X, K>

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

§Panics

Panics if burst is zero.

pub fn with_missing_key_allowed(self, allowed: bool) -> KeyedRatePolicy<X, K>

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

pub fn set_missing_key_allowed( &mut self, allowed: bool, ) -> &mut KeyedRatePolicy<X, K>

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

pub fn with_max_keys(self, max_keys: u64) -> KeyedRatePolicy<X, K>

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.

pub fn set_max_keys(&mut self, max_keys: u64) -> &mut KeyedRatePolicy<X, K>

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.

pub fn with_idle_timeout(self, idle_timeout: Duration) -> KeyedRatePolicy<X, K>

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

pub fn set_idle_timeout( &mut self, idle_timeout: Duration, ) -> &mut KeyedRatePolicy<X, K>

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

Trait Implementations§

§

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

§

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

Formats the value using the given formatter. Read more
§

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

§

type Guard = ()

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

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

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

async fn check( &self, input: Input, ) -> PolicyResult<Input, <KeyedRatePolicy<X, K> as Policy<Input>>::Guard, <KeyedRatePolicy<X, K> as Policy<Input>>::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, BucketCache<K>: Send,

§

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

§

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

§

impl<X, K> UnsafeUnpin for KeyedRatePolicy<X, K>
where X: UnsafeUnpin, BucketCache<K>: 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
§

impl<'a, T, E> AsTaggedExplicit<'a, E> for T
where T: 'a,

§

fn explicit(self, class: Class, tag: u32) -> TaggedParser<'a, Explicit, Self, E>

§

impl<'a, T, E> AsTaggedImplicit<'a, E> for T
where T: 'a,

§

fn implicit( self, class: Class, constructed: bool, tag: u32, ) -> TaggedParser<'a, Implicit, Self, E>

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
§

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

§

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.

§

impl<T> FutureExt for T

§

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

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

fn with_current_context(self) -> WithContext<Self>

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

impl<T> Instrument for T

§

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

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

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
§

impl<T> IntoRequest<T> for T

§

fn into_request(self) -> Request<T>

Wrap the input message T in a rama_grpc::Request
§

impl<L> LayerExt<L> for L

§

fn named_layer<S>(&self, service: S) -> Layered<<L as Layer<S>>::Service, S>
where L: Layer<S>,

Applies the layer to a service and wraps it in Layered.
§

impl<T> Pointable for T

§

const ALIGN: usize

The alignment of pointer.
§

type Init = T

The type for initializers.
§

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

Initializes a with the given initializer. Read more
§

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

Dereferences the given pointer. Read more
§

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

Mutably dereferences the given pointer. Read more
§

unsafe fn drop(ptr: usize)

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

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

§

fn and<P, B, E>(self, other: P) -> And<T, P>
where T: Sized + Policy<B, E>, P: Policy<B, E>,

Create a new Policy that returns Action::Follow only if self and other return Action::Follow. Read more
§

fn or<P, B, E>(self, other: P) -> Or<T, P>
where T: Sized + Policy<B, E>, P: Policy<B, E>,

Create a new Policy that returns Action::Follow if either self or other returns Action::Follow. Read more
§

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

§

fn rama_from(value: T) -> U

§

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

§

fn rama_into(self) -> U

§

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

§

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

§

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

§

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

§

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

§

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

§

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

Source§

impl<T> Same for T

Source§

type Output = T

Should always be Self
Source§

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

Source§

type Error = !

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.
§

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

§

fn vzip(self) -> V

§

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

§

const SHAPE: FieldShape<'static>

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

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

Write value to writer
§

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

§

const SHAPE: FieldShape<'static>

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

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

Write value to writer
§

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

§

const SHAPE: FieldShape<'static>

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

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

Write value to writer
§

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

§

const SHAPE: FieldShape<'static>

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

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

Write value to writer
§

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

§

const SHAPE: FieldShape<'static>

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

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

Write value to writer
§

impl<T> WithSubscriber for T

§

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
§

fn with_current_subscriber(self) -> WithDispatch<Self>

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