从头造轮子：python3 asyncio 之 run

一、知识准备

● 相对于 run_until_complete ，改动并不大，就是将入口函数重新封装了一下，基础知识主要还是 run_until_complete 的内容

● asyncio.run是Python3.7之后新增的入口函数

二、环境准备

三、 run 的实现

先来看下官方asyncio的使用方法：

|># more main.py import asyncio async def hello(): print('enter hello ...') return 'world' if __name__ == "__main__":
    rst = asyncio.run(hello())
    print(rst)
    
|># python3 main.py enter hello ... return world ...

来看下造的轮子的使用方式：

▶ more main.py from wilsonasyncio import run async def hello(): print('enter hello ...') return 'return world ...' if __name__ == "__main__":
    ret = run(hello())
    print(ret)

    
▶ python3 main.py
enter hello ... return world ...

自己造的轮子也很好的运行了，下面我们来看下轮子的代码

四、代码解析

轮子代码

1）代码组成

|># tree
.
├── eventloops.py ├── futures.py ├── main.py ├── tasks.py ├── wilsonasyncio.py

2）代码概览：

eventloops.py

tasks.py

futures.py

3）执行过程

3.1）入口函数

main.py

async def hello(): print('enter hello ...') return 'return world ...' if __name__ == "__main__":
    ret = run(hello())
    print(ret)

• ret = run(hello()) 直接调用 run ，参数是用户函数 hello() ，我们看下run的源码

def run(main): loop = get_event_loop() return loop.run_until_complete(main)

• loop = get_event_loop() 获取事件循环
• return loop.run_until_complete(main) 调用 run_until_complete

3.2）事件循环启动

def run_until_complete(self, future): future = tasks.ensure_future(future, loop=self)
        future.add_done_callback(_complete_eventloop, future)
        self.run_forever() return future.result()

• 与之前略有不同， future = tasks.ensure_future(future, loop=self) ，调用了 tasks.ensure_future

def ensure_future(coro_or_future, *, loop=None): if isinstance(coro_or_future, Future): return coro_or_future else: return loop.create_task(coro_or_future)

• 如果传入的对象是一个普通函数，那就封装成一个task；如果已经是一个future对象，那就直接返回。这一步的目的主要是确保传入的对象，是一个Future类型

剩下的部分已经没有什么新鲜的了，和 run_until_complete 一样，我们直接跳过...

3.7）执行结果

▶ python3 main.py enter hello ... return world ...

五、流程总结

六、小结

● run 与 run_until_complete 大同小异，只不过入口函数做了一些调整，使得用户调用更加的便利