利用yield的协程机制我们可以轻松地实现生产者-消费者模式:
#-*- coding:utf-8 -*-
def consumer():
while True:
d=yield
if not d:
break
print "consumer:",d
c=consumer() #创建消费者
c.send(None) #启动消费者
c.send(1) #生产数据,并提交给消费者,运行结果:consumer:1
c.send(2)
c.send(None) #生产结束,通知消费者结束,抛出StopIteration异常。
协程执行流程:
1.创建协程对象(也就是消费者)后,必须使用send(None)或next()启动;
2.协程在执行yield后让出执行绪,等待消息;
3.调用方发送send(1)消息,协程恢复执行,将接收到的数据保存到d,执行后续流程;
4.再次循环到yield,协程返回前面的处理结果,并再次让出执行绪;
5.直到关闭或被引发异常。