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

schedule_requests is not async

Browse source
This commit is contained in:
Victor Shyba 2021-10-13 15:55:02 -03:00
parent 8a2fd29645
commit ea5a0f0090
2 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
lbry/wallet/rpc/session.py
View file

@ -424,7 +424,7 @@ class RPCSession(SessionBase):
self.max_errors = 0
self._bump_errors()
else:
await self.schedule_requests(requests)
self.schedule_requests(requests)
async def _handle_request(self, request):
start = time.perf_counter()
@ -471,7 +471,7 @@ class RPCSession(SessionBase):
async def handle_request(self, request):
pass
async def schedule_requests(self, requests):
def schedule_requests(self, requests):
for request in requests:
self._task_group.add(self._handle_request(request))

9
lbry/wallet/server/session.py
View file

@ -564,7 +564,12 @@ class SessionManager:
async def consumer():
while True:
_, fut = await self.priority_queue.get()
await fut
try:
await fut
except asyncio.CancelledError:
raise
except Exception as e:
log.exception("raised while serving a request. This should never happen.")
await asyncio.gather(*(consumer() for _ in range(self.consumers)))
async def start_other(self):
@ -892,7 +897,7 @@ class LBRYElectrumX(SessionBase):
self.db: LevelDB = self.bp.db
self.time_since_last_request = time.perf_counter()
async def schedule_requests(self, requests):
def schedule_requests(self, requests):
for request in requests:
current = time.perf_counter()
elapsed = current - self.time_since_last_request