python编程(9):协程(coroutine)

时间:2021-06-04 20:07:59
def consumer():
r = ''
while True:#python的yield不但可以返回一个值,他还可以接收调用者的发出的参数
n = yield r #生成器,通过yield拿到消息(n),并把结果返回r
if not n:
return
print("[consumer] consuming %s..." %n)
r = '200 OK'#把结果返回r

def produce(c):
c.send(None)#启动生成器
n = 0
while n < 5:
n = n + 1
print("[produce]producing %s" %n)
r = c.send(n) #生产了东西,切换到consumer执行
print("[produce]consumer return:%s" %r) #produce拿到consumer的处理的结果,继续生产下一条消息
c.close() #produce决定不生产了,通过c.close关闭consumer,整个过程结束

c = consumer()
produce(c)

#协程
#子程序就是协程的一种特列

python编程(9):协程(coroutine)

注意到consumer函数是一个generator,把一个consumer传入produce后:

    首先调用c.send(None)启动生成器;

    然后,一旦生产了东西,通过c.send(n)切换到consumer执行;

    consumer通过yield拿到消息,处理,又通过yield把结果传回;

    produce拿到consumer处理的结果,继续生产下一条消息;

    produce决定不生产了,通过c.close()关闭consumer,整个过程结束。

整个流程无锁,由一个线程执行,produce和consumer协作完成任务,所以称为“协程”,而非线程的抢占式多任务