cancel other tasks if joined (#402)

Author: loongs-zhang

README.md

@@ -159,6 +159,20 @@ fn main() {
 }
 ```
 
+### wait until any task is completed or timed out
+
+```rust
+#[open_coroutine::main]
+fn main() {
+    let result = open_coroutine::any_timeout_join!(
+        std::time::Duration::from_secs(1),
+        open_coroutine::task!(|_| 1, ()),
+        open_coroutine::task!(|_| 2, ()),
+        open_coroutine::task!(|_| 3, ())
+    ).expect("timeout");
+}
+```
+
 ### scalable stack
 
 ```rust

README_ZH.md

@@ -155,6 +155,20 @@ fn main() {
 }
 ```
 
+### 等待任一任务完成或超时
+
+```rust
+#[open_coroutine::main]
+fn main() {
+    let result = open_coroutine::any_timeout_join!(
+        std::time::Duration::from_secs(1),
+        open_coroutine::task!(|_| 1, ()),
+        open_coroutine::task!(|_| 2, ()),
+        open_coroutine::task!(|_| 3, ())
+    ).expect("timeout");
+}
+```
+
 ### 可伸缩栈
 
 ```rust

core/src/co_pool/mod.rs

@@ -255,6 +255,7 @@ impl<'p> CoroutinePool<'p> {
             return;
         }
         _ = self.no_waits.insert(Box::leak(Box::from(task_name)));
+        _ = CANCEL_TASKS.remove(task_name);
     }
 
     /// Use the given `task_name` to obtain task results, and if no results are found,

open-coroutine/src/lib.rs

@@ -203,7 +203,12 @@ impl<R> JoinHandle<R> {
     }
 
     pub fn any_join<I: IntoIterator<Item = Self>>(iter: I) -> std::io::Result<Option<R>> {
-        Self::any_timeout_join(Duration::MAX, Vec::from_iter(iter).as_slice())
+        let vec = Vec::from_iter(iter);
+        Self::any_timeout_join(Duration::MAX, &vec).inspect(|_| {
+            for handle in vec {
+                _ = handle.try_cancel();
+            }
+        })
     }
 
     pub fn try_cancel(self) -> std::io::Result<()> {
@@ -239,7 +244,7 @@ impl<R> Deref for JoinHandle<R> {
 #[macro_export]
 macro_rules! any_timeout_join {
     ($time:expr, $($x:expr),+ $(,)?) => {
-        $crate::JoinHandle::any_timeout_join($time, vec![$($x),+].as_slice())
+        $crate::JoinHandle::any_timeout_join($time, &vec![$($x),+])
     }
 }