Add Semaphore bound for closed impl

Signed-off-by: Zahari Dichev <zaharidichev@gmail.com>
tokio-rs · Sep 25, 2020 · 19bfab1 · 19bfab1
1 parent edd97a2
commit 19bfab1
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/tokio/src/sync/mpsc/chan.rs b/tokio/src/sync/mpsc/chan.rs
@@ -136,6 +136,13 @@ impl<T, S> Tx<T, S> {
         self.inner.send(value);
     }
 
+    /// Wake the receive half
+    pub(crate) fn wake_rx(&self) {
+        self.inner.rx_waker.wake();
+    }
+}
+
+impl<T, S: Semaphore> Tx<T, S> {
     pub(crate) async fn closed(&mut self) {
         use std::future::Future;
         use std::pin::Pin;
@@ -162,11 +169,6 @@ impl<T, S> Tx<T, S> {
         }
         notified.await;
     }
-
-    /// Wake the receive half
-    pub(crate) fn wake_rx(&self) {
-        self.inner.rx_waker.wake();
-    }
 }
 
 impl<T, S> Clone for Tx<T, S> {