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

class TaskScheduler:def __init__(self):self._task_queue = deque()def new_task(self, task):'''Admit a newly started task to the scheduler'''self._task_queue.append(task)def run(self):'''Run until there are no more tasks'''while self._task_queue:task = self._task_queue.popleft()try:# Run until the next yield statementnext(task)self._task_queue.append(task)except StopIteration:# Generator is no longer executingpass# Example use
sched = TaskScheduler()
sched.new_task(countdown(10))
sched.new_task(countdown(5))
sched.new_task(countup(15))
sched.run()

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

def consumer():r = ''while True:n = yield rif not n:returnprint('[CONSUMER] Consuming %s...' % n)r = '200 OK'def produce(c):c.send(None)n = 0while n < 5:n = n + 1print('[PRODUCER] Producing %s...' % n)r = c.send(n)print('[PRODUCER] Consumer return: %s' % r)c.close()c = consumer()
produce(c)

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
45
46
47
48
49
50
51
52
53
54

class ActorScheduler:def __init__(self):self._actors = {}self._msg_queue = deque()def new_actor(self, name, actor):self._msg_queue.append((actor, None))self._actors[name] = actordef send(self, name, msg):actor = self._actors.get(name)if actor:self._msg_queue.append((actor, msg))def run(self):while self._msg_queue:# print("队列：", self._msg_queue)actor, msg = self._msg_queue.popleft()# print("actor", actor)# print("msg", msg)try:actor.send(msg)except StopIteration:passif __name__ == '__main__':def say_hello():while True:msg = yieldprint("say hello", msg)def say_hi():while True:msg = yieldprint("say hi", msg)def counter(sched):while True:n = yieldprint("counter:", n)if n == 0:breaksched.send('say_hello', n)sched.send('say_hi', n)sched.send('counter', n-1)sched = ActorScheduler()# 创建初始化 actorssched.new_actor('say_hello', say_hello())sched.new_actor('say_hi', say_hi())sched.new_actor('counter', counter(sched))sched.send('counter', 10)sched.run()