Merge pull request #101 from joyieldInc/FixPrivateConnLeak

fix private connection leak
joyieldInc · Feb 1, 2020 · 393ff81 · 393ff81
2 parents dacf3fb + ace6ed2
commit 393ff81
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/Handler.cpp b/src/Handler.cpp
@@ -215,7 +215,6 @@ void Handler::postAcceptConnectionEvent()
                 auto cp = mConnPool[s->server()->id()];
                 s->setStatus(Connection::LogicError);
                 addPostEvent(s, Multiplexor::ErrorEvent);
-                cp->putPrivateConnection(s);
                 c->detachConnectConnection();
                 s->detachAcceptConnection();
             }
@@ -276,6 +275,9 @@ void Handler::postConnectConnectionEvent()
                     s->status(), s->statusStr());
             mEventLoop->delSocket(s);
             s->close(this);
+            if (!s->isShared()) {
+                mConnPool[s->server()->id()]->putPrivateConnection(s);
+            }
             if (c) {
                 addPostEvent(c, Multiplexor::ErrorEvent);
                 s->detachAcceptConnection();