forked/hyper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(client): clarify some code in send_request

Browse Source
This commit is contained in:
Sean McArthur
2018-09-27 20:14:02 -07:00
parent af23dda2e9
commit bddc2d5040
1 changed files with 77 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

77
src/client/mod.rs
View File

@@ -249,24 +249,23 @@ where C: Connect + Sync + 'static,
let client = self.clone(); let client = self.clone();
let uri = req.uri().clone(); let uri = req.uri().clone();
let pool_key = (Arc::new(domain.to_string()), self.ver);
let fut = RetryableSendRequest { let fut = RetryableSendRequest {
client: client, client,
future: self.send_request(req, &domain), future: self.send_request(req, pool_key.clone()),
domain: domain, pool_key,
uri: uri, uri,
}; };
ResponseFuture::new(Box::new(fut)) ResponseFuture::new(Box::new(fut))
} }
//TODO: replace with `impl Future` when stable //TODO: replace with `impl Future` when stable
fn send_request(&self, mut req: Request<B>, domain: &str) -> Box<Future<Item=Response<Body>, Error=ClientError<B>> + Send> { fn send_request(&self, mut req: Request<B>, pool_key: PoolKey) -> Box<Future<Item=Response<Body>, Error=ClientError<B>> + Send> {
let pool_key = (Arc::new(domain.to_string()), self.ver);
let race = self.pool_checkout_or_connect(req.uri().clone(), pool_key); let race = self.pool_checkout_or_connect(req.uri().clone(), pool_key);
let ver = self.ver; let ver = self.ver;
let executor = self.executor.clone(); let executor = self.executor.clone();
let resp = race.and_then(move |mut pooled| { Box::new(race.and_then(move |mut pooled| {
let conn_reused = pooled.is_reused();
if ver == Ver::Http1 { if ver == Ver::Http1 {
// CONNECT always sends origin-form, so check it first... // CONNECT always sends origin-form, so check it first...
if req.method() == &Method::CONNECT { if req.method() == &Method::CONNECT {
@@ -283,7 +282,8 @@ where C: Connect + Sync + 'static,
); );
} }
let fut = pooled.send_request_retryable(req); let fut = pooled.send_request_retryable(req)
.map_err(ClientError::map_with_reused(pooled.is_reused()));
// As of futures@0.1.21, there is a race condition in the mpsc // As of futures@0.1.21, there is a race condition in the mpsc
// channel, such that sending when the receiver is closing can // channel, such that sending when the receiver is closing can
@@ -293,33 +293,10 @@ where C: Connect + Sync + 'static,
// To counteract this, we must check if our senders 'want' channel // To counteract this, we must check if our senders 'want' channel
// has been closed after having tried to send. If so, error out... // has been closed after having tried to send. If so, error out...
if pooled.is_closed() { if pooled.is_closed() {
drop(pooled); return Either::A(fut);
let fut = fut
.map_err(move |(err, orig_req)| {
if let Some(req) = orig_req {
ClientError::Canceled {
connection_reused: conn_reused,
reason: err,
req,
} }
} else {
ClientError::Normal(err) Either::B(fut
}
});
Either::A(fut)
} else {
let fut = fut
.map_err(move |(err, orig_req)| {
if let Some(req) = orig_req {
ClientError::Canceled {
connection_reused: conn_reused,
reason: err,
req,
}
} else {
ClientError::Normal(err)
}
})
.and_then(move |mut res| { .and_then(move |mut res| {
// If pooled is HTTP/2, we can toss this reference immediately. // If pooled is HTTP/2, we can toss this reference immediately.
// //
@@ -364,12 +341,8 @@ where C: Connect + Sync + 'static,
} }
} }
Ok(res) Ok(res)
}); }))
Either::B(fut) }))
}
});
Box::new(resp)
} }
fn pool_checkout_or_connect(&self, uri: Uri, pool_key: PoolKey) fn pool_checkout_or_connect(&self, uri: Uri, pool_key: PoolKey)
@@ -558,8 +531,8 @@ impl Future for ResponseFuture {
struct RetryableSendRequest<C, B> { struct RetryableSendRequest<C, B> {
client: Client<C, B>, client: Client<C, B>,
domain: String,
future: Box<Future<Item=Response<Body>, Error=ClientError<B>> + Send>, future: Box<Future<Item=Response<Body>, Error=ClientError<B>> + Send>,
pool_key: PoolKey,
uri: Uri, uri: Uri,
} }
@@ -598,7 +571,7 @@ where
trace!("unstarted request canceled, trying again (reason={:?})", reason); trace!("unstarted request canceled, trying again (reason={:?})", reason);
*req.uri_mut() = self.uri.clone(); *req.uri_mut() = self.uri.clone();
self.future = self.client.send_request(req, &self.domain); self.future = self.client.send_request(req, self.pool_key.clone());
} }
} }
} }
@@ -697,6 +670,24 @@ enum ClientError<B> {
} }
} }
impl<B> ClientError<B> {
fn map_with_reused(conn_reused: bool)
-> impl Fn((::Error, Option<Request<B>>)) -> Self
{
move |(err, orig_req)| {
if let Some(req) = orig_req {
ClientError::Canceled {
connection_reused: conn_reused,
reason: err,
req,
}
} else {
ClientError::Normal(err)
}
}
}
}
/// A marker to identify what version a pooled connection is. /// A marker to identify what version a pooled connection is.
#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)] #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
enum Ver { enum Ver {