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

View File

@ -43,6 +43,9 @@ class SwarmConn(INetConn):
# We *could* optimize this but it really isn't worth it.
for stream in self.streams:
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:
task.cancel()