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

refactor(lib): Remove useless uses of Pin (#2405)

Browse Source
This commit is contained in:
Paolo Barbolini
2021-01-19 23:51:34 +01:00
committed by GitHub
parent 9956587f83
commit 237b2ce083
3 changed files with 16 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/client/dispatch.rs
View File

@@ -138,25 +138,22 @@ impl<T, U> Clone for UnboundedSender<T, U> {
}
}
#[pin_project::pin_project(PinnedDrop)]
pub(crate) struct Receiver<T, U> {
#[pin]
inner: mpsc::UnboundedReceiver<Envelope<T, U>>,
taker: want::Taker,
}
impl<T, U> Receiver<T, U> {
pub(crate) fn poll_next(
self: Pin<&mut Self>,
pub(crate) fn poll_recv(
&mut self,
cx: &mut task::Context<'_>,
) -> Poll<Option<(T, Callback<T, U>)>> {
let mut this = self.project();
match this.inner.poll_recv(cx) {
match self.inner.poll_recv(cx) {
Poll::Ready(item) => {
Poll::Ready(item.map(|mut env| env.0.take().expect("envelope not dropped")))
}
Poll::Pending => {
this.taker.want();
self.taker.want();
Poll::Pending
}
}
@@ -177,12 +174,11 @@ impl<T, U> Receiver<T, U> {
}
}
#[pin_project::pinned_drop]
impl<T, U> PinnedDrop for Receiver<T, U> {
fn drop(mut self: Pin<&mut Self>) {
impl<T, U> Drop for Receiver<T, U> {
fn drop(&mut self) {
// Notify the giver about the closure first, before dropping
// the mpsc::Receiver.
self.as_mut().taker.cancel();
self.taker.cancel();
}
}
@@ -279,8 +275,8 @@ mod tests {
impl<T, U> Future for Receiver<T, U> {
type Output = Option<(T, Callback<T, U>)>;
fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
self.poll_next(cx)
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
self.poll_recv(cx)
}
}

14
src/proto/h1/dispatch.rs
View File

@@ -44,10 +44,8 @@ cfg_server! {
}
cfg_client! {
#[pin_project::pin_project]
pub(crate) struct Client<B> {
callback: Option<crate::client::dispatch::Callback<Request<B>, http::Response<Body>>>,
#[pin]
rx: ClientRx<B>,
rx_closed: bool,
}
@@ -557,12 +555,12 @@ cfg_client! {
type RecvItem = crate::proto::ResponseHead;
fn poll_msg(
self: Pin<&mut Self>,
mut self: Pin<&mut Self>,
cx: &mut task::Context<'_>,
) -> Poll<Option<Result<(Self::PollItem, Self::PollBody), crate::common::Never>>> {
let this = self.project();
debug_assert!(!*this.rx_closed);
match this.rx.poll_next(cx) {
let mut this = self.as_mut();
debug_assert!(!this.rx_closed);
match this.rx.poll_recv(cx) {
Poll::Ready(Some((req, mut cb))) => {
// check that future hasn't been canceled already
match cb.poll_canceled(cx) {
@@ -578,7 +576,7 @@ cfg_client! {
headers: parts.headers,
extensions: parts.extensions,
};
*this.callback = Some(cb);
this.callback = Some(cb);
Poll::Ready(Some(Ok((head, body))))
}
}
@@ -586,7 +584,7 @@ cfg_client! {
Poll::Ready(None) => {
// user has dropped sender handle
trace!("client tx closed");
*this.rx_closed = true;
this.rx_closed = true;
Poll::Ready(None)
}
Poll::Pending => Poll::Pending,

2
src/proto/h2/client.rs
View File

@@ -213,7 +213,7 @@ where
}
};
match Pin::new(&mut self.req_rx).poll_next(cx) {
match self.req_rx.poll_recv(cx) {
Poll::Ready(Some((req, cb))) => {
// check that future hasn't been canceled already
if cb.is_canceled() {