基础协程实现方法
协程的实现主要靠的是yield关键字,yield的作用
挂起当前函数,将yield后面的值当做返回给调用生成器的地方;能够在唤醒生成器函数的时候,回复代码继续紧接着从上次执行的地方执行(可以接受额外的参数)
def func1(n):
for i in range(n):
print('func1:before yield,i=%d' % i)
x = yield
print('func1:after yield,i=%d,x=%s' % (i,x))
def func2(n):
for i in range(n):
print('func2:before yield,i=%d' % i)
x = yield
print('func2:after yield,i=%d,x=%s' % (i,x))
a = func1(5)
print(type(a))
==><class 'generator'>
很明显,协程的实现实际上是用的生成器。
a.send(1)
==>TypeError: can't send non-None value to a just-started generator
next(a)
==>func1:before yield,i=0
a.send(1)
==>func1:after yield,i=0,x=1
==>func1:before yield,i=1
next(a)
==>func1:after yield,i=1,x=None
==>func1:before yield,i=2
所以,必须先使用next或者send(None)启动生成器。
之后可以继续使用next不断让生成器继续执行或者使用send()继续执行(send可以发送参数)。
当然,我们在a执行几次后改为执行b是可以的,只要控制好谁去send()。
a = func1(5)
b = func2(5)
next(a)
a.send(1)
next(b)
a.send(2)
b.send(3)
==========================================
func1:before yield,i=0
func1:after yield,i=0,x=1
func1:before yield,i=1
func2:before yield,i=0
func1:after yield,i=1,x=2
func1:before yield,i=2
func2:after yield,i=0,x=3
func2:before yield,i=1
使用greenlet库
pip3 install greenlet