可以使用asyncio.Semaphore和async with结合来限制并发请求的数量,从而解决aiohttp.ClientSession().get在大量迭代后停止工作的问题。
示例代码:
import asyncio import aiohttp
async def fetch(session, url): async with session.get(url) as response: return await response.text()
async def main(): async with aiohttp.ClientSession() as session: semaphore = asyncio.Semaphore(50) # 在这里设置并发请求数量 tasks = [] for i in range(100): task = asyncio.ensure_future(worker(session, semaphore, f'http://example.com/{i}')) tasks.append(task) await asyncio.gather(*tasks)
async def worker(session, semaphore, url): async with semaphore: res = await fetch(session, url) print(res)
if name == 'main': loop = asyncio.get_event_loop() loop.run_until_complete(main())
上一篇:aiohttp-graphql 异步执行器 GraphQLLocatedError: 'NoneType' 对象不可调用。
下一篇:aiohttp.ClientSession().post和requests.post之间有什么区别?我使用这两个包时从服务器得到不同的响应