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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
use https::{header::{RANGE, ACCEPT_RANGES, CONTENT_RANGE, CONTENT_LENGTH}, StatusCode};
use futures::future::{FutureExt, LocalBoxFuture};
use tcp::Socket;
use crate::{gateway::GatewayContext,
middleware::{MiddlewareResult, Middleware},
request::HttpRequest,
response::HttpResponse};
///
/// Http静态资源范围加载器
///
pub struct RangeLoad;
unsafe impl Send for RangeLoad {}
unsafe impl Sync for RangeLoad {}
impl<S: Socket> Middleware<S, GatewayContext> for RangeLoad {
fn request<'a>(&'a self,
_context: &'a mut GatewayContext,
req: HttpRequest<S>)
-> LocalBoxFuture<'a, MiddlewareResult<S>> {
let future = async move {
//继续请求处理
MiddlewareResult::ContinueRequest(req)
};
future.boxed_local()
}
fn response<'a>(&'a self,
_context: &'a mut GatewayContext,
req: HttpRequest<S>,
resp: HttpResponse)
-> LocalBoxFuture<'a, MiddlewareResult<S>> {
let mut response = resp;
let future = async move {
if let Some(range_value) = req.headers().get(RANGE) {
if let Ok(r) = range_value.to_str() {
if let None = r.find(',') {
//只支持单个指定范围
let str = r.trim().to_lowercase();
let tmp: Vec<&str> = str.split("=").collect();
let range_str = tmp[1].trim();
let mut vec: Vec<&str> = range_str.split("-").collect();
if let Ok(start) = vec[0].parse::<usize>() {
//获取开始范围成功，则获取响应体大小
let mut body_len = 0;
if let Some(body) = response.as_body() {
if let Some(len) = body.len() {
body_len = len;
}
}
let mut is_end = false; //默认未指定结束范围
let mut end = body_len - 1; //未指定结束范围，则默认为所有数据
if let Ok(e) = vec[1].parse::<usize>() {
//已指定结束范围
is_end = true;
end = e;
}
if start > end || end >= body_len {
//客户端需要的静态资源范围越界，则立即返回错误
if let Some(body) = response.as_mut_body() {
body.reset(&[]);
}
response
.status(StatusCode::RANGE_NOT_SATISFIABLE.as_u16())
.header(ACCEPT_RANGES.as_str(), "bytes")
.header(CONTENT_LENGTH.as_str(), "0");
return MiddlewareResult::Break(response);
}
if let Some(body) = response.as_mut_body() {
let mut buf = Vec::with_capacity(0);
if let Some(bin) = body.as_slice() {
buf = Vec::from(&bin[start..(end + 1)]);
}
body.reset(buf.as_slice()); //重置响应体为指定范围的数据
}
//设置范围响应状态码和响应头
if !is_end {
//未指定结束范围
response
.status(StatusCode::PARTIAL_CONTENT.as_u16())
.header(ACCEPT_RANGES.as_str(), "bytes")
.header(CONTENT_RANGE.as_str(), ["bytes", " ", range_str, end.to_string().as_str(), "/", body_len.to_string().as_str()].concat().as_str());
} else {
//已指定结束范围
response
.status(StatusCode::PARTIAL_CONTENT.as_u16())
.header(ACCEPT_RANGES.as_str(), "bytes")
.header(CONTENT_RANGE.as_str(), ["bytes", " ", range_str, "/", body_len.to_string().as_str()].concat().as_str());
}
}
}
}
}
//继续响应处理
MiddlewareResult::ContinueResponse((req, response))
};
future.boxed_local()
}
}
impl RangeLoad {
/// 构建指定根目录的文件上传处理器
pub fn new() -> Self {
RangeLoad
}
}