varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2025-12-31 20:36:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Check if entry exists in dictionary before delete

Browse Source
This commit is contained in:
NIC619
2019-11-17 21:52:05 +08:00
parent 64c49f809f
commit c0522c1bd9
3 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
libp2p/stream_muxer/mplex/mplex.py
View File

@ -297,7 +297,8 @@ class Mplex(IMuxedConn):
# the entry of this stream, to avoid others from accessing it.
if is_local_closed:
async with self.streams_lock:
del self.streams[stream_id]
if stream_id in self.streams:
del self.streams[stream_id]
async def _handle_reset(self, stream_id: StreamID) -> None:
async with self.streams_lock:
@ -315,7 +316,8 @@ class Mplex(IMuxedConn):
if not stream.event_local_closed.is_set():
stream.event_local_closed.set()
async with self.streams_lock:
del self.streams[stream_id]
if stream_id in self.streams:
del self.streams[stream_id]
async def _cleanup(self) -> None:
if not self.event_shutting_down.is_set():