Madiator2011/lbry-sdk
1
0
Fork 0
forked from LBRYCommunity/lbry-sdk
Code Pull requests Activity

Merge pull request #1950 from lbryio/fix-slow-ping-queue-shutdown

Browse source 
Fix slow ping queue shutdown
This commit is contained in:
Jack Robison 2019-02-21 18:12:22 -05:00 committed by GitHub
commit 0c67469321
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
lbrynet/dht/protocol/protocol.py
View file

@ -190,7 +190,7 @@ class PingQueue:
self._pending_contacts: typing.Dict['KademliaPeer', float] = {}
self._process_task: asyncio.Task = None
self._running = False
self._running_pings: typing.List[asyncio.Task] = []
self._running_pings: typing.Set[asyncio.Task] = set()
@property
def running(self):
@ -215,7 +215,7 @@ class PingQueue:
task = self._loop.create_task(ping_task())
task.add_done_callback(lambda _: None if task not in self._running_pings else self._running_pings.remove(task))
self._running_pings.append(task)
self._running_pings.add(task)
async def _process(self): # send up to 1 ping per second
while True:
@ -241,7 +241,7 @@ class PingQueue:
self._process_task.cancel()
self._process_task = None
while self._running_pings:
self._running_pings[0].cancel()
self._running_pings.pop().cancel()
class KademliaProtocol(DatagramProtocol):