reqwest-ratelimit 0.5.0

Rate-limit middleware for reqwest.
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

//! Middleware to rate-limit requests built on [`reqwest_middleware`].
//!
//! You're expected to provide your own [`RateLimiter`] implementation.
//!
//! ## Example
//!
//! ```
#![doc = include_str!("example.rs")]
//! ```
use std::future::Future;

use async_trait::async_trait;
use http::Extensions;
use reqwest::{Request, Response};
use reqwest_middleware::{Next, Result};

/// Request rate limiter.
pub trait RateLimiter: Send + Sync + 'static {
    /// Acquires a permit to issue the next request.
    fn acquire_permit(&self) -> impl Future<Output = ()> + Send + '_;
}

/// Creates a new [`Middleware`] rate-limiting all requests using the provided [`RateLimiter`].
pub fn all<R>(rate_limiter: R) -> Middleware<R> {
    Middleware { rate_limiter }
}

/// Request rate-limiting middleware.
pub struct Middleware<R> {
    rate_limiter: R,
}

#[async_trait]
impl<R: RateLimiter> reqwest_middleware::Middleware for Middleware<R> {
    async fn handle(
        &self,
        req: Request,
        extensions: &'_ mut Extensions,
        next: Next<'_>,
    ) -> Result<Response> {
        self.rate_limiter.acquire_permit().await;
        next.run(req, extensions).await
    }
}