import asyncio
import concurrent.futures
import time
def cpu_bound():
# CPU-bound operations will block the event loop:
# in general it is preferable to run them in a
# process pool.
return sum(i * i for i in range(10**7))
async def do_one(pool):
loop = asyncio.get_running_loop()
return await loop.run_in_executor(pool, cpu_bound)
async def do_all(pool):
coros = [do_one(pool) for _ in range(10)]
return await asyncio.gather(*coros, return_exceptions=True)
async def main():
start_time = time.perf_counter()
# 1. Run in the default loop's executor:
result = await do_all(None)
print(f"default thread pool: {time.perf_counter()-start_time:.2f}s", result)
start_time = time.perf_counter()
# 2. Run in a custom process pool:
with concurrent.futures.ProcessPoolExecutor() as pool:
result = await do_all(pool)
print(f"custom process pool: {time.perf_counter()-start_time:.2f}s", result)
asyncio.run(main())
asyncio和ProcessPoolExecutor
最新推荐文章于 2024-08-26 07:38:45 发布