设计模式系列——策略模式

前言

软件开发中,经常遇到一种情况:实现一个功能可以有多种算法或策略,我们根据实际情况选择不同的算法或者策略来完成该功能。
针对这种情况,常规实现方法是将多种算法写在一个类中,然后通过if-else或者switch-case等条件语句来选择具体的算法,但是当很多个算法集中在一个类时,这个类就会变得臃肿,维护成本变高,另外当我们想增加一种算法,或者想修改其中某一算法时,需要修改这个类的源代码,容易引发错误。这就明显违反了开闭原则和单一职责原则。这种情况下可以考虑使用策略模式。
感谢《Android源码设计模式解析与实战》!

策略模式的定义

策略模式定义了一系列的算法,并将每一个算法封装起来,而且使他们还可以相互替换。策略模式让算法独立于使用它的客户而独立变化。

策略模式的使用场景

  • 针对同一类型问题的多种处理方式,仅仅是具体行为有差别时。
  • 需要安全地封装多种同一类型的操作时。
  • 出现同一抽象类有多个子类,而又需要使用if-else或者switch-case来选择具体子类时。

策略模式的UML类图

设计模式系列——策略模式_第1张图片
策略模式UML类图.png
  • Context: 用来操作策略的上下文环境。
  • Stragety: 策略的抽象。
  • ConcreteStragetyA、ConcreteStragetyB: 具体的策略实现。

策略模式的简单示例

我们以交通工具为例,分别计算一下公交车和地铁的价格。公交车和地铁的计费方式分别对应不同的算法。

  • 普通实现
public class PriceCalculator {
    //公交车类型
    private static final int BUS = 1;
    //地铁类型
    private static final int SUBWAY = 2;

    public static void main(String[] args) {
        PriceCalculator calculator = new PriceCalculator();
        System.out.println("坐20公里的公交车票价为:  " + calculator.calculatePrice(20, BUS));
        System.out.println("坐20公里的地铁票价为:  " + calculator.calculatePrice(20, SUBWAY));
    }
     //计算价格
    private int calculatePrice(int km, int type) {
        if (type == BUS) {
            return busPrice(km);
        } else if (type == SUBWAY) {
            return subwayPrice(km);
        }
        return 0;
    }

    /**
     * 北京公交车,十公里之内一元钱,超过十公里之后,每五公里加一元。
     *
     * @param km 公里
     * @return
     */
    private int busPrice(int km) {
        //超过十公里的总距离
        int extraTotal = km - 10;
        //超过的距离是5公里的倍数
        int extraFactor = extraTotal / 5;
        //超过的距离对5公里取余
        int fraction = extraTotal % 5;
        //价格计算
        int price = 1 + extraFactor * 1;
        return fraction > 0 ? ++price : price;
    }

    /**
     * 6公里(含)内3元;6~12公里(含)4元;12~22公里(含)5元;22~32公里(含)6元
     *
     * @param km
     * @return
     */
    private int subwayPrice(int km) {
        if (km <= 6) {
            return 3;
        } else if (km > 6 && km <= 12) {
            return 4;
        } else if (km > 12 && km <= 22) {
            return 5;
        } else if (km > 22 && km <= 32) {
            return 6;
        }
        //其他距离简化为7元;
        return 7;
    }
}

以上示例中,PriceCalculator类中承担了计算公交车和地铁乘坐价格的职责,并且通过if-else的形式来判断使用哪种计算方式,很明显违反了单一职责原则。
如果此时增加出行方式出租车,大致代码如下:

public class PriceCalculator {
    //公交车类型
    private static final int BUS = 1;
    //地铁类型
    private static final int SUBWAY = 2;
    //出租车类型
    private static final int TAXI = 3;

    //计算价格
    private int calculatePrice(int km, int type) {
        if (type == BUS) {
            return busPrice(km);
        } else if (type == SUBWAY) {
            return subwayPrice(km);
        } else if (type == TAXI) {//**增加了判断**
            return taxiPrice(km);
        }
        return 0;
    }

    /**
     * 假设每公里2元
     *
     * @param km
     * @return
     */
    private int taxiPrice(int km) {//**增加了计算方式**
        return km * 2;
    }
}

可以看出,当我们每增加一种出行方式时,都需要在PriceCalculator中增加一个方法来计算该方式的出行价格,并在calculatePrice中进行判断,大量重复使用if-else。另外当计算方式发生变化时,需要直接在PriceCalculator类中进行修改,很可能一段代码是其他几个计算方式公用的,这就容易引起错误。在应对变化时,这类代码必然先得臃肿,难以维护。策略模式就很好的解决了以上问题。

  • 策略模式实现
    定义一个抽象的计算乘车价格的接口。
