graphul 1.0.1

Optimize, speed, scale your microservices and save money 💵
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

use crate::Body;

pub use axum::middleware::{self, from_fn, from_fn_with_state};

pub type Next = middleware::Next<Body>;
pub use tower_http as tower;

pub mod limit {
    use std::time::Duration;

    pub use ::tower::limit::{ConcurrencyLimit, RateLimit};
    use ::tower::limit::{
        ConcurrencyLimitLayer as OriginalConcurrencyLimitLayer,
        GlobalConcurrencyLimitLayer as OriginalGlobalConcurrencyLimitLayer,
        RateLimitLayer as OriginalRateLimitLayer,
    };
    use tower_http::add_extension::{AddExtension, AddExtensionLayer};

    /// Wrapper of RateLimitLayer of tower
    /// Enforces a rate limit on the number of requests the underlying
    /// service can handle over a period of time.
    #[derive(Debug, Clone)]
    pub struct RateLimitLayer(AddExtensionLayer<OriginalRateLimitLayer>);
    /// Wrapper of ConcurrencyLimitLayer of tower
    /// Enforces a limit on the concurrent number of requests the underlying
    /// service can handle.
    #[derive(Debug, Clone)]
    pub struct ConcurrencyLimitLayer(AddExtensionLayer<OriginalConcurrencyLimitLayer>);
    /// Wrapper of GlobalConcurrencyLimitLayer of tower
    /// Enforces a limit on the concurrent number of requests the underlying
    /// service can handle.
    ///
    /// Unlike [`ConcurrencyLimitLayer`], which enforces a per-service concurrency
    /// limit, this layer accepts a owned semaphore (`Arc<Semaphore>`) which can be
    /// shared across multiple services.
    ///
    /// Cloning this layer will not create a new semaphore.
    #[derive(Debug, Clone)]
    pub struct GlobalConcurrencyLimitLayer(AddExtensionLayer<OriginalGlobalConcurrencyLimitLayer>);

    impl RateLimitLayer {
        /// Create new rate limit layer.
        pub fn new(limit: u64, per: Duration) -> Self {
            Self(AddExtensionLayer::new(OriginalRateLimitLayer::new(
                limit, per,
            )))
        }
    }

    impl<S> tower_layer::Layer<S> for RateLimitLayer {
        type Service = AddExtension<S, OriginalRateLimitLayer>;
        fn layer(&self, service: S) -> Self::Service {
            self.0.layer(service)
        }
    }

    impl ConcurrencyLimitLayer {
        /// Create a new concurrency limit layer.
        pub fn new(max: usize) -> Self {
            Self(AddExtensionLayer::new(OriginalConcurrencyLimitLayer::new(
                max,
            )))
        }
    }

    impl<S> tower_layer::Layer<S> for ConcurrencyLimitLayer {
        type Service = AddExtension<S, OriginalConcurrencyLimitLayer>;
        fn layer(&self, service: S) -> Self::Service {
            self.0.layer(service)
        }
    }

    impl GlobalConcurrencyLimitLayer {
        /// Create a new `GlobalConcurrencyLimitLayer`.
        pub fn new(max: usize) -> Self {
            Self(AddExtensionLayer::new(
                OriginalGlobalConcurrencyLimitLayer::new(max),
            ))
        }
    }

    impl<S> tower_layer::Layer<S> for GlobalConcurrencyLimitLayer {
        type Service = AddExtension<S, OriginalGlobalConcurrencyLimitLayer>;
        fn layer(&self, service: S) -> Self::Service {
            self.0.layer(service)
        }
    }
}