asyncio.Semaphore

发布时间 2023-05-08 12:19:17作者: AngDH

 

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协程完成。