fix http server callback

Author: qicosmos

include/ylt/standalone/cinatra/coro_http_server.hpp

@@ -121,7 +121,7 @@ class coro_http_server {
 
     // close current connections.
     {
-      std::scoped_lock lock(conn_mtx_);
+      std::scoped_lock lock(*conn_mtx_);
       for (auto &conn : connections_) {
         conn.second->close(false);
       }
@@ -595,7 +595,7 @@ class coro_http_server {
   }
 
   size_t connection_count() {
-    std::scoped_lock lock(conn_mtx_);
+    std::scoped_lock lock(*conn_mtx_);
     return connections_.size();
   }
 
@@ -708,17 +708,20 @@ class coro_http_server {
         conn->init_ssl(cert_file_, key_file_, passwd_);
       }
 #endif
-
+      std::weak_ptr<std::mutex> weak(conn_mtx_);
       conn->set_quit_callback(
-          [this](const uint64_t &id) {
-            std::scoped_lock lock(conn_mtx_);
-            if (!connections_.empty())
-              connections_.erase(id);
+          [this, weak](const uint64_t &id) {
+            auto mtx = weak.lock();
+            if (mtx) {
+              std::scoped_lock lock(*mtx);
+              if (!connections_.empty())
+                connections_.erase(id);
+            }
           },
           conn_id);
 
       {
-        std::scoped_lock lock(conn_mtx_);
+        std::scoped_lock lock(*conn_mtx_);
         connections_.emplace(conn_id, conn);
       }
 
@@ -758,7 +761,7 @@ class coro_http_server {
     std::unordered_map<uint64_t, std::shared_ptr<coro_http_connection>> conns;
 
     {
-      std::scoped_lock lock(conn_mtx_);
+      std::scoped_lock lock(*conn_mtx_);
       for (auto it = connections_.begin();
            it != connections_.end();)  // no "++"!
       {
@@ -966,7 +969,7 @@ class coro_http_server {
   uint64_t conn_id_ = 0;
   std::unordered_map<uint64_t, std::shared_ptr<coro_http_connection>>
       connections_;
-  std::mutex conn_mtx_;
+  std::shared_ptr<std::mutex> conn_mtx_ = std::make_shared<std::mutex>();
   std::chrono::steady_clock::duration check_duration_ =
       std::chrono::seconds(15);
   std::chrono::steady_clock::duration timeout_duration_{};