摘要：標(biāo)準(zhǔn)的異常處理是這樣的這段代碼會(huì)打印出而不會(huì)打印出，因?yàn)楫惓?huì)中斷當(dāng)前流程，跳轉(zhuǎn)到部分去繼續(xù)執(zhí)行。這種行為類似里的。如何實(shí)現(xiàn)的其實(shí)原理上很簡(jiǎn)單。的時(shí)候把當(dāng)前協(xié)程的狀態(tài)保存起來，如果決定要，就把協(xié)程的時(shí)刻的狀態(tài)重新恢復(fù)然后從那個(gè)點(diǎn)繼續(xù)執(zhí)行。

標(biāo)準(zhǔn)的異常處理是這樣的

try:
    print("hello")
    raise Exception()
    print("!!!")
except:
    print("world")
    print("???")

這段代碼會(huì)打印出???而不會(huì)打印出!!!，因?yàn)楫惓?huì)中斷當(dāng)前流程，跳轉(zhuǎn)到except部分去繼續(xù)執(zhí)行。但是有的時(shí)候我們希望的這樣的行為：

try:
    print("hello")
    print(Scheduler.interrupt())
    print("!!!")
except ProcessInterrupt as pi:
    pi.resume("world")
    print("???")

這段代碼打印出!!!而不是???，因?yàn)閞esume的時(shí)候把執(zhí)行重新跳轉(zhuǎn)回interrupt的地方了。這種行為類似vba里的on error resume next（https://msdn.microsoft.com/en-us/library/5hsw66as.aspx）。

如何實(shí)現(xiàn)的？其實(shí)原理上很簡(jiǎn)單。interrupt的時(shí)候把當(dāng)前協(xié)程的狀態(tài)保存起來（pickle.dumps），如果決定要resume，就把協(xié)程interrupt的時(shí)刻的狀態(tài)重新恢復(fù)（pickle.loads）然后從那個(gè)點(diǎn)繼續(xù)執(zhí)行。

完整的代碼（需要pypy或者stackless python）：

import greenlet
import cPickle as pickle
import traceback
import threading
import functools


class ProcessInterrupt(Exception):
    def __init__(self, interruption_point, pi_args):
        self.interruption_point = interruption_point
        self.stacktrace = traceback.extract_stack()
        self.pi_args = pi_args

    def resume(self, resume_with=None):
        Scheduler.resume(self.interruption_point, resume_with)

    def __repr__(self):
        return ">>>ProcessInterrupt>>>%s" % repr(self.stacktrace)

    def __str__(self):
        return repr(self)

    def __unicode__(self):
        return repr(self)


class Scheduler(object):
    current = threading.local()

    def __init__(self):

        if getattr(self.current, "instance", None):
            raise Exception("can not have two scheduler in one thread")
        self.scheduler_greenlet = greenlet.getcurrent()
        self.current.instance = self

    def __call__(self, action, action_args):
        next = action, action_args
        while next:
            action, action_args = next
            if "init" == action:
                next = action_args["init_greenlet"].switch()
            elif "interrupt" == action:
                interruption_point = pickle.dumps(action_args["switched_from"])
                should_resume, resume_with = False, None
                next = action_args["switched_from"].switch(
                    should_resume, resume_with, interruption_point)
            elif "resume" == action:
                should_resume, resume_with, interruption_point = True, action_args["resume_with"], action_args[
                    "interruption_point"]
                next = pickle.loads(action_args["interruption_point"]).switch(
                    should_resume, resume_with, interruption_point)
            else:
                raise NotImplementedError("unknown action: %s" % action)

    @classmethod
    def start(cls, init_func, *args, **kwargs):
        scheduler = Scheduler()
        init_greenlet = greenlet.greenlet(functools.partial(init_func, *args, **kwargs))
        scheduler("init", {
            "init_greenlet": init_greenlet,
        })

    @classmethod
    def interrupt(cls, pi_args=None):
        should_resume, resume_with, interruption_point = cls.switch_to_scheduler("interrupt", {
            "switched_from": greenlet.getcurrent()
        })
        if should_resume:
            return resume_with
        else:
            pi = ProcessInterrupt(interruption_point, pi_args)
            raise pi

    @classmethod
    def resume(cls, interruption_point, resume_with=None):
        cls.switch_to_scheduler("resume", {
            "interruption_point": interruption_point,
            "resume_with": resume_with
        })

    @classmethod
    def switch_to_scheduler(cls, *args, **kwargs):
        return cls.current.instance.scheduler_greenlet.switch(*args, **kwargs)


if "__main__" == __name__:
    def init():
        try:
            print("hello")
            print(Scheduler.interrupt())
            print("!!!")
        except ProcessInterrupt as pi:
            pi.resume("world")
            print("???")

        try:
            print("hello")
            raise Exception()
            print("!!!")
        except:
            print("world")
            print("???")

    Scheduler.start(init)