StarMirror/py-libp2p
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix:

Browse Source 
Force context switch before canceling swarm connection tasks
This commit is contained in:
NIC619 2019-11-04 18:27:31 +08:00
parent a8d9536b08
commit 97b3aca535
No known key found for this signature in database
GPG Key ID: 570C35F5C2D51B17
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
libp2p/network/connection/swarm_connection.py
View File

@ -43,6 +43,9 @@ class SwarmConn(INetConn):
# We *could* optimize this but it really isn't worth it. # We *could* optimize this but it really isn't worth it.
for stream in self.streams: for stream in self.streams:
await stream.reset() await stream.reset()
# Force context switch for stream handlers to process the stream reset event we just emit
# before we cancel the stream handler tasks.
await asyncio.sleep(0.1)
for task in self._tasks: for task in self._tasks:
task.cancel() task.cancel()