解决多if语句- 策略模式

策略模式

背景

在软件开发中常常遇到这种情况,实现某一个功能有多种算法或者策略,我们需要根据环境或者条件的不同选择不同的算法或者策略来完成该功能。如支付环境需要根据传入的条件判断是微信支付、支付宝支付、银行卡支付等等,一种常用的方法是硬编码(Hard Coding)在一个类中,如需要提供多种查找算法,可以将这些算法写到一个类中,在该类中提供多个方法,每一个方法对应一个具体的查找算法;当然也可以将这些查找算法封装在一个统一的方法中,通过if…else…或者case等条件判断语句来进行选择。
这两种实现方法我们都可以称之为硬编码,如果需要增加一种新的查找算法,需要修改封装算法类的源代码;更换查找算法,也需要修改客户端调用代码。在这个算法类中封装了大量查找算法,该类代码将较复杂,维护较为困难。如果我们将这些策略包含在客户端,这种做法更不可取,将导致客户端程序庞大而且难以维护,如果存在大量可供选择的算法时问题将变得更加严重。

方案

把一个类中经常改变或者将来可能改变的部分提取出来,作为一个接口,然后在类中包含这个对象的实例,这样类的实例在运行时就可以随意调用实现了这个接口的类的行为。

优点

1、可以动态的改变对象的行为

缺点

1、客户端必须知道所有的策略类,并自行决定使用哪一个策略类
2、策略模式将造成产生很多策略类

组成

环境类(Context):用一个ConcreteStrategy对象来配置。维护一个对Strategy对象的引用。可定义一个接口来让Strategy访问它的数据。
抽象策略类(Strategy):定义所有支持的算法的公共接口。 Context使用这个接口来调用某ConcreteStrategy定义的算法。
具体策略类(ConcreteStrategy):以Strategy接口实现某具体算法。

示例(以旅游为选择交通工具为背景)

/**
 * @Author: zjr
 * 策略模式
 * @Date: 21-4-8
 */
public interface TravelStrategy {
    /**
     * 行程需要支付的费用算法
     * @return
     */
    double travelAlgorithm();

    /**
     * 获取类型
     * @return
     */
    String getTravelType();
}

/**
 * @Author: zjr
 * 火车
 * @Date: 21-4-8
 */
@Service
public class AirStrategy implements TravelStrategy{

    @Override
    public double travelAlgorithm() {
        System.out.println("乘坐飞机。。。。。。");
        return 1000.0;
    }

    @Override
    public String getTravelType() {
        return "Air";
    }
}

/**
 * @Author: zjr
 * 火车
 * @Date: 21-4-8
 */
@Service
public class TrainStrategy implements TravelStrategy{

    @Override
    public double travelAlgorithm() {
        System.out.println("乘坐火车。。。。。。");
        return 200.0;
    }

    @Override
    public String getTravelType() {
        return "Train";
    }
}

/**
 * @Author: zjr
 * @Date: 21-4-8
 */
public class OrderCheckOut {
    Map<String, TravelStrategy> strategyHashedMap = new HashMap<>(8);

    public OrderCheckOut() {
        AirStrategy airStrategy = new AirStrategy();
        TrainStrategy trainStrategy = new TrainStrategy();
        strategyHashedMap.put(airStrategy.getTravelType(),airStrategy);
        strategyHashedMap.put(trainStrategy.getTravelType(),trainStrategy);
    }

    public double travelAlgorithm(String travelType) {
        TravelStrategy travelStrategy = strategyHashedMap.get(travelType);
        if(null == travelStrategy){
            travelStrategy = new TrainStrategy();
        }
        return travelStrategy.travelAlgorithm();
    }
}

测试

/**
 * @Author: zjr
 * @Date: 21-4-8
 */
public class StrategyMain {

    public static void main(String[] args) {
        OrderCheckOut orderCheckOut = new OrderCheckOut();
        double fee = orderCheckOut.travelAlgorithm("Air");
        System.out.println(fee);
    }
}

这样的也可以优化多个if语句


版权声明:本文为weixin_40245601原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。