增强型的 yield 大致模拟了下 StacklessPython 的 api

 

主要是因为相对 StacklessPython 来说 yield 有这么几个限制：

函数只能通过 yield 来挂起，这导致实现 channel 的时候只能通过 yield command(c.send, value)，value = yield command(c.receive) 这样的语法来使当前 tasklet 挂起 (一定条件下) ，不像 stacklesspython 直接 some_channel.send(...) 就有可能挂起当前 tasklet 。 yield 只能向上一层，而不能直接 yield 到调用栈的最上层（或者甚至是指定 yield 到哪一层!!），这导致我们的 tasklet 只能在函数调用的第一层进行 yield 才能将执行权切换给调度程序，像下面:

def a_task():
    a_func()
def a_func():
    # 在这个函数里是没用办法将执行权切换给调度程序的。
不像 StacklessPython ，随便哪里调用 stackless.schedule() 都可以把执行权交出去。

本来还想把 evolution 也转过来的，发现 pygame 还没为 windows 编译 python2.5 的版本，自己编译太麻烦，还是等 python2.5 正式发布了再说吧。

#coding:utf-8
from collections import deque
'''
http://www.stackless.com/
'''
debuglevel = 0
readys = deque() # 就绪队列
class command(object):
    def __init__(self, func, *args, **kw):
        self.func = func
        self.args = args
        self.kw = kw

    def __call__(self, task):
        return self.func(task, *self.args, **self.kw)
class channel(object):
    ''' http://www.stackless.com/wiki/Channels '''
    def __init__(self):
        self.senders = deque() # 发送队列 元素：(tasklet, obj)
        self.receivers = deque() # 接收队列 元素：tasklet

    def send(self, sender, obj):
        '''向channel中发送数据，如果没用接受者，则让该tasklet等待'''
        if debuglevel:
            print 'tasklet:',sender,'send data:',obj,';receivers:',len(self.receivers)
        if self.receivers:
            receiver = self.receivers.popleft()
            ready(sender, None)
            run(receiver, obj)
        else:
            self.senders.append( (sender, obj) )
    def receive(self, receiver):
        ''' 从channel中接收数据，如果没用发送者，则让该tasklet等待'''
        if debuglevel:
            print 'tasklet:',receiver,'receive data ;senders:',len(self.senders)
        if self.senders:
            sender, obj = self.senders.popleft()
            ready(receiver,obj)
            run(sender, None)
        else:
            self.receivers.append(receiver)
class tasklet(object):
    '''http://www.stackless.com/wiki/Tasklets'''
    def __init__(self, func):
        self.func = func

    def __call__(self, *arg, **kw):
        '''将 genarator 加到就绪队列'''
        ready(self.func(*arg, **kw), None)
def run(task, obj):
    ''' 执行task '''
    try:
        result = task.send(obj)
    except StopIteration:
        pass
    else:
        if isinstance(result, command):
            # 给tasklet以执行 "系统命令" 的机会
            # 用户代码: yield command(func, ... )
            result(task)
        else:
            ready(task, None) # 加到就绪队列队尾，等待调度执行
def ready(task, obj):
    ''' 加入就绪队列 等待调度 '''
    readys.append( (task, obj) )
def schedule():
    ''' 调度就绪队列中的 tasklet '''
    while readys:
        task, obj = readys.popleft()
        run(task, obj)
# ============ test ===========
def simple_task(a):
    while True:
        print a
        yield None
def receiver(id, c):
    value = yield command(c.receive)
    print id,'receive', value
    while value:
        value = yield command(c.receive)
        print id,'received', value
def sender(id, c):
    value = 20
    while value:
        print id,'send',value
        yield command(c.send, value)
        value -= 1
def test_simple():
    tasklet(simple_task)(0)
    tasklet(simple_task)(10)
    tasklet(simple_task)(100)
def test_channel():
    c = channel()
    tasklet(receiver)('receiver1', c)
    tasklet(receiver)('receiver2', c)
    tasklet(sender)('sender1', c)
    c1 = channel()
    tasklet(receiver)('receiver3', c1)
    tasklet(sender)('sender2', c1)
if __name__=='__main__':
    #test_simple()
    test_channel()
    schedule()