526互联

策略模式代替老式的if-else

发布时间 2023-11-09 09:43:43作者: chelsey3tsf

最近重构项目发现存在巨多的if else ，if …else if … else…这确实是我们在各种项目研发中的一个利器，但是一旦他过于多，就会使逻辑混乱，让代码的可读性大大下降，因而打算使用策略模式来代替过多的if else。

策略模式的介绍
策略模式
引入百科的解释：策略模式作为一种软件设计模式，指对象有某个行为，但是在不同的场景中，该行为有不同的实现算法。比如每个人都要“交个人所得税”，但是“在美国交个人所得税”和“在中国交个人所得税”就有不同的算税方法。

策略模式：顾名思义就是当要在不同的情况下做出该情况下所适用的策略，按照面向对象的是思想就是，就使用继承和多态机制，实现同一行为在不同场景下具备不同实现。

策略模式的本质：分离算法，选择实现。

主要解决
从上述的策略模式的本质中的选择实现是不是可以联想到if…else if… else…没错，在有多种算法相似的情况下，使用 if…else 或 switch…case 所带来的复杂性和臃肿性。

优缺点
优点

策略的多样性，能够自由切换
有效的避免了多重判断，提高了代码的可读性，扩展性，封装性
降低了操作难度以及错误率。
缺点

增加了开发难度，策略类数量增多，并且所要策略类都要对外开放，以便客户端访问
你还在用if else吗？
这里给大家举一个典型的例子
我们每天扫码支付，选用支付方式，选择的微信，支付宝，或者是直接银行卡支付。想必大家已经屡见不鲜了。
那我们现在我们需要做一个支付系统，通过扫描二维码对一个商户进行付款，我们可以选用我们已经绑定的支付宝账号，微信，或者所绑定的银行，直接进行支付，由于每家商户不同，可能付款的时候可能也会有一些优惠政策的不同，并且他们为外界所提供的API也会有所不同。例如，支付宝可以使用红包进行抵用，直接使用银行卡付款，每家银行可能会有不同的折扣。具体情况比较复杂，这里就不做考虑，为了为接下来的策略模式做铺垫，我们这里就简单的分为，微信支付，支付宝支付，银行卡支付，抵用券支付。
我相信很多人拿到需求后，最先考虑的写法就是：

package pay;

public class Example {

    public String selectPayWay(Integer payCode){
        if(PayWayEnum.WEICHAT_PAY.equals(payCode)){
            //do something
            return "微信支付成功";
        }else if(PayWayEnum.AL_PAY.equals(payCode)){
            // do something
            return "支付宝支付成功";
        }else if(PayWayEnum.CARD_PAY.equals(payCode)){
            // do something
            return "银行卡成功";
        }else if(PayWayEnum.PONIT_COUPON_PAY.equals(payCode)){
            // do something
            return "优惠券支付成功";
        }else {
            return "";
        }
    }

}

可以看出上面四种不同的支付方式在同一个方法体内部，不利于扩展和维护，并且也不符合面向对象设计原则。对以上的代码使用策略模式进行代替，类图如下：

从 上面的UML 类图中，我们可以看到，策略模式 主要包含三种角色：

  • 上下文角色（Context）：用来操作策略的上下文环境，屏蔽高层模块（客户端）对策略，算法的直接访问，封装可能存在的变化；
  • 抽象策略角色（Strategy）：规定策略或算法的行为；
  • 具体策略角色（ConcreteStrategy）：具体的策略或算法实现；

java代码实现具体如下

package strategy;

public interface PayStrategy {
    String selectPayWay(Integer paycode);
}

规定具体策略行为

package strategy;

public class PayContext {

    private PayStrategy payStrategy;

    public String selectPayWay(Integer payCode){
        payStrategy =  StrategyFactory.getInstance().create(payCode);
        return payStrategy.selectPayWay(payCode);
    }

    public PayStrategy getPayStrategy() {

        return payStrategy;
    }

    public void setPayStrategy(PayStrategy payStrategy) {
        this.payStrategy = payStrategy;
    }
}

支付宝付款实现类

package strategy.paystrategy;

import strategy.PayStrategy;

public class ALPayStrategy implements PayStrategy {

    @Override
    public String selectPayWay(Integer paycode) {
        //do something
        return "支付宝支付成功";
    }
}

银行卡支付实现类

package strategy.paystrategy;

import strategy.PayStrategy;

public class CardPayStrategy implements PayStrategy {
    @Override
    public String selectPayWay(Integer paycode) {
        //do something
        return "银行卡支付成功";
    }
}

点券支付实现类

package strategy.paystrategy;

import strategy.PayStrategy;

public class PointCouponPayStrategy implements PayStrategy {

    @Override
    public String selectPayWay(Integer paycode) {
        //do something
        return "点券支付成功";
    }
}

微信支付实现类

package strategy.paystrategy;

import strategy.PayStrategy;

public class WeChatPayStrategy implements PayStrategy {

    @Override
    public String selectPayWay(Integer paycode) {
        //do something
        return "微信支付成功";
    }

}

创建策略类的工厂

package strategy;

import pay.PayWayEnum;
import strategy.paystrategy.ALPayStrategy;
import strategy.paystrategy.CardPayStrategy;
import strategy.paystrategy.PointCouponPayStrategy;
import strategy.paystrategy.WeChatPayStrategy;

import java.util.HashMap;
import java.util.Map;

public class StrategyFactory {

    private static StrategyFactory factory = new StrategyFactory();

    private StrategyFactory(){}

    private static Map strategyMap = new HashMap<>(16);

    static {
        strategyMap.put(PayWayEnum.AL_PAY.getCode(),new ALPayStrategy());
        strategyMap.put(PayWayEnum.WEICHAT_PAY.getCode(),new WeChatPayStrategy());
        strategyMap.put(PayWayEnum.CARD_PAY.getCode(),new CardPayStrategy());
        strategyMap.put(PayWayEnum.PONIT_COUPON_PAY.getCode(),new PointCouponPayStrategy());
    }

    public PayStrategy create(Integer payCode){
        return (PayStrategy) strategyMap.get(payCode);
    }

    public static StrategyFactory getInstance(){
        return factory;
    }

}

实现客户端

package strategy;

public class Client {

    public static void main(String[] args) {
        PayContext payContext = new PayContext();
        System.out.println(payContext.selectPayWay(100));
    }
}

最后在贴一个枚举类

package pay;

public enum PayWayEnum {

    /**微信支付*/
    WEICHAT_PAY("微信支付",100),
    /**支付宝支付*/
    AL_PAY("支付宝支付",101),
    /**银行卡支付*/
    CARD_PAY("银行卡支付",102),
    /**点券支付*/
    PONIT_COUPON_PAY("点券支付",103);

    private String msg;
    private Integer code;

    PayWayEnum(String msg, Integer code) {
        this.msg = msg;
        this.code = code;
    }

    public String getMsg() {
        return msg;
    }

    public Integer getCode() {
        return code;
    }
}