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
use futures::future::BoxFuture;
use http::{
header::{HeaderValue, IntoHeaderName},
HeaderMap, HttpTryFrom,
};
use crate::{
middleware::{Middleware, Next},
Context, Response,
};
#[derive(Clone, Default)]
pub struct DefaultHeaders {
headers: HeaderMap,
}
impl DefaultHeaders {
pub fn new() -> DefaultHeaders {
DefaultHeaders::default()
}
#[inline]
pub fn header<K, V>(mut self, key: K, value: V) -> Self
where
K: IntoHeaderName,
HeaderValue: HttpTryFrom<V>,
{
let value = HeaderValue::try_from(value)
.map_err(Into::into)
.expect("Cannot create default header");
self.headers.append(key, value);
self
}
}
impl<Data: Send + Sync + 'static> Middleware<Data> for DefaultHeaders {
fn handle<'a>(&'a self, cx: Context<Data>, next: Next<'a, Data>) -> BoxFuture<'a, Response> {
box_async! {
let mut res = await!(next.run(cx));
let headers = res.headers_mut();
for (key, value) in self.headers.iter() {
headers.entry(key).unwrap().or_insert_with(|| value.clone());
}
res
}
}
}