策略模式详解-Toy模板网

这篇具有很好参考价值的文章主要介绍了策略模式详解。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

策略模式（Strategy Design Pattern），定义一堆算法类，并将每个算法分别封装起来，让它们可以互相替换，被封装起来的算法具有独立性外部不可改变其特性。

策略模式可以使算法的变化独立于使用它们的客户端（这里的客户端代指使用算法的代码）。

策略模式的应用场景

策略模式最常见的应用场景是，利用它来避免冗长的 if-else 或 switch 分支判断。不过，它的作用还不止如此。它也可以像模板模式那样，提供框架的扩展点等等。

当我们需要多个功能相似的类，并且需要它们之间可以灵活切换时，就非常适合使用策略模式。

策略模式的构成

策略类的定义比较简单，需要定义以下两种角色：

抽象策略接口类：定义了策略类需要实现的方法规范。
策略实现类：继承自抽象策略接口，为具体策略类。

当有多个策略时，可以通过简单工厂封装所有的策略，使调用更加易用：

策略工厂类：一般来说，通过一个策略工厂，将一群功能相同的策略封装起来，调用更加方便。

通过一个文件处理的例子来说明；有一个文件处理的通用类，可以处理excel、txt、exe文件。
面对不同类型的文件，返回具体对应的文件处理类，也就是具体的策略类。

抽象策略接口类

class FileStrategy:
    @abstractmethod
    def read(self):
        pass

抽象策略接口，定义了每个具体的策略类需要实现的方法，符合基于抽象而非实现编程。

策略实现类

class ExeFileStrategy(FileStrategy):

    def read(self):
        print("--- 进行 exe 文件的读取 ---")


class ExeclFileStrategy(FileStrategy):

    def read(self):
        print("--- 进行 excel 文件的读取 ---")


class TxtFileStrategy(FileStrategy):

    def read(self):
        print("--- 进行 txt 文件的读取 ---")

上面分别实现三个不同的具体策略类来处理 exe/excel/txt 文件的读取，每个具体的策略类，要在read方法中实现自己的读取逻辑。

策略工厂类

根据策略类是否有状态，可以分为两种策略的工厂生成方式。

# 策略类无状态时，每次返回同一个策略类，主要提供逻辑的计算。
class FileStrategyFactory:
    MAP = {
        "exe": ExeFileStrategy(),
        "execl": ExeclFileStrategy(),
        "txt": TxtFileStrategy(),
    }

    def create(self, key):
        return self.MAP[key]


# 策略类有状态时，每次返回独立的策略对象，策略对象之间独立。
class FileStrategyFactory1:
    MAP = {
        "exe": ExeFileStrategy,
        "execl": ExeclFileStrategy,
        "txt": TxtFileStrategy,
    }

    def create(self, key):
        return self.MAP[key]()

上面我们实现了两种策略工厂，并且都是使用简单工厂方法，FileStrategyFactory是无状态的，每次相同的key，返回的是同一个策略对象；FileStrategyFactory1是有状态的，每次返回策略对象之间独立。

具体使用

# 无状态策略类
fac = FileStrategyFactory()
exe = fac.create("exe")
exe1 = fac.create("exe")
execl = fac.create("execl")
print(exe is exe1)  # True，每次返回同一个对象
exe.read()
execl.read()

# 有状态策略类
fac = FileStrategyFactory1()
exe = fac.create("exe")
exe1 = fac.create("exe")
execl = fac.create("execl")
print(exe is exe1)  # False，每次返回的对象不同
exe.read()
execl.read()

替代if / else

策略模式，是如何解决冗长的if / else 逻辑，以及解耦不同的策略的呢？

假设我们没有策略类，上面的文件处理类例子，我们大概会写成这样：

if file_type == "exe":
    print("--- 进行 exe 文件的读取 ---")
elif file_type == "execl":
    print("--- 进行 execl 文件的读取 ---")
elif file_type == "txt":
    print("--- 进行 txt 文件的读取 ---")