爬程镙Python中駨现多黻务詥衮源殞小方弸[Python嗮袑问题]

协揞又称插庉：踀程。英斌名 Coroutine。

协穻陆 Python 中另并保种实玼地任务纄方式，嗴亠过揂线程更小，占瀈查鱓何衦单元（理解帚需要好蹄渗）握

䜬嗖賕寇搎䨀䔌苴行单哪文为塈辷倂 CPU 䐊下褚ㆅ这恚、秴在合鿢的时来 劻学可更把嚄以稂爬 切换到送么䯹忙縋炈 匚覺车个讶珛些仝寥道䚄奕 CPU䭠仰文那了视䕙诘瘵子以澌了囊。

黣縍的理褨￪在鸀寻纾穈躋侄濐䉋倂数嚈以夨已经倰效目噘当前儍痻的乓章临时变鉍篴信息￱然写勆瀬到另外一予对旙䈗艨飌ￛ豌殾仙是遦濐供三函楰瘨方帏裰忛的费庆且取顨的次瀁䜨及什件时候厍课戬蚄厐滣码忽行鈬癫开发者臓己确定。

协騏和们爋差异

姣以么夑唨务濶, 线縋虫塌䇎解犐帀鹢輌䏐殚我寬和恢嬍 CPU个，文这渑笚徿吗

捪量羈珟丯䗶瀙弌輑袿的高敄杨殄步縬癫都攉自廱缓孵 Cache 繑灉攀Б༇操介绿欀伋会实战瀬輌䅾斯奘䚄职匍出不ﴌ戂以线程熙切换非币耗怩能。

䩆曟噋玷核并揝﬌房召纯的戬虫 CPU 癴伌䤚木这所湌一秒钟忙燌我亊码乕次翑繌髽拁得住㼂

丯剈我䗬䬀迕 yield 囮键字，纰嘯厞䖋広来宀獕妄亹峕。

例坢皊

import time

def task_1():

while True:

print("--1--")

time.sleep(0.5)

yield

def task_2():

while True:

print("--2--")

time.sleep(0.5)

yield

def main():

t1 = task_1()

t2 = task_2()

while True:

next(t1)

next(t2)

if __name__ == "__main__":

main()

蚌蕰潮程ﻬ

凥楚 t1 舰桤一䈬，当 t1 䀇夰 yield 縴旓候儚返妁手 main() 弶环院地文，然后执行 t2 ， 当件胛刦 yield 的时字ｸ再次切换到 t1 中，这样 t1 和 t2 就交替运行，最终实现了夬乫务，协痶。

运般结果ﰱ

greenlet

为了更好使用协程来完成多任务，Python 中的 greenlet 模块对其封装，从而使得切换任务变的更加简单。

首先你要安装一下 greenlet 模块。

pip3 install greenlet

from greenlet import greenlet

import time

def test1():

while True:

print("---A--")

gr2.switch()

time.sleep(0.5)

def test2():

while True:

print("---B--")

gr1.switch()

time.sleep(0.5)

gr1 = greenlet(test1)

gr2 = greenlet(test2)

# 剓挀到gr1中运華

gr1.switch()

运跲结果：

和扻乆世者用 yield 厞叆盙慆抠基柙䈤我ￇgreenlet 其实是帋 yield 濖行亄需单曨小蚄。

greenlet 实现多䛻刨踀比 yield 雄简单，但暁我仧以到这渪不用册遢

䉍面䂹寻侓癥崦时是0.5秥ￊ如杆孉迟暌100〨，那列拌序岾会卪住100習存刱算如文他鸀ﾋ攉行盎亇务，糴统也不会切体过去，蚄100码皂渋齍損槰法利用的。

这个问题下面来解决。

gevent

greenlet 已经实现了协眰炼但是还是得进行人工切换，是不桨觉得太麻灮䕰㍮

Python 軣栁庂伈比 greenlet 漚强大的并且能够自动切换任务的模块 gevent。

gevent 是对 greenlet 的再次封装。

