fix coroutine quit after destruct (#377)

qicosmos · Jul 26, 2023 · 0cc289a · 0cc289a
1 parent 93bd6b3
commit 0cc289a
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/include/cinatra/coro_http_client.hpp b/include/cinatra/coro_http_client.hpp
@@ -1513,12 +1513,21 @@ class coro_http_client {
 
     websocket ws{};
     while (true) {
+      std::weak_ptr socket = socket_;
       if (auto [ec, _] = co_await async_read(read_buf_, header_size); ec) {
         data.net_err = ec;
         data.status = 404;
-        close_socket(*socket_);
+        auto sock = socket.lock();
+        if (!sock) {
+          co_return;
+        }
+        if (!sock->has_closed_) {
+          close_socket(*sock);
+        }
+
         if (on_ws_msg_)
           on_ws_msg_(data);
+
         co_return;
       }