fix erasing unclosed stream in OnClosed

ykt66 · Sep 5, 2016 · ab238e3 · ab238e3
1 parent b2d270c
commit ab238e3
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/sofa/pbrpc/rpc_client_impl.cc b/src/sofa/pbrpc/rpc_client_impl.cc
@@ -402,6 +402,12 @@ void RpcClientImpl::OnClosed(const RpcClientStreamPtr& stream)
         return;
 
     ScopedLocker<FastLock> _(_stream_map_lock);
+    // to prevent erasing the new connection
+    StreamMap::iterator find = _stream_map.find(stream->remote_endpoint());
+    if (find != _stream_map.end() && !find->second->is_closed())
+    {
+        return;
+    }
     _stream_map.erase(stream->remote_endpoint());
 }