在没有极端标识的情况下维护python中的可滚动代码流

Question

我遇到过一种情况,我正在处理一段代码,我命令对远程对象进行更改(这是我无法复制以克服一个克隆),然后请求远程对象进行某些操作.新状态并通过一系列相反的命令恢复我对它所做的所有更改.问题是,如果在所有这些更改的中间我遇到错误,我希望能够回滚到目前为止所做的所有更改.

我想到的最合适的解决方案是python try-finally工作流程,但是当命令序列很长时,这是相当有问题的:

try:
    # perform action
    try:
        # perform action
        try: 
            # ...
        finally:
            # unroll
    finally:
        # unroll
finally:
    # unroll

这样,我需要的命令越多,我的缩进和嵌套就越深入,代码的可读性就越少.我已经考虑过其他一些解决方案,比如维护一个堆栈,每个命令都会推送回滚动作,但这可能会变得相当复杂,我不喜欢将绑定方法推入堆栈.我还考虑为我执行的每个操作递增一个计数器然后在一个单独的最终根据计数器决定我想要的回滚类型,但同样,这些代码的可维护性变得很痛苦.

我搜索"交易"和"回滚"的大多数点击都与数据库相关,并且不太适合更通用的代码......任何人都对如何系统地压扁这种暴行有一个好主意？

Answer 1

Context Manager对象和with语句不会改善这种情况吗？特别是如果你可以使用一个Python版本,其中with语句支持多个上下文表达式,如2.7或3.x. 这是一个例子:

class Action(object):
    def __init__(self, count):
        self.count = count
    def perform(self):
        print "perform " + str(self.count)
        if self.count == 2:
            raise Exception("self.count is " + str(self.count))
    def commit(self):
        print "commit " + str(self.count)
    def rollback(self):
        print "rollback " + str(self.count)
    def __enter__(self):
        perform()
        return self
    def __exit__(self, exc_type, exc_value, traceback):
        if exc_value is None:
            self.commit()
        else:
            self.rollback()

with Action(1), Action(2), Action(3):
    pass

您必须将代码移动到一组"事务"类,Action如上所述,其中要执行的操作在__enter__()方法中执行,如果这通常终止,则可以保证__exit()__将调用相应的方法.

请注意,我的例子并不完全符合你的例子; 你必须调整在__enter__()方法中执行的内容以及在with语句体内执行的内容.在这种情况下,您可能希望使用以下语法:

with Action(1) as a1, Action(2) as a2:
    pass

能够Action从with语句的正文中访问对象.