LBRYCommunity/lbry-sdk
1
0
Fork 1
Code Issues Pull requests Projects Releases 282 Packages Wiki Activity

fix slow ping queue shutdown

Browse source
This commit is contained in:
Jack Robison 2019-02-21 17:44:02 -05:00
parent 7fe9513857
commit fc64d8d1a8
No known key found for this signature in database
GPG key ID: DF25C68FE0239BB2
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._pending_contacts: typing.Dict['KademliaPeer', float] = {}
self._process_task: asyncio.Task = None self._process_task: asyncio.Task = None
self._running = False self._running = False
self._running_pings: typing.List[asyncio.Task] = [] self._running_pings: typing.Set[asyncio.Task] = set()
@property @property
def running(self): def running(self):
@ -215,7 +215,7 @@ class PingQueue:
task = self._loop.create_task(ping_task()) 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)) 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 async def _process(self): # send up to 1 ping per second
while True: while True:
@ -241,7 +241,7 @@ class PingQueue:
self._process_task.cancel() self._process_task.cancel()
self._process_task = None self._process_task = None
while self._running_pings: while self._running_pings:
self._running_pings[0].cancel() self._running_pings.pop().cancel()
class KademliaProtocol(DatagramProtocol): class KademliaProtocol(DatagramProtocol):