在使用 asyncio 时,可以利用 asyncio.gather() 或者是 asyncio.wait() 进行一次执行多个协程。这样,如果一个协程出现了时间阻塞,那么其他协程依然可以同时执行,以达到更高的效率。
另外,在使用 asyncio.sleep() 时,需要注意传入的时间参数单位为秒而非毫秒。可以通过将时间除以 1000 的方式将毫秒转换为秒。此外,如果需要更精细的调度管理,可以使用 aiohttp-limiter 进行更灵活的控制。以下是一个使用 asyncio.gather() 解决 Asyncio Sleep 满足率限制的代码示例:
import asyncio
import aiohttp
async def fetch_page(session, url):
async with session.get(url) as response:
if response.status == 429:
await asyncio.sleep(int(response.headers['Retry-After']))
elif response.status == 200:
await asyncio.sleep(0.5)
else:
print(f"Unhandled status code {response.status} for {url}")
async def main():
async with aiohttp.ClientSession() as session:
tasks = []
for i in range(10):
task = fetch_page(session, f"http://example.com/page/{i}")
tasks.append(task)
await asyncio.gather(*tasks)
if __name__ == '__main__':
asyncio.run(main())
在这个示例代码中,当 HTTP 响应状态为 429 时,我们使用了 asyncio.sleep(),它的参数获取自响应头的 Retry-After 值。当响应状态为 200 时,我们延迟了 0.5 秒 而不是使用了 asyncio.sleep()。