8.21 實(shí)現(xiàn)訪問(wèn)者模式

問(wèn)題

你要處理由大量不同類(lèi)型的對(duì)象組成的復(fù)雜數(shù)據(jù)結(jié)構(gòu)，每一個(gè)對(duì)象都需要需要進(jìn)行不同的處理。比如，遍歷一個(gè)樹(shù)形結(jié)構(gòu)，然后根據(jù)每個(gè)節(jié)點(diǎn)的相應(yīng)狀態(tài)執(zhí)行不同的操作。

解決方案

這里遇到的問(wèn)題在編程領(lǐng)域中是很普遍的，有時(shí)候會(huì)構(gòu)建一個(gè)由大量不同對(duì)象組成的數(shù)據(jù)結(jié)構(gòu)。假設(shè)你要寫(xiě)一個(gè)表示數(shù)學(xué)表達(dá)式的程序，那么你可能需要定義如下的類(lèi)：

class Node:
    pass

class UnaryOperator(Node):
    def __init__(self, operand):
        self.operand = operand

class BinaryOperator(Node):
    def __init__(self, left, right):
        self.left = left
        self.right = right

class Add(BinaryOperator):
    pass

class Sub(BinaryOperator):
    pass

class Mul(BinaryOperator):
    pass

class Div(BinaryOperator):
    pass

class Negate(UnaryOperator):
    pass

class Number(Node):
    def __init__(self, value):
        self.value = value

然后利用這些類(lèi)構(gòu)建嵌套數(shù)據(jù)結(jié)構(gòu)，如下所示：

# Representation of 1 + 2 * (3 - 4) / 5
t1 = Sub(Number(3), Number(4))
t2 = Mul(Number(2), t1)
t3 = Div(t2, Number(5))
t4 = Add(Number(1), t3)

這樣做的問(wèn)題是對(duì)于每個(gè)表達(dá)式，每次都要重新定義一遍，有沒(méi)有一種更通用的方式讓它支持所有的數(shù)字和操作符呢。這里我們使用訪問(wèn)者模式可以達(dá)到這樣的目的：

class NodeVisitor:
    def visit(self, node):
        methname = 'visit_' + type(node).__name__
        meth = getattr(self, methname, None)
        if meth is None:
            meth = self.generic_visit
        return meth(node)

    def generic_visit(self, node):
        raise RuntimeError('No {} method'.format('visit_' + type(node).__name__))

為了使用這個(gè)類(lèi)，可以定義一個(gè)類(lèi)繼承它并且實(shí)現(xiàn)各種 visit_Name() 方法，其中Name是node類(lèi)型。例如，如果你想求表達(dá)式的值，可以這樣寫(xiě)：

class Evaluator(NodeVisitor):
    def visit_Number(self, node):
        return node.value

    def visit_Add(self, node):
        return self.visit(node.left) + self.visit(node.right)

    def visit_Sub(self, node):
        return self.visit(node.left) - self.visit(node.right)

    def visit_Mul(self, node):
        return self.visit(node.left) * self.visit(node.right)

    def visit_Div(self, node):
        return self.visit(node.left) / self.visit(node.right)

    def visit_Negate(self, node):
        return -node.operand

使用示例：

>>> e = Evaluator()
>>> e.visit(t4)
0.6
>>>

作為一個(gè)不同的例子，下面定義一個(gè)類(lèi)在一個(gè)棧上面將一個(gè)表達(dá)式轉(zhuǎn)換成多個(gè)操作序列：

class StackCode(NodeVisitor):
    def generate_code(self, node):
        self.instructions = []
        self.visit(node)
        return self.instructions

    def visit_Number(self, node):
        self.instructions.append(('PUSH', node.value))

    def binop(self, node, instruction):
        self.visit(node.left)
        self.visit(node.right)
        self.instructions.append((instruction,))

    def visit_Add(self, node):
        self.binop(node, 'ADD')

    def visit_Sub(self, node):
        self.binop(node, 'SUB')

    def visit_Mul(self, node):
        self.binop(node, 'MUL')

    def visit_Div(self, node):
        self.binop(node, 'DIV')

    def unaryop(self, node, instruction):
        self.visit(node.operand)
        self.instructions.append((instruction,))

    def visit_Negate(self, node):
        self.unaryop(node, 'NEG')

使用示例：

>>> s = StackCode()
>>> s.generate_code(t4)
[('PUSH', 1), ('PUSH', 2), ('PUSH', 3), ('PUSH', 4), ('SUB',),
('MUL',), ('PUSH', 5), ('DIV',), ('ADD',)]
>>>

討論

剛開(kāi)始的時(shí)候你可能會(huì)寫(xiě)大量的if/else語(yǔ)句來(lái)實(shí)現(xiàn)，這里訪問(wèn)者模式的好處就是通過(guò) getattr() 來(lái)獲取相應(yīng)的方法，并利用遞歸來(lái)遍歷所有的節(jié)點(diǎn)：

def binop(self, node, instruction):
    self.visit(node.left)
    self.visit(node.right)
    self.instructions.append((instruction,))

還有一點(diǎn)需要指出的是，這種技術(shù)也是實(shí)現(xiàn)其他語(yǔ)言中switch或case語(yǔ)句的方式。比如，如果你正在寫(xiě)一個(gè)HTTP框架，你可能會(huì)寫(xiě)這樣一個(gè)請(qǐng)求分發(fā)的控制器：

class HTTPHandler:
    def handle(self, request):
        methname = 'do_' + request.request_method
        getattr(self, methname)(request)
    def do_GET(self, request):
        pass
    def do_POST(self, request):
        pass
    def do_HEAD(self, request):
        pass

訪問(wèn)者模式一個(gè)缺點(diǎn)就是它嚴(yán)重依賴(lài)遞歸，如果數(shù)據(jù)結(jié)構(gòu)嵌套層次太深可能會(huì)有問(wèn)題，有時(shí)候會(huì)超過(guò)Python的遞歸深度限制(參考 sys.getrecursionlimit() )。

可以參照8.22小節(jié)，利用生成器或迭代器來(lái)實(shí)現(xiàn)非遞歸遍歷算法。

在跟解析和編譯相關(guān)的編程中使用訪問(wèn)者模式是非常常見(jiàn)的。Python本身的 ast 模塊值的關(guān)注下，可以去看看源碼。9.24小節(jié)演示了一個(gè)利用 ast 模塊來(lái)處理Python源代碼的例子。