asyncio 中给running 的loop 动态添加 Future Task

时间：2021-06-07 20:24:57

https://my.oschina.net/backbye/blog/1919486

asyncio 提供了两个给运行中的事件循环（loop) 添加 事件的方法

call_soon_threadsafe() 、run_coroutine_threadsafe()

因为我们一般的场景中会另起一个线程来启动loop（ loop.run_forever() ），所以上面两个方法设计成线程安全

但是上面两个方法并不能满足我们全部的需要：

1. run_coroutine_threadsafe 只能用于传入一个协程，而不能直接传入task对象（这样才方便我们在主进程中随时查看任务的执行情况和获取结果），当然可以去修改源码，但是不建议这么做

2. run_forever 虽然解决了可以随时加入任务的问题，但是带来一个新的问题：这个线程会一直运行下去，除非ctrl-C, 因为1，所以我们无法根据任务的执行情况来loop.stop, 而且由于线程安全，即使你执行了loop.stop() 也不会起任何作用

所以源头要回到线程安全上去解决，查看源码发现是通过一个管道文件来实现的，可以理解为内置了一个队列，一切对loop的修改都要通过发送一个信号到这个队列，才能同步到running loop的那个线程中，从而产生效果。

发送信号代码：loop._csock.send(b'\0')

下面给个示例，自己去体会

import asyncio

import threading

def start_loop(loop):

    asyncio.set_event_loop(loop)

    loop.run_forever()

    print(asyncio.Task.all_tasks().pop().result())

async def do_some_work(name):

    for i in range(5):

        print(f"{name}: is working")

        await asyncio.sleep(1)

    return True

loop = asyncio.new_event_loop()

threading.Thread(target=start_loop, args=(loop,)).start()

task = loop.create_task(do_some_work("Lili"))

loop._csock.send(b'\0')

while True:

    if task.done():

        loop.stop()

        loop._csock.send(b'\0')

        break

　　

相关文章