python-协程

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22

import time
import random

def producer():
    res = []
    for i in range(9999990):
        res.append(i)
    return res


def consumer(c):
    x = 0
    for i in c:
        x += i

    return x


start = time.time()
res = producer()
print(consumer(res))
print(time.time() - start)

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19

def producer(g):
    res = ""
    for i in range(9999990):
        res = g.send(i)
    return res


def consumer():
    num = 0
    while True:
        x = yield num  # x是send发送来的值
        num += x


start = time.time()
c1 = consumer()
next(c1)
print(producer(c1))
print(time.time() - start)

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

from greenlet import greenlet

def test1():
    print(12)
    gr2.switch()  # 切换到gr2 
    print(34)


def test2():
    print(56)
    gr1.switch()  # 切换到gr1
    print(78)


gr1 = greenlet(test1)
gr2 = greenlet(test2)
gr1.switch()  # 执行过程 test1[print(12)] --> test2[print(56)] --> test1[print(34)] 程序结束

# 下面的执行效率比yield要慢很多，不知道为什么
X = 0
NUM = 0

def producer(area):
    global X
    for i in range(area):  # 9999990
        X = i
        g2.switch()

def consumer():
    global NUM
    while True:
        NUM += X
        g1.switch()


# 不可以传函数参数
g1 = greenlet(producer)
g2 = greenlet(consumer)

start = time.time()
# 切换到g1运行,可以传参
g1.switch(9999990)
print(NUM)
print(time.time() - start)

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

import gevent
from gevent import monkey

# 默认不支持阻塞其他IO操作[默认支持gevent.sleep(3)],要支持其他IO操作需要打补丁。
monkey.patch_all()


def eat(name):
    print("%s eat 1 " % name)
    # gevent.sleep(3)
    time.sleep(2)
    print("%s eat 2 " % name)


def play(name):
    print("%s play 1 " % name)
    time.sleep(3)
    print("%s play 2 " % name)


# 创建协程任务，gevent遇到阻塞就会切换执行协程任务
g1 = gevent.spawn(eat, "soulchild")
g2 = gevent.spawn(play, "soulchild")
print("ok")
for i in range(5):
    print(i)
    # 模拟IO阻塞,这时g1和g2会被调用
    time.sleep(0.1)
# 模拟IO阻塞,这时g1和g2会被调用
gevent.joinall([g1, g2])