上篇文章我们说过由于GIL锁的限制,导致Python不能充分利用多线程来实现高并发,在某些情况下使用多线程可能比单线程效率更低,所以Python中出现了协程。
协程(coroutine) 又称微线程,是一中轻量级的线程,它可以在函数的特定位置暂停或恢复,同时调用者可以从协程中获取状态或将状态传递给协程。进程和线程都是通过CPU的调度实现不同任务的有序执行,而协程是由用户程序自己控制调度的,也没有线程切换的开销,所以执行效率极高。
生成器方式实现
早先的协程是使用生成器关键字yield来实现的,和生成器很相似。下面利用一生产者-消费者模型来介绍如何使用yield协程来进行任务的切换。
import time
def consumer():
r = None
while True:
n = yield r
if not n:
return
print("[消费者] 消费'{}'...".format(n))
time.sleep(1)
r = '消费完成'
def producer(c):
next(c)
n = 0
while n < 5:
n = n + 1
print("[生产者] 生产'{}'...".format(n))
res = c.send(n)
print("[生产者] 消费者返回'{}'".format(res))
c.close()
if __name__ == '__main__':
c = consumer()
producer(c)
执行结果如下:
上述代码中的consumer函数是一个生成器,
- 把consumer传入producer函数,调用next(c)启动该生成器;
- 生产者producer函数生产了n,通过c.send(n)切换到consumer执行;
- consumer函数通过yield拿到生产者生产的消息n,处理后,又通过yield把结果r返回。
- producer拿到consumer处理的结果,继续生产下一条消息,直至生产结束,通过c.close()关闭consumer,整个过程结束。
这段代码全程在一个线程中完成,producer和co