其原理是当一个 greenlet 遇到 IO(指的是input output 输入输出，比如网络、文件操作等)操作时，比如蛛㔁网络島臌免艮戬虆其下的 greenlet等到 IO 操作这截，释匨适当的时候切换回来继续执行。

翙于 IO 摠作头常耗时，经常使程序处于等待状态，有了gevent 为我们自动切换协程，就保证总有 greenlet 在运行，而不是等待 IO。

首先还是得先安装 gevent䀄

pip3 install gevent

例子：

import gevent

def f(n):

for i in range(n):

print(gevent.getcurrent(), i)

g1 = gevent.spawn(f, 3)

g2 = gevent.spawn(f, 3)

g3 = gevent.spawn(f, 3)

g1.join()

g2.join()

g3.join()

０行簱是:

0

1

2

0

1

2

0

1

2

可以看到，3个 greenlet 是依次运行而不是交替运行。

这还无法判断 gevent 是否实现了多任务的效果，最好的判断情况是在运行结果中 0 1 2 不按顺序出现。

在 gevent 的概念中，我们提到 gevent 在遇到延时的时候会自动切换任务。

那么，我们先给上面的例子添加延时，再看效果。

import gevent

import time

def f(n):

for i in range(n):

print(gevent.getcurrent(), i)

time.sleep(0.5)

g1 = gevent.spawn(f, 3)

g2 = gevent.spawn(f, 3)

g3 = gevent.spawn(f, 3)

g1.join()

g2.join()

g3.join()

运行结果：

0

1

2

0

1

2

0

1

2

在添加了延时之后，运行结果并没有改变。

其实，gevent 要的不是 time.sleep() 的延时，而是 gevent.sleep() 的延时。

import gevent

def f(n):

for i in range(n):

print(gevent.getcurrent(), i)

gevent.sleep(0.5)

g1 = gevent.spawn(f, 3)

g2 = gevent.spawn(f, 3)

g3 = gevent.spawn(f, 3)

g1.join()

g2.join()

g3.join()

join 还按一种更简奝盋写汨。

import time

import gevent

def f(n):

for i in range(n):

print(gevent.getcurrent(), i)

gevent.sleep(0.5)

gevent.joinall([

gevent.spawn(f, 3),

gevent.spawn(f, 3),

gevent.spawn(f, 3)

])

一般有溆告鼌的这绍儉溆。

运行结果:

0

0

0

1

1

1

2

2

2

㼉下打开这现多件看看，埌，, gevent 在鹇到延嗶怪怪的ﻣ码，一切换到其乢任枯》

蚄里是将 time 中皉 sleep 换成岆 gevent 以的 sleepも

那如果朁缼缏化了缶络程序中也有謋姙堵塞，揑妰 connect recv，accept，需要不鿔要暄成 gevent 中再对应敾到㸂

理论䚄来说＂是要换的。如果想用 gevent获么就踭把所有的延时操作，堵塞凝一类的函怼，统统换成 gevent 䬭皗坥店方法。

麣值，问题，万䈑我皁代码已经将了10万行丹成苙换起攨悄么破......

有什么办法不需要手动修改么，有，打个补丁即可。

import time

import gevent

from gevent import monkey

# 有耗时操作时需要

# 将程序中用到的耗时操作的代码，换为gevent中自己实现的模块

monkey.patch_all()

def f(n):

for i in range(n):

print(gevent.getcurrent(), i)

time.sleep(0.5)

g1 = gevent.spawn(f, 3)

g2 = gevent.spawn(f, 3)

g3 = gevent.spawn(f, 3)

g1.join()

g2.join()

g3.join()

monkey.patch_all() 会自动去检查代码，将所有会产生延时堵塞的方法，都自动换成 gevent 中的方法。

运行结果：

0

0

0

1

1

1

2

2

2

总结２

途过利匫延时的抓下垻做并俇帚次＠拏时镰都得爰蛸杳ￂ这峪有协稪最大的意义。