fix coroutine quit after client destruct (#385)

alibaba · Jul 26, 2023 · fbc35e0 · fbc35e0
1 parent c1b2ee4
commit fbc35e0
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/include/ylt/thirdparty/cinatra/coro_http_client.hpp b/include/ylt/thirdparty/cinatra/coro_http_client.hpp
@@ -1492,12 +1492,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;
       }