下面将一个经典的消费者和生产者的案例进行分析:
import timedef consumer(): r = '' while True: n = yield r if not n: return print('[CONSUMER] Consuming %s...' % n) time.sleep(1) r = '200 OK'def produce(c): c.next() n = 0 while n < 5: n = n + 1 print('[PRODUCER] Producing %s...' % n) r = c.send(n) print('[PRODUCER] Consumer return: %s' % r) c.close()if __name__=='__main__': c = consumer() produce(c)
协程涉及函数 yield、send、next。堵塞、传递(发送)、继续
从main 开始看起,首先调用consumer(),将其返回结果给c,然后将c给produce()函数。
consumer() 函数中通过yield(阻塞),然后进行produce(c)[c代表consumer()的消息],通过c.next()恢复consumer()。将n + 1后c.send(n)发送给c,依此类推n=5时关闭。
官方解释如下:(http://www.liaoxuefeng.com/wiki/001374738125095c955c1e6d8bb493182103fac9270762a000/0013868328689835ecd883d910145dfa8227b539725e5ed000)
注意到consumer函数是一个generator(生成器),把一个consumer传入produce后:首先调用c.next()启动生成器;然后,一旦生产了东西,通过c.send(n)切换到consumer执行;consumer通过yield拿到消息,处理,又通过yield把结果传回;produce拿到consumer处理的结果,继续生产下一条消息;produce决定不生产了,通过c.close()关闭consumer,整个过程结束。整个流程无锁,由一个线程执行,produce和consumer协作完成任务,所以称为“协程”,而非线程的抢占式多任务。