Python的协程很像生成器,但并不是生成数据,大多数时候扮演了数据消费者的作用。换句话说,协程是一个在每次使用send方法发送数据后就会被唤醒的函数。
协程的要点是将“yield”关键字写在表达式的右边。下面是一个打印出所发送的值的协程例子:
def coroutine():
print('My coroutine')
while True:
val = yield
print('Got', val)
>>> co = coroutine()
>>> next(co)
My coroutine
>>> co.send(1)
Got 1
>>> co.send(2)
Got 2
>>> co.send(3)
Got 3
首先要调用“next”来将协程推进。你可以看到它执行了一个打印。最终,函数进行到“yield”表达式了,然后就会等待唤醒。之后,每次有值被发送过来,协程就会从“yield”处唤醒,将值复制给val并打印出它。
使用close()
方法可以关闭这个协程。
>>> co.close()
>>> co.send(4)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
StopIteration
- 创建一个名为“square”的协程,它会打印出所发送的值的平方。
- 创建一个名为“minimize”的协程,它会保存并打印出所发送的值中最小的一个值。
协程可以用在部署数据管道中,其中一个协程会发送数据到下一个在数据管道中的协程。协程使用send()
方法来将数据压入管道。
下面的例子实现了一个小型管道,值会从生产者协程发送到消费者协程打印出来:
def producer(consumer):
print("Producer ready")
while True:
val = yield
consumer.send(val * val)
def consumer():
print("Consumer ready")
while True:
val = yield
print('Consumer got', val)
如前所述,在发送任何数据前都别忘了调用一次“next”。
>>> cons = consumer()
>>> prod = producer(cons)
>>> next(prod)
Producer ready
>>> next(cons)
Consumer ready
>>> prod.send(1)
Consumer got 1
>>> prod.send(2)
Consumer got 4
>>> prod.send(3)
Consumer got 9
同样的,对于协程来说,数据可以被发送到不同的地方。下面的例子部署了两个消费者,第一个它只会打印0-10之间的内容,第二个则是10-20。
def producer(consumers):
print("Producer ready")
try:
while True:
val = yield
for consumer in consumers:
consumer.send(val * val)
except GeneratorExit:
for consumer in consumers:
consumer.close()
def consumer(name, low, high):
print("%s ready" % name)
try:
while True:
val = yield
if low < val < high:
print('%s got' % name, val)
except GeneratorExit:
print("%s closed" % name)
不要忘了调用下“next”呦。
>>> con1 = consumer('Consumer 1', 00, 10)
>>> con2 = consumer('Consumer 2', 10, 20)
>>> prod = producer([con1, con2])
>>> next(prod)
Producer ready
>>> next(con1)
Consumer 1 ready
>>> next(con2)
Consumer 2 ready
>>> prod.send(1)
Consumer 1 got 1
>>> prod.send(2)
Consumer 1 got 4
>>> prod.send(3)
Consumer 1 got 9
>>> prod.send(4)
Consumer 2 got 16
>>> prod.close()
Consumer 1 closed
Consumer 2 closed
数据被发送到所有的消费者中,但只有第二个执行了打印命令。注意这里使用的“GeneratorExit”异常。一般都会用捕获异常的方式来通知下游协程这个管道没有用了,赶紧关闭了吧。
-
部署一个生产者-消费者管道,生产者会将值的平方发送给两个消费者。其中一个会储存并打印出所发送的值中最小的一个,另一个则是最大的一个。
-
部署一个生产者-消费者管道,生产者会将值的平方发送给两个消费者,一次只发送给其中的一个。第一个值会发送给消费者1号,第二个则会发送给消费者2号,第三个又发送给消费者1号.... 关闭生产者时会强制让消费者打印出它们所接收到的值的列表。