Skip to content

Commit

Permalink
feat(client): add conn::http1::Connection::without_shutdown() method (
Browse files Browse the repository at this point in the history
  • Loading branch information
kiron1 committed Nov 20, 2023
1 parent 4899703 commit ad50497
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions src/client/conn/http1.rs
Expand Up @@ -93,6 +93,16 @@ where
.expect("algready upgraded")
.poll_without_shutdown(cx)
}

/// Prevent shutdown of the underlying IO object at the end of service the request,
/// instead run `into_parts`. This is a convenience wrapper over `poll_without_shutdown`.
pub fn without_shutdown(self) -> impl Future<Output = crate::Result<Parts<T>>> {
let mut conn = Some(self);
futures_util::future::poll_fn(move |cx| -> Poll<crate::Result<Parts<T>>> {
ready!(conn.as_mut().unwrap().poll_without_shutdown(cx))?;
Poll::Ready(Ok(conn.take().unwrap().into_parts()))
})
}
}

/// A builder to configure an HTTP connection.
Expand Down

0 comments on commit ad50497

Please sign in to comment.