public interface CalculateStrategy {
    /**
     * 按距离计算价格
     *
     * @param km 公里
     * @return 返回价格
     */
    int calculatePrice(int km);
}

每一种出行方式都对应一个独立的计算策略类,这些策略类都实现了CalculateStrategy接口。

//公交价格计算策略
public class BusStrategy implements CalculateStrategy {
    /**
     * 北京公交车,十公里之内一元钱,超过十公里之后,每五公里加一元。
     *
     * @param km 公里
     * @return
     */
    @Override
    public int calculatePrice(int km) {
        //超过十公里的总距离
        int extraTotal = km - 10;
        //超过的距离是5公里的倍数
        int extraFactor = extraTotal / 5;
        //超过的距离对5公里取余
        int fraction = extraTotal % 5;
        //价格计算
        int price = 1 + extraFactor * 1;
        return fraction > 0 ? ++price : price;
    }
}
//地铁价格计算策略
public class SubwayStrategy implements CalculateStrategy {
    /**
     * 6公里(含)内3元;6~12公里(含)4元;12~22公里(含)5元;22~32公里(含)6元
     *
     * @param km
     * @return
     */
    @Override
    public int calculatePrice(int km) {
        if (km <= 6) {
            return 3;
        } else if (km > 6 && km <= 12) {
            return 4;
        } else if (km > 12 && km <= 22) {
            return 5;
        } else if (km > 22 && km <= 32) {
            return 6;
        }
        //其他距离简化为7元;
        return 7;
    }
}

我们再创建一个扮演Context角色的类。

//出行方式价格计算器
public class TrafficCalculator {

    private CalculateStrategy mStrategy;

    public static void main(String[] args) {
        TrafficCalculator calculator = new TrafficCalculator();
        //设置计算策略
        calculator.setStrategy(new BusStrategy());
        //计算价格
        System.out.println("坐20公里的公交车票价为:  " + calculator.calculatePrice(20));
    }

    private void setStrategy(CalculateStrategy strategy) {
        this.mStrategy = strategy;
    }

    public int calculatePrice(int km) {
        return mStrategy.calculatePrice(km);
    }
}

经过重构以后,去掉了大量重复的if-else语句,结构便得清晰。另外在隐藏实现的同时,可扩展性变强。例如当我们想增加出租车为出行策略时,只需增加一个出租车计算策略类,然后将该策略设置给TrafficCalculator。如下:

//出租车价格计算策略
public class TaxiStrategy implements CalculateStrategy {
    /**
     * 假设每公里2元
     *
     * @param km 公里
     * @return
     */
    @Override
    public int calculatePrice(int km) {
        return km * 2;
    }
}

将策略注入到TrafficCalculator中。

//出行方式价格计算器
public class TrafficCalculator {

    private CalculateStrategy mStrategy;

    public static void main(String[] args) {
        TrafficCalculator calculator = new TrafficCalculator();
        //设置计算策略
        calculator.setStrategy(new TaxiStrategy());
        //计算价格
        System.out.println("坐20公里的出租车票价为:  " + calculator.calculatePrice(20));
    }

    private void setStrategy(CalculateStrategy strategy) {
        this.mStrategy = strategy;
    }

    public int calculatePrice(int km) {
        return mStrategy.calculatePrice(km);
    }
}

通过上述实例我们可以清晰地看出二者的区别所在。前者通过if-else来解决问题,虽然实现较为简单,类型层级单一,但暴露的问题非常明显,即代码臃肿,逻辑复杂,难以升级和维护,没有结构可言;后者则是通过建立抽象,将不同的策略构建成一个具体的策略实现,通过不同的策略实现算法替换。在简化逻辑、结构的同时,增强了系统的可读性、稳定性、可扩展性,这对于较为复杂的业务逻辑先得更为直观,扩展也更为方便。

总结

策略模式主要是用来分离算法,在相同的行为抽象下有不同的具体实现策略。这个模式很好的演示了开闭原则,也就是定义抽象,注入不容的实现,从而达到很好的可扩展性。

优点
  • 结构清晰明了、使用简单直观;
  • 耦合度相对而言较低,扩展方便;
  • 操作封装也更为彻底,数据更为安全。
缺点
  • 随着策略的增加,子类也会变得繁多。

你可能感兴趣的:(设计模式系列——策略模式)