Remove client node from Nodes on connection close

resetius · Dec 16, 2023 · 3e908e2 · 3e908e2
1 parent 902c6af
commit 3e908e2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/server.cpp b/src/server.cpp
@@ -114,8 +114,9 @@ NNet::TVoidSuspendedTask TNode<TSocket>::DoConnect() {
 
 template<typename TSocket>
 NNet::TVoidTask TRaftServer<TSocket>::InboundConnection(TSocket socket) {
+    std::shared_ptr<TNode<TSocket>> client;
     try {
-        auto client = std::make_shared<TNode<TSocket>>(
+        client = std::make_shared<TNode<TSocket>>(
             "client", std::move(socket), TimeSource
         );
         Nodes.insert(client);
@@ -128,6 +129,7 @@ NNet::TVoidTask TRaftServer<TSocket>::InboundConnection(TSocket socket) {
     } catch (const std::exception & ex) {
         std::cerr << "Exception: " << ex.what() << "\n";
     }
+    Nodes.erase(client);
     co_return;
 }