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

reduce timeout to 10, fix lints

Browse source
This commit is contained in:
Victor Shyba 2022-03-09 16:42:20 -03:00
parent eccf0e6234
commit 0e4f1eae5b
1 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
lbry/torrent/tracker.py
View file

@ -60,7 +60,7 @@ def encode(obj):
class UDPTrackerClientProtocol(asyncio.DatagramProtocol): class UDPTrackerClientProtocol(asyncio.DatagramProtocol):
def __init__(self, timeout = 30.0): def __init__(self, timeout=10.0):
self.transport = None self.transport = None
self.data_queue = {} self.data_queue = {}
self.timeout = timeout self.timeout = timeout
@ -137,7 +137,8 @@ class TrackerClient:
async def start(self): async def start(self):
self.transport, _ = await asyncio.get_running_loop().create_datagram_endpoint( self.transport, _ = await asyncio.get_running_loop().create_datagram_endpoint(
lambda: self.client, local_addr=("0.0.0.0", 0)) lambda: self.client, local_addr=("0.0.0.0", 0))
self.EVENT_CONTROLLER.stream.listen(lambda request: self.on_hash(request[1]) if request[0] == 'search' else None) self.EVENT_CONTROLLER.stream.listen(
lambda request: self.on_hash(request[1]) if request[0] == 'search' else None)
def stop(self): def stop(self):
if self.transport is not None: if self.transport is not None:
@ -185,11 +186,11 @@ class TrackerClient:
self.results[info_hash] = (time.time() + 60.0, result) self.results[info_hash] = (time.time() + 60.0, result)
log.debug("Tracker timed out: %s:%d", tracker_host, tracker_port) log.debug("Tracker timed out: %s:%d", tracker_host, tracker_port)
return None return None
log.debug("Announced: %s found %d peers for %s on %s", tracker_host, len(result.peers), info_hash.hex()[:8]) log.debug("Announced: %s found %d peers for %s", tracker_host, len(result.peers), info_hash.hex()[:8])
return result return result
def subscribe_hash(hash: bytes, on_data): def subscribe_hash(info_hash: bytes, on_data):
TrackerClient.EVENT_CONTROLLER.add(('search', hash)) TrackerClient.EVENT_CONTROLLER.add(('search', info_hash))
TrackerClient.EVENT_CONTROLLER.stream.where(lambda request: request[0] == hash).add_done_callback( TrackerClient.EVENT_CONTROLLER.stream.where(lambda request: request[0] == info_hash).add_done_callback(
lambda request: on_data(request.result()[1])) lambda request: on_data(request.result()[1]))