prevent memory leak risk

Author: netcan

include/asyncio/gather.h

@@ -28,7 +28,13 @@ class GatherAwaiter: NonCopyable {
     void await_suspend(std::coroutine_handle<Promise> continuation) noexcept {
         continuation_ = &continuation.promise();
         // set continuation_ to PENDING, don't schedule anymore, until it resume continuation_
-        continuation_->set_state(PromiseState::PENDING);
+        continuation.promise().set_state(PromiseState::PENDING);
+    }
+
+    ~GatherAwaiter() {
+        if (continuation_) {
+            continuation_->set_state(PromiseState::UNSCHEDULED);
+        }
     }
 
     template<concepts::Awaitable... Futs>

include/asyncio/handle.h

@@ -15,7 +15,7 @@ enum class PromiseState: uint8_t {
     PENDING,
 };
 
-struct Handle {
+struct Handle { // type erase for EventLoop
     virtual void run() = 0;
     std::string frame_name() const {
         const auto& frame_info = get_frame_info();

include/asyncio/wait_for.h

@@ -21,10 +21,16 @@ struct WaitForAwaiter: NonCopyable {
     }
 
     template<typename Promise>
-    void await_suspend(std::coroutine_handle<Promise> caller) noexcept {
-        continuation_ = &caller.promise();
+    void await_suspend(std::coroutine_handle<Promise> continuation) noexcept {
+        continuation_ = &continuation.promise();
         // set continuation_ to PENDING, don't schedule anymore, until it resume continuation_
-        continuation_->set_state(PromiseState::PENDING);
+        continuation.promise().set_state(PromiseState::PENDING);
+    }
+
+    ~WaitForAwaiter() {
+        if (continuation_) {
+            continuation_->set_state(PromiseState::UNSCHEDULED);
+        }
     }
 
     template<concepts::Awaitable Fut>