src/client/http_client.rs: directly return H2Client on upgrade

This commit is contained in:
Dietmar Maurer
2019-05-13 16:44:59 +02:00
parent 10130cf48c
commit 850ac6d0f0
2 changed files with 5 additions and 4 deletions

View File

@ -242,7 +242,7 @@ impl HttpClient {
pub fn h2upgrade(
&mut self, path:
&str, param: Option<Value>
) -> impl Future<Item=h2::client::SendRequest<bytes::Bytes>, Error=Error> {
) -> impl Future<Item=H2Client, Error=Error> {
let mut req = Self::request_builder(&self.server, "GET", path, param).unwrap();
@ -279,7 +279,9 @@ impl HttpClient {
hyper::rt::spawn(connection);
// Wait until the `SendRequest` handle has available capacity.
h2.ready().map_err(Error::from)
h2.ready()
.map(H2Client::new)
.map_err(Error::from)
})
})
}