leyline-http 0.1.1

HTTP client that mimics browsers on the wire
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
89
90
91
92
93

use std::time::Duration;

const DEFAULT_TOTAL: Option<Duration> = Some(Duration::from_secs(300));
const DEFAULT_CONNECT: Option<Duration> = Some(Duration::from_secs(10));
const DEFAULT_ERROR_BODY: Option<Duration> = Some(Duration::from_secs(10));

#[derive(Debug, Clone, Copy, Default, PartialEq, Eq)]
#[non_exhaustive]
pub struct TimeoutConfig {
    total: Option<Option<Duration>>,
    connect: Option<Option<Duration>>,
    read: Option<Option<Duration>>,
    response_header: Option<Option<Duration>>,
    body: Option<Option<Duration>>,
    error_body: Option<Option<Duration>>,
}

impl TimeoutConfig {
    pub fn new() -> Self {
        Self::default()
    }

    pub fn total(mut self, d: impl Into<Option<Duration>>) -> Self {
        self.total = Some(d.into());
        self
    }

    pub fn connect(mut self, d: impl Into<Option<Duration>>) -> Self {
        self.connect = Some(d.into());
        self
    }

    pub fn read(mut self, d: impl Into<Option<Duration>>) -> Self {
        self.read = Some(d.into());
        self
    }

    pub fn response_header(mut self, d: impl Into<Option<Duration>>) -> Self {
        self.response_header = Some(d.into());
        self
    }

    pub fn body(mut self, d: impl Into<Option<Duration>>) -> Self {
        self.body = Some(d.into());
        self
    }

    pub fn error_body(mut self, d: impl Into<Option<Duration>>) -> Self {
        self.error_body = Some(d.into());
        self
    }

    pub(crate) fn over(self, base: &TimeoutConfig) -> TimeoutConfig {
        TimeoutConfig {
            total: self.total.or(base.total),
            connect: self.connect.or(base.connect),
            read: self.read.or(base.read),
            response_header: self.response_header.or(base.response_header),
            body: self.body.or(base.body),
            error_body: self.error_body.or(base.error_body),
        }
    }

    pub(crate) fn body_limit(&self) -> Option<Duration> {
        self.body.flatten()
    }

    pub(crate) fn error_body_limit(&self) -> Option<Duration> {
        self.error_body.unwrap_or(DEFAULT_ERROR_BODY)
    }

    pub(crate) fn total_limit(&self) -> Option<Duration> {
        self.total.unwrap_or(DEFAULT_TOTAL)
    }

    pub(crate) fn connect_limit(&self) -> Option<Duration> {
        self.connect.unwrap_or(DEFAULT_CONNECT)
    }

    pub(crate) fn read_limit(&self) -> Option<Duration> {
        self.read.flatten()
    }

    pub(crate) fn response_header_limit(&self) -> Option<Duration> {
        self.response_header.flatten()
    }
}

impl From<Duration> for TimeoutConfig {
    fn from(total: Duration) -> Self {
        Self::new().total(total)
    }
}