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

perf(client): change HttpConnecting to hold Arc<Config> instead of inlined fields

Browse Source
This commit is contained in:
Sean McArthur
2019-10-23 15:48:47 -07:00
parent f71304b449
commit 8b878a805a
3 changed files with 71 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

82
src/client/connect/http.rs
View File

@@ -228,6 +228,16 @@ impl<R> HttpConnector<R> {
}
}
impl<R: Resolve> HttpConnector<R> {
fn invalid_url(&self, err: InvalidUrl) -> HttpConnecting<R> {
HttpConnecting {
config: self.config.clone(),
state: State::Error(Some(io::Error::new(io::ErrorKind::InvalidInput, err))),
port: 0,
}
}
}
// R: Debug required for now to allow adding it to debug output later...
impl<R: fmt::Debug> fmt::Debug for HttpConnector<R> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
@@ -263,15 +273,15 @@ where
if self.config.enforce_http {
if dst.uri.scheme_part() != Some(&Scheme::HTTP) {
return invalid_url(InvalidUrl::NotHttp);
return self.invalid_url(InvalidUrl::NotHttp);
}
} else if dst.uri.scheme_part().is_none() {
return invalid_url(InvalidUrl::MissingScheme);
return self.invalid_url(InvalidUrl::MissingScheme);
}
let host = match dst.uri.host() {
Some(s) => s,
None => return invalid_url(InvalidUrl::MissingAuthority),
None => return self.invalid_url(InvalidUrl::MissingAuthority),
};
let port = match dst.uri.port_part() {
Some(port) => port.as_u16(),
@@ -279,16 +289,9 @@ where
};
HttpConnecting {
state: State::Lazy(self.resolver.clone(), host.into(), self.config.local_address),
handle: self.config.handle.clone(),
connect_timeout: self.config.connect_timeout,
happy_eyeballs_timeout: self.config.happy_eyeballs_timeout,
keep_alive_timeout: self.config.keep_alive_timeout,
nodelay: self.config.nodelay,
config: self.config.clone(),
state: State::Lazy(self.resolver.clone(), host.into()),
port,
reuse_address: self.config.reuse_address,
send_buffer_size: self.config.send_buffer_size,
recv_buffer_size: self.config.recv_buffer_size,
}
}
}
@@ -321,21 +324,6 @@ impl HttpInfo {
}
}
fn invalid_url<R: Resolve>(err: InvalidUrl) -> HttpConnecting<R> {
HttpConnecting {
state: State::Error(Some(io::Error::new(io::ErrorKind::InvalidInput, err))),
handle: None,
keep_alive_timeout: None,
nodelay: false,
port: 0,
connect_timeout: None,
happy_eyeballs_timeout: None,
reuse_address: false,
send_buffer_size: None,
recv_buffer_size: None,
}
}
#[derive(Debug, Clone, Copy)]
enum InvalidUrl {
MissingScheme,
@@ -362,23 +350,16 @@ impl StdError for InvalidUrl {
#[must_use = "futures do nothing unless polled"]
#[pin_project]
pub struct HttpConnecting<R: Resolve = GaiResolver> {
config: Arc<Config>,
#[pin]
state: State<R>,
handle: Option<Handle>,
connect_timeout: Option<Duration>,
happy_eyeballs_timeout: Option<Duration>,
keep_alive_timeout: Option<Duration>,
nodelay: bool,
port: u16,
reuse_address: bool,
send_buffer_size: Option<usize>,
recv_buffer_size: Option<usize>,
}
#[pin_project]
enum State<R: Resolve> {
Lazy(R, String, Option<IpAddr>),
Resolving(#[pin] R::Future, Option<IpAddr>),
Lazy(R, String),
Resolving(#[pin] R::Future),
Connecting(ConnectingTcp),
Error(Option<io::Error>),
}
@@ -389,22 +370,23 @@ impl<R: Resolve> Future for HttpConnecting<R> {
#[project]
fn poll(self: Pin<&mut Self>, cx: &mut task::Context<'_>) -> Poll<Self::Output> {
let mut me = self.project();
let config: &Config = &me.config;
loop {
let state;
#[project]
match me.state.as_mut().project() {
State::Lazy(ref resolver, ref mut host, ref local_addr) => {
State::Lazy(ref resolver, ref mut host) => {
// If the host is already an IP addr (v4 or v6),
// skip resolving the dns and start connecting right away.
if let Some(addrs) = dns::IpAddrs::try_parse(host, *me.port) {
state = State::Connecting(ConnectingTcp::new(
**local_addr, addrs, *me.connect_timeout, *me.happy_eyeballs_timeout, *me.reuse_address));
config.local_address, addrs, config.connect_timeout, config.happy_eyeballs_timeout, config.reuse_address));
} else {
let name = dns::Name::new(mem::replace(host, String::new()));
state = State::Resolving(resolver.resolve(name), **local_addr);
state = State::Resolving(resolver.resolve(name));
}
},
State::Resolving(future, local_addr) => {
State::Resolving(future) => {
let addrs = ready!(future.poll(cx))?;
let port = *me.port;
let addrs = addrs
@@ -412,24 +394,24 @@ impl<R: Resolve> Future for HttpConnecting<R> {
.collect();
let addrs = dns::IpAddrs::new(addrs);
state = State::Connecting(ConnectingTcp::new(
*local_addr, addrs, *me.connect_timeout, *me.happy_eyeballs_timeout, *me.reuse_address));
config.local_address, addrs, config.connect_timeout, config.happy_eyeballs_timeout, config.reuse_address));
},
State::Connecting(ref mut c) => {
let sock = ready!(c.poll(cx, &me.handle))?;
let sock = ready!(c.poll(cx, &config.handle))?;
if let Some(dur) = me.keep_alive_timeout {
sock.set_keepalive(Some(*dur))?;
if let Some(dur) = config.keep_alive_timeout {
sock.set_keepalive(Some(dur))?;
}
if let Some(size) = me.send_buffer_size {
sock.set_send_buffer_size(*size)?;
if let Some(size) = config.send_buffer_size {
sock.set_send_buffer_size(size)?;
}
if let Some(size) = me.recv_buffer_size {
sock.set_recv_buffer_size(*size)?;
if let Some(size) = config.recv_buffer_size {
sock.set_recv_buffer_size(size)?;
}
sock.set_nodelay(*me.nodelay)?;
sock.set_nodelay(config.nodelay)?;
let extra = HttpInfo {
remote_addr: sock.peer_addr()?,