import time
# python生成器的例子(协程)
def consumer():
r = ''
while True:
n = yield r
if not n:
return
print('[消费者] Consuming %s...' % n)
time.sleep(1)
r = '200 OK'
def produce(c):
# 启动生成器
c.__next__()
n = 0
while n < 3:
n = n + 1
print('[生产者] Producing %s...' % n)
r = c.send(n)
print('[生产者] Consumer return: %s' % r)
c.close()
if __name__ == '__main__':
produce(consumer())
执行结果如下:
[生产者] Producing 1...
[消费者] Consuming 1...
[生产者] Consumer return: 200 OK
[生产者] Producing 2...
[消费者] Consuming 2...
[生产者] Consumer return: 200 OK
[生产者] Producing 3...
[消费者] Consuming 3...
[生产者] Consumer return: 200 OK
注意到consumer函数是一个generator(生成器),把一个consumer传入produce后:
首先调用c.__next__()启动生成器;
然后,一旦生产了东西,通过c.send(n) 切换到 consumer执行;
consumer 通过 yield 拿到消息,处理,又通过yield把结果传回;
produce 拿到 consumer 处理的结果,继续生产下一条消息;
produce决 定不生产了,通过c.close()关闭consumer,整个过程结束。
整个流程无锁,由一个线程执行,produce和consumer协作完成任务,所以称为“协程”,而非线程的抢占式多任务。一句话总结协程的特点:“子程序就是协程的一种特例。”