软件构造:策略模式
本文是软件构造系列的第七篇文章。其余文章在:
https://ruanx.net/tag/software-construction/
求最小生成树有 Kruskal 和 Prim 两种算法;从长沙去哈尔滨可以乘坐火车或者飞机;在 Steam 购买游戏可以选择支付宝、微信或 visa 信用卡支付。这些都是同一个任务的多种实现方式。以 Steam 为例,支付的逻辑可能写成下面的形式:
class BuyGame {
void pay(String via) {
if ("Alipay".equals(via)) {
System.out.println("Pay with Alipay");
}
if ("Wechat".equals(via)) {
System.out.println("Pay with Wechat");
}
if ("Visa".equals(via)) {
System.out.println("Pay with visa card");
}
}
}
客户端通过参数传入支付方式:
public static void main(String[] args) {
BuyGame order = new BuyGame();
order.pay("Alipay");
}
看起来是解决了问题,但考虑一个情形:现在除了上述几种支付方式外,要添加 master 信用卡、银联、Paypal 支付。我们将不得不修改 BuyGame
类的代码,这违背了开闭原则。
策略模式可以解决这个问题。它的思想是基于单一责任:把算法实现与算法使用分离开,用若干个独立的类(策略类)专门实现算法。算法的调用者(称为“环境类”)在执行任务时,委派一个策略类来完成任务。
按照策略模式,首先需要把支付方式分拆到自己的类里面:
interface PaymentStrategy {
void pay();
}
class AlipayStrategy implements PaymentStrategy {
public void pay() {
System.out.println("Pay with Alipay");
}
}
class WechatStrategy implements PaymentStrategy {
public void pay() {
System.out.println("Pay with Wechat");
}
}
class VisaStrategy implements PaymentStrategy {
public void pay() {
System.out.println("Pay with visa card");
}
}
BuyGame
委派一个支付策略来实现支付逻辑:
class BuyGame {
private PaymentStrategy via;
public BuyGame(PaymentStrategy via) {
this.via = via;
}
public void pay() {
via.pay();
}
}
现在,客户想通过 Alipay 来支付一笔订单,只需要把一个 AlipayStrategy
实例交给 BuyGame,再调用 pay()
接口:
public static void main(String[] args) {
BuyGame order = new BuyGame(new AlipayStrategy());
order.pay();
}
很明显,想要添加新的支付方式,只需要添加几个对 PaymentStrategy
的实现类即可。无需改动其他部分的代码;client 代码也无需改变,完美地实现了开闭原则。
另外,策略模式还有一个好处:既然算法实现是分拆成了多个类来实现,它们就可以通过继承来复用代码。例如,visa 和 master 支付都有 “输入信用卡号” 等步骤,可以用一个父类实现这些共性操作,再由 visa 和 master 继承,去实现个性操作。单一责任使得这些继承只需要关注算法的共性,而无需关注执行算法的环境,这给开发者和用户带来了很大便利。