tideway 0.1.0

A batteries-included Rust web framework built on Axum for building SaaS applications quickly
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

use super::config::TimeoutConfig;
use tower_http::timeout::TimeoutLayer;

/// Build a tower-http TimeoutLayer from a TimeoutConfig
pub fn build_timeout_layer(config: &TimeoutConfig) -> Option<TimeoutLayer> {
    if !config.enabled {
        return None;
    }

    Some(TimeoutLayer::new(config.duration()))
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_disabled_timeout() {
        let config = TimeoutConfig {
            enabled: false,
            ..Default::default()
        };
        let layer = build_timeout_layer(&config);
        assert!(layer.is_none());
    }

    #[test]
    fn test_enabled_timeout() {
        let config = TimeoutConfig {
            enabled: true,
            timeout_seconds: 60,
        };
        let layer = build_timeout_layer(&config);
        assert!(layer.is_some());
    }
}