fix poll_shutdown

narrowlink · Mar 4, 2024 · e4a3b6e · e4a3b6e
1 parent 868a11f
commit e4a3b6e
Showing 1 changed file with 1 addition and 8 deletions.
diff --git a/src/stream/tcp.rs b/src/stream/tcp.rs
@@ -478,19 +478,12 @@ impl AsyncWrite for IpStackTcpStream {
         std::task::Poll::Ready(Ok(()))
     }
 
-    #[allow(unused_mut, unused_variables)]
     fn poll_shutdown(
         mut self: std::pin::Pin<&mut Self>,
         cx: &mut std::task::Context<'_>,
     ) -> std::task::Poll<Result<(), std::io::Error>> {
-        /*
         let notified = self.shutdown.get_or_insert(Notify::new()).notified();
-        match Pin::new(&mut Box::pin(notified)).poll(cx) {
-            std::task::Poll::Ready(_) => std::task::Poll::Ready(Ok(())),
-            std::task::Poll::Pending => std::task::Poll::Pending,
-        }
-        // */
-        std::task::Poll::Ready(Ok(()))
+        Pin::new(&mut Box::pin(notified)).poll(cx).map(Ok)
     }
 }