设计模式 - 21.访问者模式 [行为型模式]

将作用于某种数据结构中的各元素的操作分离出来封装成独立的类，使其在不改变数据结构的前提下可以添加作用于这些元素的新的操作，为数据结构中的每个元素提供多种访问方式。它将对数据的操作与数据结构进行分离，是行为类模式中最复杂的一种模式。

import abc

class Container:
    def __init__(self):
        self.pool = []

    def add(self, obj):
        self.pool.append(obj)

    def remove(self, obj):
        if obj in self.pool:
            self.pool.remove(obj)

    def deal(self, cls:'ICls'):
        if cls.__class__ == ClsA:
            for i,k in enumerate(self.pool):
                cls.deal1(i,k)
        elif cls.__class__ == ClsB:
            for i,k in enumerate(self.pool):
                cls.deal2(i,k)

class ICls(metaclass=abc.ABCMeta):
    
    def __init__(self, type):
        self.type = type

    # @abc.abstractmethod
    def deal1(self, id, obj):
        pass

    # @abc.abstractmethod
    def deal2(self, id, obj):
        pass

class ClsA(ICls):
    def deal1(self, id, obj):
        print('cls%d'%self.type, obj)

class ClsB(ICls):
    def deal2(self, id, obj):
        print('cls%d'%self.type, [id, obj])



if __name__ == '__main__':
    a = ClsA(1)
    b = ClsB(2)
    container = Container()
    container.add({'a':123})
    container.add({'color':0xff00ff})
    container.deal(a)
    print('====')
    container.deal(b)

输出

cls1 {'a': 123}
cls1 {'color': 16711935}
====
cls2 [0, {'a': 123}]
cls2 [1, {'color': 16711935}]

如果语言支持多态, 条件可以改为多态