tower-limit 0.3.1

Limit maximum request rate to a `Service`.
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

use std::time::Duration;
use tokio::time;
use tokio_test::{assert_pending, assert_ready, assert_ready_ok};
use tower_limit::rate::RateLimitLayer;
use tower_test::{assert_request_eq, mock};

#[tokio::test]
async fn reaching_capacity() {
    time::pause();

    let rate_limit = RateLimitLayer::new(1, Duration::from_millis(100));

    let (mut service, mut handle) = mock::spawn_layer(rate_limit);

    assert_ready_ok!(service.poll_ready());

    let response = service.call("hello");

    assert_request_eq!(handle, "hello").send_response("world");

    assert_eq!(response.await.unwrap(), "world");
    assert_pending!(service.poll_ready());

    assert_pending!(handle.poll_request());

    time::advance(Duration::from_millis(101)).await;

    assert_ready_ok!(service.poll_ready());

    let response = service.call("two");

    assert_request_eq!(handle, "two").send_response("done");

    assert_eq!(response.await.unwrap(), "done");
}