fixes #206

boostorg · Oct 22, 2024 · cf6f0e5 · cf6f0e5
1 parent 2eb10b3
commit cf6f0e5
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/include/boost/cobalt/channel.hpp b/include/boost/cobalt/channel.hpp
@@ -85,6 +85,13 @@ struct channel
       this->unlink();
     }
 
+    void interrupt_await()
+    {
+      this->cancelled = true;
+      if (awaited_from)
+        awaited_from.release().resume();
+    }
+
     struct cancel_impl;
     bool await_ready() { return !chn->buffer_.empty(); }
     template<typename Promise>
@@ -118,6 +125,13 @@ struct channel
       this->unlink();
     }
 
+    void interrupt_await()
+    {
+      this->cancelled = true;
+      if (awaited_from)
+        awaited_from.release().resume();
+    }
+
     struct cancel_impl;
 
     bool await_ready() { return !chn->buffer_.full(); }
@@ -234,6 +248,13 @@ struct channel<void>
       this->unlink();
     }
 
+    void interrupt_await()
+    {
+      this->cancelled = true;
+      if (awaited_from)
+        awaited_from.release().resume();
+    }
+
     struct cancel_impl;
     bool await_ready() { return (chn->n_ > 0); }
     template<typename Promise>
@@ -261,6 +282,13 @@ struct channel<void>
       this->unlink();
     }
 
+    void interrupt_await()
+    {
+      this->cancelled = true;
+      if (awaited_from)
+        awaited_from.release().resume();
+    }
+
     struct cancel_impl;
     bool await_ready()
     {