Merge pull request baidu#109 from zd-double/new_branch

fix erasing unclosed stream
ykt66 · Sep 6, 2016 · 18b244c · 18b244c
2 parents 03ccd2c + ab238e3
commit 18b244c
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());
 }