trypema 1.1.0

High-performance rate limiting primitives in Rust, designed for concurrency safety, low overhead, and predictable latency.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

//! Async runtime abstraction for criterion benchmarks.
//!
//! Include in a bench file with:
//! ```ignore
//! #[path = "runtime.rs"]
//! mod runtime;
//! ```
//!
//! Mirrors the structure of `src/tests/runtime.rs` and `stress/src/runtime.rs`,
//! adapted for the bench context where:
//! - The runtime handle is built once and reused across all iterations.
//! - `block_on` takes the handle by reference (not owned).
//! - `spawn` / `join` / `async_sleep` are omitted — benchmarks do not need them.

/// The runtime handle used to drive async bench futures.
///
/// - **tokio**: `tokio::runtime::Runtime` (multi-thread, 2 workers)
/// - **smol**: `()` — `smol::block_on` requires no persistent handle
#[cfg(feature = "redis-tokio")]
pub type Runtime = tokio::runtime::Runtime;

#[cfg(all(feature = "redis-smol", not(feature = "redis-tokio")))]
pub type Runtime = ();

/// Build the runtime handle once per bench binary.
#[cfg(feature = "redis-tokio")]
pub fn build() -> Runtime {
    tokio::runtime::Builder::new_multi_thread()
        .enable_all()
        .worker_threads(2)
        .build()
        .unwrap()
}

#[cfg(all(feature = "redis-smol", not(feature = "redis-tokio")))]
pub fn build() -> Runtime {}

/// Drive a future to completion on the given runtime handle.
#[cfg(feature = "redis-tokio")]
pub fn block_on<F, T>(rt: &Runtime, f: F) -> T
where
    F: std::future::Future<Output = T>,
{
    rt.block_on(f)
}

#[cfg(all(feature = "redis-smol", not(feature = "redis-tokio")))]
pub fn block_on<F, T>(_rt: &Runtime, f: F) -> T
where
    F: std::future::Future<Output = T>,
{
    smol::block_on(f)
}