Call Pin::set and Pin::as_mut as method

Author: taiki-e

futures-core/src/future/future_obj.rs

@@ -173,7 +173,7 @@ where
     F: Future<Output = T> + 'a
 {
     fn into_raw(mut self) -> *mut () {
-        let mut_ref: &mut F = unsafe { Pin::get_unchecked_mut(Pin::as_mut(&mut self)) };
+        let mut_ref: &mut F = unsafe { Pin::get_unchecked_mut(self.as_mut()) };
         mut_ref as *mut F as *mut ()
     }
 
@@ -214,7 +214,7 @@ mod if_alloc {
         F: Future<Output = T> + 'a
     {
         fn into_raw(mut self) -> *mut () {
-            let mut_ref: &mut F = unsafe { Pin::get_unchecked_mut(Pin::as_mut(&mut self)) };
+            let mut_ref: &mut F = unsafe { Pin::get_unchecked_mut(self.as_mut()) };
             let ptr = mut_ref as *mut F as *mut ();
             mem::forget(self); // Don't drop the box
             ptr

futures-core/src/stream/stream_obj.rs

@@ -215,7 +215,7 @@ mod if_alloc {
         where F: Stream<Item = T> + 'a
     {
         fn into_raw(mut self) -> *mut () {
-            let mut_ref: &mut F = unsafe { Pin::get_unchecked_mut(Pin::as_mut(&mut self)) };
+            let mut_ref: &mut F = unsafe { Pin::get_unchecked_mut(self.as_mut()) };
             let ptr = mut_ref as *mut F as *mut ();
             mem::forget(self); // Don't drop the box
             ptr

futures-util/src/future/fuse.rs

@@ -42,7 +42,7 @@ impl<Fut: Future> Future for Fuse<Fut> {
             None => return Poll::Pending,
         };
 
-        Pin::set(&mut self.as_mut().future(), None);
+        self.as_mut().future().set(None);
         Poll::Ready(v)
     }
 }

futures-util/src/future/into_stream.rs

@@ -35,7 +35,7 @@ impl<Fut: Future> Stream for IntoStream<Fut> {
             None => return Poll::Ready(None),
         };
 
-        Pin::set(&mut self.as_mut().future(), None);
+        self.as_mut().future().set(None);
         Poll::Ready(Some(v))
     }
 }

futures-util/src/future/maybe_done.rs

@@ -106,7 +106,7 @@ impl<Fut: Future> Future for MaybeDone<Fut> {
                 MaybeDone::Gone => panic!("MaybeDone polled after value taken"),
             }
         };
-        Pin::set(&mut self, MaybeDone::Done(res));
+        self.set(MaybeDone::Done(res));
         Poll::Ready(())
     }
 }

futures-util/src/sink/with_flat_map.rs

@@ -98,7 +98,7 @@ where
                 };
             }
         }
-        Pin::set(&mut stream, None);
+        stream.set(None);
         Poll::Ready(Ok(()))
     }
 }

futures-util/src/stream/for_each.rs

@@ -55,7 +55,7 @@ impl<St, Fut, F> Future for ForEach<St, Fut, F>
             if let Some(future) = self.as_mut().future().as_pin_mut() {
                 ready!(future.poll(cx));
             }
-            self.as_mut().future().as_mut().set(None);
+            self.as_mut().future().set(None);
 
             match ready!(self.as_mut().stream().poll_next(cx)) {
                 Some(e) => {

futures-util/src/stream/unfold.rs

@@ -93,7 +93,7 @@ impl<T, F, Fut, It> Stream for Unfold<T, F, Fut>
     ) -> Poll<Option<It>> {
         if let Some(state) = self.as_mut().state().take() {
             let fut = (self.as_mut().f())(state);
-            Pin::set(&mut self.as_mut().fut(), Some(fut));
+            self.as_mut().fut().set(Some(fut));
         }
 
         let step = ready!(self.as_mut().fut().as_pin_mut().unwrap().poll(cx));

futures-util/src/try_future/flatten_sink.rs

@@ -55,7 +55,7 @@ where
             Waiting(f) => try_ready!(f.try_poll(cx)),
             Closed => panic!("poll_ready called after eof"),
         };
-        Pin::set(&mut self.as_mut(), FlattenSink(Ready(resolved_stream)));
+        self.set(FlattenSink(Ready(resolved_stream)));
         if let Ready(resolved_stream) = self.project_pin() {
             resolved_stream.poll_ready(cx)
         } else {
@@ -95,7 +95,7 @@ where
             Waiting(_) | Closed => Poll::Ready(Ok(())),
         };
         if res.is_ready() {
-            Pin::set(&mut self, FlattenSink(Closed));
+            self.set(FlattenSink(Closed));
         }
         res
     }

futures-util/src/try_stream/try_for_each.rs

@@ -45,12 +45,12 @@ impl<St, Fut, F> Future for TryForEach<St, Fut, F>
             if let Some(future) = self.as_mut().future().as_pin_mut() {
                 try_ready!(future.try_poll(cx));
             }
-            Pin::set(&mut self.as_mut().future(), None);
+            self.as_mut().future().set(None);
 
             match ready!(self.as_mut().stream().try_poll_next(cx)) {
                 Some(Ok(e)) => {
                     let future = (self.as_mut().f())(e);
-                    Pin::set(&mut self.as_mut().future(), Some(future));
+                    self.as_mut().future().set(Some(future));
                 }
                 Some(Err(e)) => return Poll::Ready(Err(e)),
                 None => return Poll::Ready(Ok(())),