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

use a semaphore to avoid false timeouts on large syncs

Browse source
This commit is contained in:
Victor Shyba 2019-08-23 13:31:17 -03:00
parent 089cefb77b
commit b9b8178e30
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
torba/torba/client/basenetwork.py
View file

@ -29,6 +29,8 @@ class ClientSession(BaseClientSession):
self.pending_amount = 0 self.pending_amount = 0
self._on_connect_cb = on_connect_callback or (lambda: None) self._on_connect_cb = on_connect_callback or (lambda: None)
self.trigger_urgent_reconnect = asyncio.Event() self.trigger_urgent_reconnect = asyncio.Event()
# one request per second of timeout, conservative default
self._semaphore = asyncio.Semaphore(self.timeout)
@property @property
def available(self): def available(self):
@ -54,8 +56,12 @@ class ClientSession(BaseClientSession):
return result return result
async def send_request(self, method, args=()): async def send_request(self, method, args=()):
log.debug("send %s to %s:%i", method, *self.server)
self.pending_amount += 1 self.pending_amount += 1
async with self._semaphore:
return await self._send_request(method, args)
async def _send_request(self, method, args=()):
log.debug("send %s to %s:%i", method, *self.server)
try: try:
if method == 'server.version': if method == 'server.version':
reply = await self.send_timed_server_version_request(args, self.timeout) reply = await self.send_timed_server_version_request(args, self.timeout)