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