Python——使用asyncio包处理并发

使用 asyncio 包处理并发

asyncio包：使用事件循环驱动的协程实现并发。

线程与协程的对比

'\ thinking' 旋转等待效果

In [1]: import threading
 
In [2]: import itertools
 
In [3]: import time,sys
 
In [4]: class Signal:  # 定义一个简单的可变对象；go 属性 从外部控制线程
   ...:     go = True
 
In [5]: def spin(msg,signal):
   ...:     w,flush = sys.stdout.write,sys.stdout.flush
   ...:     for char in itertools.cycle('|/-\\'):  # 从序列中反复不断的生成元素
   ...:         status = char + ' ' + msg
   ...:         w(status)
   ...:         flush()
   ...:         w('\x08' * len(status))  # 退格键：\x08 文本动画的诀窍所在
   ...:         time.sleep(1)
   ...:         if not signal.go:
   ...:             break
   ...:     w(' ' * len(status) + '\x08' * len(status))
 
In [6]: def slow():
   ...:     time.sleep(3)
   ...:     return 42
 
In [9]: def super():
   ...:     signal = Signal()
   ...:     sp = threading.Thread(target=spin,args=('thinking',signal))
   ...:     print('============')
   ...:     sp.start()
   ...:     res = slow()
   ...:     signal.go = False
   ...:     sp.join()
   ...:     return res

注意：Python 没有提供终止线程的 API ，这是有意为之的。若想关闭线程，必须给线程发送消息。这里用的是 signal.go 属性。干净的规则的退出。

适合 asyncio API 的协程：

1 定义体必须使用 yield from ，而不能使用 yield

2 协程要由调用方驱动，并由调用方通过 yield from 调用

3 或者把协程传给 asyncio 包中的某个函数，比如 asyncio.async()

4 @asyncio.coroutine 装饰器应该用在协程上

asyncio 实现 动画效果

In [1]: import asyncio
 
In [3]: import itertools
 
In [4]: import sys
 
# 交给 asyncio 处理的协程需要使用该装饰器装饰。这不是强制要求，但是强烈建议这么做。
In [5]: @asyncio.coroutine
   ...: def spin(msg):  # 不需要多线程的关闭参数
   ...:     w,flush = sys.stdout.write, sys.stdout.flush
   ...:     for char in itertools.cycle('|/-\\'):
   ...:         status = char + ' ' + msg
   ...:         w(status)
   ...:         flush()
   ...:         w('\x08' * len(status))
   ...:         try:
   ...:             yield from asyncio.sleep(.1)  # 不会阻塞事件循环
                # spin 函数苏醒后，取消请求 异常，退出循环
   ...:         except asyncio.CancelledError:
   ...:             break
   ...:     write(' ' * len(status) + '\x08' * len(status))
   ...:
 
In [6]: @asyncio.coroutine
   ...: def slow():
            # 把控制权交给主循环，休眠结束后，结束这个协程
   ...:     yield from asyncio.sleep(3)
   ...:     return 42
   ...:
 
In [9]: @asyncio.coroutine
   ...: def sup():
            # asyncio 排定spin协程的运行时间，封装成一个 Task对象 sp
   ...:     sp = asyncio.async(spin('thinking!'))
   ...:     print('spin obj:',sp)
            # sup 也是协程，因此，可以使用 yield from 驱动 slow()
   ...:     res = yield from slow()
            sp.cancel()
   ...:     return res
   ...:
 
In [10]: def main():
    ...:     loop = asyncio.get_event_loop()
    ...:     res = loop.run_until_complete(sup())
    ...:     loop.close()
    ...:     print('answer:',res)
    ...:
 
In [11]: main()
D:\python36\Scripts\ipython:3: DeprecationWarning: asyncio.async() function is deprecated, use ensure_future()
spin obj: <Task pending coro=<spin() running at <ipython-input-5-0304845f34e1>:1>>
answer: 42!

除非想阻塞主线程，从而冻结事件循环或整个应用，否则不要在 asyncio 协程中使用 time.sleep() 。如果协程需要一段时间内什么也不做，应该使用 yield from asyncio.sleep() 。

@asyncio.coroutine 装饰器不是强制要求，但是强烈建议这么做，因为这样能

1 把协程凸显出来，有助于调试。

2 如果还未产出值，协程就被垃圾回收了（意味着有操作未完成，因此有可能是个缺陷），那就可以发出警告了。

3 这个装饰器不会预激协程。