Fix timeouts due to hitting the gateway rate limit

This commit is contained in:
Rapptz 2020-09-07 21:50:00 -04:00
parent 7db0008190
commit 0ea0f483bc

View File

@ -753,7 +753,7 @@ class ConnectionState:
self._chunk_requests.append(request)
await self.chunker(guild.id, nonce=request.nonce)
if wait:
await request.future
return await request.future
return request.future
async def _chunk_and_dispatch(self, guild, unavailable):
@ -1063,7 +1063,7 @@ class AutoShardedConnectionState(ConnectionState):
log.debug('Guild ID %d requires chunking, will be done in the background.', guild.id)
if len(current_bucket) >= max_concurrency:
try:
await utils.sane_wait_for(current_bucket, timeout=max_concurrency * 10)
await utils.sane_wait_for(current_bucket, timeout=max_concurrency * 70.0)
except asyncio.TimeoutError:
fmt = 'Shard ID %s failed to wait for chunks from a sub-bucket with length %d'
log.warning(fmt, self.shard_id, len(current_bucket))