策略模式是一种行为设计模式,它通过定义一系列算法,将每个算法封装起来,并使它们可以互相替换,从而让算法的变化独立于使用算法的客户。策略模式使得算法可以在不影响客户端的情况下发生变化。
策略模式的组成部分:
- 策略接口(Strategy): 定义所有支持的算法的公共接口。
- 具体策略(Concrete Strategy): 实现策略接口的各个具体算法。
- 上下文(Context): 维护一个对策略对象的引用,并且可以接受客户端的请求,委托给策略对象来执行。
策略模式的优点:
- 易于扩展: 可以方便地增加新的策略而不影响客户代码。
- 避免使用多重条件语句: 通过策略模式,可以将条件判断逻辑转移到不同的策略类中。
策略模式的缺点:
- 策略数量增多: 当策略过多时,可能会导致类的数量增加,增加系统复杂性。
- 客户端必须知道不同的策略: 客户端需要了解各个策略之间的区别,以便选择合适的策略。
示例代码:
以下是一个简单的策略模式的例子:
from abc import ABC, abstractmethod
# 策略接口
class Strategy(ABC):
@abstractmethod
def do_operation(self, num1: int, num2: int) -> int:
pass
# 具体策略1
class OperationAdd(Strategy):
def do_operation(self, num1: int, num2: int) -> int:
return num1 + num2
# 具体策略2
class OperationSubtract(Strategy):
def do_operation(self, num1: int, num2: int) -> int:
return num1 - num2
# 上下文
class Context:
def __init__(self, strategy: Strategy):
self._strategy = strategy
def execute_strategy(self, num1: int, num2: int) -> int:
return self._strategy.do_operation(num1, num2)
# 客户端代码
if __name__ == "__main__":
context = Context(OperationAdd())
print("10 + 5 =", context.execute_strategy(10, 5))
context = Context(OperationSubtract())
print("10 - 5 =", context.execute_strategy(10, 5))
在这个例子中,Strategy
是策略接口,OperationAdd
和 OperationSubtract
是具体策略类,而 Context
类持有一个策略对象并使用它来执行具体的算法。