trypema 1.1.0

High-performance rate limiting primitives in Rust, designed for concurrency safety, low overhead, and predictable latency.
Documentation
use std::{future::Future, time::Duration};

#[cfg(feature = "redis-tokio")]
pub(super) type JoinHandle<T> = tokio::task::JoinHandle<T>;

#[cfg(all(feature = "redis-smol", not(feature = "redis-tokio")))]
pub(super) type JoinHandle<T> = smol::Task<T>;

#[cfg(feature = "redis-tokio")]
pub(super) fn block_on<F, T>(f: F) -> T
where
    F: Future<Output = T>,
{
    tokio::runtime::Runtime::new().unwrap().block_on(f)
}

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

#[cfg(feature = "redis-tokio")]
pub(super) fn spawn<F, T>(f: F) -> JoinHandle<T>
where
    F: Future<Output = T> + Send + 'static,
    T: Send + 'static,
{
    tokio::spawn(f)
}

#[cfg(all(feature = "redis-smol", not(feature = "redis-tokio")))]
pub(super) fn spawn<F, T>(f: F) -> JoinHandle<T>
where
    F: Future<Output = T> + Send + 'static,
    T: Send + 'static,
{
    smol::spawn(f)
}

#[cfg(feature = "redis-tokio")]
pub(super) async fn join<T>(h: JoinHandle<T>) -> T
where
    T: Send + 'static,
{
    h.await.unwrap()
}

#[cfg(all(feature = "redis-smol", not(feature = "redis-tokio")))]
pub(super) async fn join<T>(h: JoinHandle<T>) -> T
where
    T: Send + 'static,
{
    h.await
}

#[cfg(feature = "redis-tokio")]
pub(super) async fn async_sleep(d: Duration) {
    tokio::time::sleep(d).await;
}

#[cfg(all(feature = "redis-smol", not(feature = "redis-tokio")))]
pub(super) async fn async_sleep(d: Duration) {
    smol::Timer::after(d).await;
}