Python使用asyncio包处理并发的实现代码
程序员老华 人气:0使用 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 这个装饰器不会预激协程。
加载全部内容