actix/actix-extras
1
0
Fork 0
mirror of https://github.com/actix/actix-extras.git synced 2024-11-28 09:42:40 +01:00
Code Issues Releases Wiki Activity

calculate client request timeout

Browse Source
This commit is contained in:
Nikolay Kim 2018-07-17 02:30:21 +06:00
parent bccd7c7671
commit 1af5aa3a3e
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/client/pipeline.rs
View File

@ -71,7 +71,7 @@ pub struct SendRequest {
conn: Option<Addr<ClientConnector>>, conn: Option<Addr<ClientConnector>>,
conn_timeout: Duration, conn_timeout: Duration,
wait_timeout: Duration, wait_timeout: Duration,
timeout: Option<Delay>, timeout: Option<Duration>,
} }
impl SendRequest { impl SendRequest {
@ -115,7 +115,7 @@ impl SendRequest {
/// Request timeout is the total time before a response must be received. /// Request timeout is the total time before a response must be received.
/// Default value is 5 seconds. /// Default value is 5 seconds.
pub fn timeout(mut self, timeout: Duration) -> Self { pub fn timeout(mut self, timeout: Duration) -> Self {
self.timeout = Some(Delay::new(Instant::now() + timeout)); self.timeout = Some(timeout);
self self
} }
@ -185,9 +185,10 @@ impl Future for SendRequest {
_ => IoBody::Done, _ => IoBody::Done,
}; };
let timeout = self.timeout.take().unwrap_or_else(|| { let timeout = self
Delay::new(Instant::now() + Duration::from_secs(5)) .timeout
}); .take()
.unwrap_or_else(|| Duration::from_secs(5));
let pl = Box::new(Pipeline { let pl = Box::new(Pipeline {
body, body,
@ -201,7 +202,7 @@ impl Future for SendRequest {
decompress: None, decompress: None,
should_decompress: self.req.response_decompress(), should_decompress: self.req.response_decompress(),
write_state: RunningState::Running, write_state: RunningState::Running,
timeout: Some(timeout), timeout: Some(Delay::new(Instant::now() + timeout)),
close: self.req.method() == &Method::HEAD, close: self.req.method() == &Method::HEAD,
}); });
self.state = State::Send(pl); self.state = State::Send(pl);