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
use crate::metadata::MetadataMap;
#[derive(Debug)]
pub struct Response<T> {
metadata: MetadataMap,
message: T,
}
impl<T> Response<T> {
pub fn new(message: T) -> Self {
Response {
metadata: MetadataMap::new(),
message,
}
}
pub fn get_ref(&self) -> &T {
&self.message
}
pub fn get_mut(&mut self) -> &mut T {
&mut self.message
}
pub fn metadata(&self) -> &MetadataMap {
&self.metadata
}
pub fn metadata_mut(&mut self) -> &mut MetadataMap {
&mut self.metadata
}
pub fn into_inner(self) -> T {
self.message
}
pub(crate) fn from_http(res: http::Response<T>) -> Self {
let (head, message) = res.into_parts();
Response {
metadata: MetadataMap::from_headers(head.headers),
message,
}
}
pub fn into_http(self) -> http::Response<T> {
let mut res = http::Response::new(self.message);
*res.version_mut() = http::Version::HTTP_2;
*res.headers_mut() = self.metadata.into_headers();
res
}
pub fn map<F, U>(self, f: F) -> Response<U>
where
F: FnOnce(T) -> U,
{
let message = f(self.message);
Response {
metadata: self.metadata,
message,
}
}
}