fix(async): complete pending operations during shutdown

Author: GaspardKirira

include/vix/async/core/scheduler.hpp

@@ -199,18 +199,12 @@ namespace vix::async::core
 
         {
           std::unique_lock<std::mutex> lock(m_);
+
           cv_.wait(lock, [this]()
                    { return stop_requested_.load(std::memory_order_acquire) ||
                             !handle_q_.empty() ||
                             !fn_q_.empty(); });
 
-          if (stop_requested_.load(std::memory_order_acquire))
-          {
-            handle_q_.clear();
-            fn_q_.clear();
-            break;
-          }
-
           if (!handle_q_.empty())
           {
             h = handle_q_.front();
@@ -221,6 +215,10 @@ namespace vix::async::core
             fn = std::move(fn_q_.front());
             fn_q_.pop_front();
           }
+          else if (stop_requested_.load(std::memory_order_acquire))
+          {
+            break;
+          }
         }
 
         if (h)

src/net/asio_await.hpp

@@ -81,11 +81,25 @@ namespace vix::async::net::detail
       vix::async::core::io_context *ctx,
       std::coroutine_handle<> h) noexcept
   {
-    if (!ctx || !h)
+    if (!h)
     {
       return;
     }
 
+    if (!ctx)
+    {
+      h.resume();
+      return;
+    }
+
+    auto &sched = ctx->get_scheduler();
+
+    if (sched.stop_requested())
+    {
+      h.resume();
+      return;
+    }
+
     ctx->post(h);
   }
 

src/net/asio_tcp.cpp

@@ -154,6 +154,18 @@ namespace vix::async::net
     void close() noexcept override
     {
       std::error_code ec;
+
+      if (!sock_.is_open())
+      {
+        return;
+      }
+
+      sock_.cancel(ec);
+      ec.clear();
+
+      sock_.shutdown(tcp::socket::shutdown_both, ec);
+      ec.clear();
+
       sock_.close(ec);
     }
 
@@ -245,6 +257,15 @@ namespace vix::async::net
     void close() noexcept override
     {
       std::error_code ec;
+
+      if (!acc_.is_open())
+      {
+        return;
+      }
+
+      acc_.cancel(ec);
+      ec.clear();
+
       acc_.close(ec);
     }
 

src/net/asio_udp.cpp

@@ -143,6 +143,15 @@ namespace vix::async::net
     void close() noexcept override
     {
       std::error_code ec;
+
+      if (!sock_.is_open())
+      {
+        return;
+      }
+
+      sock_.cancel(ec);
+      ec.clear();
+
       sock_.close(ec);
     }