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

use cache_concurrent to avoid requesting the same connection_id multiple times

Browse source
This commit is contained in:
Victor Shyba 2022-03-09 20:11:30 -03:00
parent 4e09b35012
commit 6e5c7a1927
1 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
lbry/torrent/tracker.py
View file

@ -5,7 +5,7 @@ import logging
import time import time
from collections import namedtuple from collections import namedtuple
from lbry.utils import resolve_host, async_timed_cache from lbry.utils import resolve_host, async_timed_cache, cache_concurrent
from lbry.wallet.stream import StreamController from lbry.wallet.stream import StreamController
log = logging.getLogger(__name__) log = logging.getLogger(__name__)
@ -83,6 +83,7 @@ class UDPTrackerClientProtocol(asyncio.DatagramProtocol):
return decode(ConnectResponse, return decode(ConnectResponse,
await self.request(ConnectRequest(0x41727101980, 0, transaction_id), tracker_ip, tracker_port)) await self.request(ConnectRequest(0x41727101980, 0, transaction_id), tracker_ip, tracker_port))
@cache_concurrent
@async_timed_cache(CONNECTION_EXPIRES_AFTER_SECONDS) @async_timed_cache(CONNECTION_EXPIRES_AFTER_SECONDS)
async def ensure_connection_id(self, peer_id, tracker_ip, tracker_port): async def ensure_connection_id(self, peer_id, tracker_ip, tracker_port):
# peer_id is just to ensure cache coherency # peer_id is just to ensure cache coherency
@ -98,9 +99,7 @@ class UDPTrackerClientProtocol(asyncio.DatagramProtocol):
return decode(AnnounceResponse, await self.request(req, tracker_ip, tracker_port)) return decode(AnnounceResponse, await self.request(req, tracker_ip, tracker_port))
async def scrape(self, infohashes, tracker_ip, tracker_port, connection_id=None): async def scrape(self, infohashes, tracker_ip, tracker_port, connection_id=None):
if not connection_id: connection_id = await self.ensure_connection_id(None, tracker_ip, tracker_port)
reply = await self.connect(tracker_ip, tracker_port)
connection_id = reply.connection_id
transaction_id = random.getrandbits(32) transaction_id = random.getrandbits(32)
reply = await self.request( reply = await self.request(
ScrapeRequest(connection_id, 2, transaction_id, infohashes), tracker_ip, tracker_port) ScrapeRequest(connection_id, 2, transaction_id, infohashes), tracker_ip, tracker_port)