asyncio.Semaphore
asyncio.Semaphore是一个异步信号量,用于协调多个协程对共享资源的访问。
异步信号量在协程中的使用方式与线程中的普通信号量类似,但是它是非阻塞的。当信号量的计数器为0时,协程将会被阻塞,直到其他协程释放了该信号量。
import asyncio async def worker(semaphore): async with semaphore: # do something await asyncio.sleep(3) print("Done") async def main(): semaphore = asyncio.Semaphore(5) tasks = [worker(semaphore) for _ in range(10)] await asyncio.gather(*tasks) print("Done all") asyncio.run(main())
在上面的代码中,我们创建了一个大小为5的异步信号量,并创建了10个worker协程来尝试获取信号量。每个worker协程使用async with语句获取信号量,当信号量的计数器为0时,协程将会被阻塞,直到其他协程释放了该信号量。
最后,我们使用asyncio.gather()函数来等待所有worker协程完成。
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步