fix

ljcui · Dec 9, 2024 · e92a201 · e92a201
1 parent 2ec4c3f
commit e92a201
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 2 deletions.
diff --git a/src/bolt/blocking_queue.h b/src/bolt/blocking_queue.h
@@ -32,6 +32,9 @@ class BlockingQueue {
         }
         condition_.notify_one();
     }
+    void Notify() {
+        condition_.notify_one();
+    }
     T Pop() {
         std::unique_lock<std::mutex> lock(mutex_);
         condition_.wait(lock, [this] {return !queue_.empty();});

diff --git a/src/bolt/connection.cpp b/src/bolt/connection.cpp
@@ -22,6 +22,7 @@
 #include "bolt/connection.h"
 #include "bolt/messages.h"
 #include "bolt/to_string.h"
+#include "server/bolt_session.h"
 
 namespace bolt {
 using namespace boost::asio;
@@ -100,6 +101,10 @@ void BoltConnection::Start() {
 
 void BoltConnection::Close() {
     Connection::Close();
+    if (context_) {
+        auto session = (BoltSession*)GetContext();
+        session->msgs.Notify();
+    }
 }
 
 void BoltConnection::DoSend() {

diff --git a/src/core/thread_id.h b/src/core/thread_id.h
@@ -18,7 +18,7 @@
 #include "fma-common/assert.h"
 
 namespace lgraph {
-static const int LGRAPH_MAX_THREADS = 480;
+static const int LGRAPH_MAX_THREADS = 65000;
 
 class ThreadIdAssigner {
     static const int N = LGRAPH_MAX_THREADS;

diff --git a/src/server/bolt_handler.cpp b/src/server/bolt_handler.cpp
@@ -121,7 +121,7 @@ void BoltFSM(std::shared_ptr<BoltConnection> conn) {
         session->state = SessionState::FAILED;
     };
     while (!conn->has_closed()) {
-        auto msg = session->msgs.Pop(std::chrono::milliseconds(100));
+        auto msg = session->msgs.Pop(std::chrono::milliseconds(10));
         if (!msg) {  // msgs pop timeout
             continue;
         }