之前都是手动切换有规律,为了更好的使用协程来完成多任务,Python中greelet模块对其进行封装,从而让切换任务更加简单;
安装:
pip3 install greelet
直接看一下greelet的使用
- g1 = greenlet(fun) #开启协程
- g1.switch() #切换的开关
注意:需要从greenlet模块中导入greenlet类,名字相同
from greenlet import greenlet
def work1():
for i in range(5):
print('work1---{}'.format(i))
def work2():
for i in range(5):
print('work2---{}'.format(i))
def work3():
for i in range(5):
print('work3---{}'.format(i))
g1 = greenlet(work1)
g2 = greenlet(work2)
g3 = greenlet(work3)
g1.switch()
g2.switch()
g3.switch()
运行结果:
work1---0
work1---1
work1---2
work1---3
work1---4
work2---0
work2---1
work2---2
work2---3
work2---4
work3---0
work3---1
work3---2
work3---3
work3---4
但是发现是一个一个执行,那么如何交替执行呢,只需要互相调用切换开关即可:
from greenlet import greenlet
def work1():
for i in range(5):
g2.switch() #调用切换
print('work1---{}'.format(i))
def work2():
for i in range(5):
g1.switch()
print('work2---{}'.format(i))
g1 = greenlet(work1)
g2 = greenlet(work2)
g1.switch()
运行结果:
work1---0
work2---0
work1---1
work2---1
work1---2
work2---2
work1---3
work2---3
work1---4
work2---4
但是实际还是switch()手动切换和生成器类似,如何进行自动切换呢?
下一节看一下自动切换;