Asyncio Semaphore

新的启动事件循环

import asyncio
import logging
from contextlib import closing

import nest_asyncio

nest_asyncio.apply( )  # 在jupyter需要这个，不然asyncio运行出错
logging.basicConfig(  # 用日志打印输出信息
    level = logging.INFO,
    format = "%(asctime)s %(process)d %(thread)d [*] %(message)s"
)


async def myWorker(name, semaphore):
    await semaphore.acquire( )
    logging.info(f"{name}:成功获取信号量")
    await asyncio.sleep(3)
    logging.info(f"{name}:释放信号量")
    semaphore.release( )


async def main( ):
    logging.info("主协程")
    mySemaphore = asyncio.Semaphore(value = 2)  # 有界信号量（等待其中两个协程结束）
    await asyncio.wait([
        myWorker("A", mySemaphore),
        myWorker("B", mySemaphore),
        myWorker("C", mySemaphore),
        myWorker("D", mySemaphore),
        myWorker("E", mySemaphore),
    ])


if __name__ == "__main__":
    # 修改了这里
    with closing(asyncio.get_event_loop( )) as loop:
        loop.run_until_complete(main( ))

旧的启动事件循环

import asyncio
import time
import logging
import nest_asyncio
nest_asyncio.apply()  # 在jupyter需要这个，不然asyncio运行出错
logging.basicConfig(  # 用日志打印输出信息
    level=logging.INFO,
    format="%(asctime)s %(process)d %(thread)d [*] %(message)s"
)


async def myWorker(semaphore):
    await semaphore.acquire()
    logging.info("成功获取信号量")
    await asyncio.sleep(3)
    logging.info("释放信号量")
    semaphore.release()


async def main(loop):
    logging.info("主协程")
    mySemaphore = asyncio.Semaphore(value=2)  # 有界信号量（等待其中两个协程结束）
    return await asyncio.wait([
        myWorker(mySemaphore),
        myWorker(mySemaphore),
        myWorker(mySemaphore),
    ])

if __name__ == '__main__':
    loop = asyncio.get_event_loop()
    try:
        done, pending = loop.run_until_complete(main(loop))
        logging.info("所有任务完成")
    except KeyboardInterrupt as e:
        asyncio.Task.all_tasks()
        asyncio.gather(*asyncio.Task.all_tasks()).cancel()
        loop.stop()
#         loop.run_forever()  # 在jupyter不能手动关闭
    finally:
        pass
#         loop.close()  # 在jupyter不能手动关闭

2019-11-15 17:32:19,892 12912 9828 [*] 主协程
2019-11-15 17:32:19,894 12912 9828 [*] 成功获取信号量
2019-11-15 17:32:19,896 12912 9828 [*] 成功获取信号量
2019-11-15 17:32:22,897 12912 9828 [*] 释放信号量
2019-11-15 17:32:22,901 12912 9828 [*] 释放信号量
2019-11-15 17:32:22,907 12912 9828 [*] 成功获取信号量
2019-11-15 17:32:25,916 12912 9828 [*] 释放信号量
2019-11-15 17:32:25,921 12912 9828 [*] 所有任务完成