设计模式-行为型模式-策略模式

最新推荐文章于 2024-09-03 22:52:48 发布

帅帅气气的黑猫警长

最新推荐文章于 2024-09-03 22:52:48 发布

阅读量3.6k

点赞数

分类专栏：设计模式文章标签：设计模式策略模式

本文链接：https://blog.csdn.net/Hao_ge_666/article/details/134491818

版权

设计模式专栏收录该内容

5 篇文章 0 订阅

订阅专栏

一、什么是策略模式

策略模式是一种行为设计模式，它允许在运行时选择算法或行为，并将其封装成独立的对象，使得这些算法或行为可以相互替换，而不影响使用它们的客户端。（ChatGPT生成）

主要组成部分：

1、策略（Strategy）： 定义了一个算法族、行为或方法，并将其封装在一个接口或抽象类中，使得这些算法可以相互替换。

2、具体策略（Concrete Strategy）： 实现了策略接口，提供了具体的算法或行为实现。

3、上下文（Context）： 持有一个策略对象的引用，在运行时可动态切换不同的策略。

原理：

1、策略模式允许在运行时动态选择算法或行为，通过将具体的算法或行为封装成策略对象，然后将这些策略对象注入到上下文中。

2、上下文根据不同的需求或条件选择合适的策略对象，并使用它执行特定的算法或行为。

优点：

1、易于扩展和维护： 可以方便地添加新的策略或修改现有策略，不影响原有代码。

2、避免条件语句： 可以避免大量的条件语句，提高代码可读性和可维护性。

3、代码复用： 可以通过共享策略对象来提高代码的复用性。

二、场景模拟

为了应用策略模式，我们模拟一个场景。在一个企业资源管理（ERP）系统中，可能会涉及到成本的计算，其中需要计算的成本例如：钢材、油漆、人工、运输等等，虽然说计算成本都是“单价*数量”，但是有些成本，例如钢材，不仅要计算钢材的成本，而且放置钢材或者其它费用，也需要进行一定的计费；对于人工成本，还需要计算企业担负的五险一金等。

如上图所示，钢材是由吨数*单价计算的，但是要考虑保管费或其他费用，假设一吨n元，后续还可能考虑其它费用等；人工费用，雇员一人可能工资12000元，公司要负担其12%公积金或者医疗保险等；运输要考虑车辆维护费或者日常损耗。（以上仅是模拟，真实环境可能更加复杂或简单）。

三、业务实现

3.1、使用IF-ELSE方法

public static BigDecimal useIfElse(String name, BigDecimal amount, BigDecimal price){

        if (name.equals("plate")) {
            // 单价×数量，再加上每吨200元的其它费用
            return amount.multiply(price).add(amount.multiply(new BigDecimal("200.00")));
        } else if (name.equals("paints")) {
            // 单价×数量，再加上每吨200元的其它费用
            return amount.multiply(price).add(amount.multiply(new BigDecimal("100.00")));
        } else if (name.equals("employer")) {
            // 人员×数量，再加上每吨200元的其它费用
            BigDecimal sum = amount.multiply(price); // 基本工资
            sum = sum.add(amount.multiply(price.multiply(new BigDecimal("0.12")))); // 加上额外支付的12%公积金
            return sum;
        } else if (name.equals("haulage")) {
            // 吨数×数量，再加上每吨300元的损耗费用
            return amount.multiply(price).add(amount.multiply(new BigDecimal("300.00")));
        } else {
            log.error("无法匹配");
            return new BigDecimal(0);
        }

    }

我们用IF-ELSE方法，确实是可以实现功能，可以看出来很麻烦，看起来很混乱，没有结构，也没办法实现一些定制功能，后续变动起来也比骄困难，因为有一些东西都是写死的。我们来测试一下：

public static void main(String[] args) {

        // 钢板单价及数量
        BigDecimal plateAmount = new BigDecimal(15);
        BigDecimal platePrice = new BigDecimal("888.88");

        // 油漆价格及数量
        BigDecimal paintsAmount = new BigDecimal(20);
        BigDecimal paintsPrice = new BigDecimal("490.00");

        // 雇员工资及人数
        BigDecimal employerAmount = new BigDecimal(10);
        BigDecimal employerPrice = new BigDecimal("10000.00");

        // 运输价格及吨数
        BigDecimal haulageAmount = new BigDecimal(25);
        BigDecimal haulagePrice = new BigDecimal("3000.00");

        // 计算各个成本
        BigDecimal plate = useIfElse("plate", plateAmount, platePrice);
        BigDecimal paints = useIfElse("paints", paintsAmount, paintsPrice);
        BigDecimal employer = useIfElse("employer", employerAmount, employerPrice);
        BigDecimal haulage = useIfElse("haulage", haulageAmount, haulagePrice);

        BigDecimal sum = plate.add(paints).add(employer).add(haulage);

        log.info("钢材:{}元,油漆{}元,雇员:{}元,运输:{}元,总计:{}元", plate, paints, employer, haulage, sum);

    }

可以看到确实正确的计算出了结果，但是这种方式不具有拓展性，随着程序功能的增加，会越来越混乱。那有些同学可能会说，为什么不用Switch-Case方法？那就来实践一下。

3.2、使用Switch-Case方法

public static BigDecimal useSwitch(String name, BigDecimal amount, BigDecimal price){

        switch (name) {
            case "plate":
                // 单价×数量，再加上每吨200元的其它费用
                return amount.multiply(price).add(amount.multiply(new BigDecimal("200.00")));
            case "paints":
                // 单价×数量，再加上每吨200元的其它费用
                return amount.multiply(price).add(amount.multiply(new BigDecimal("100.00")));
            case "employer":
                // 人员×数量，再加上每吨200元的其它费用
                BigDecimal sum = amount.multiply(price); // 基本工资
                sum = sum.add(amount.multiply(price.multiply(new BigDecimal("0.12")))); // 加上额外支付的12%公积金
                return sum;
            case "haulage":
                // 吨数×数量，再加上每吨300元的损耗费用
                return amount.multiply(price).add(amount.multiply(new BigDecimal("300.00")));
            default:
                log.error("无法匹配");
                return new BigDecimal(0);
        }

    }

使用Switch-Case方法后，确实代码变得比较清晰了，但是一些东西仍然是写死的，不具备很好的扩展性。

3.3、使用策略模式重构代码

其中ICosts是一个接口，里面只有一个方法，就是成本的计算。他的实现类包含钢材、油漆、人工等等，可以有很多。Context是策略控制类，外部可以传递不同策略执行计算成本方法。

3.3.1、编写成本计算接口

public interface ICosts<T, K, V> {

    /**
     * 成本计算
     * @param amount 数量
     * @param price 单价
     * @param other 其它信息
     * @return 成本金额
     */
    BigDecimal calculateCosts(T amount, BigDecimal price, Map<K, V> other);
}

解释一下amount为什么要用T表示，因为这里数量不仅仅是BigDecimal形式的，可能会涉及到人、车辆等等，这种是没有小数点的……；关于other就是一些附加信息，例如钢材的除锈费用、维护费用，人员的五险一金比例等等。

3.3.2、具体成本实现类

钢材

public class GCCosts implements ICosts<BigDecimal, String, BigDecimal> {

    /*
    * 钢材成本
    *   1、基本成本：吨数 * 价格；
    *   2、存放成本：吨数 * 存放消费；
    *   3、防锈成本：吨数 * 防锈成本；
    * */
    @Override
    public BigDecimal calculateCosts(BigDecimal amount, BigDecimal price, Map<String, BigDecimal> other) {

        BigDecimal sum = amount.multiply(price);

        // 计算存放成本
        if (other.containsKey("CF")) {
            sum = sum.add(amount.multiply(other.get("CF")));
        }
        // 计算防锈成本
        if (other.containsKey("CX")) {
            sum = sum.add(amount.multiply(other.get("CX")));
        }

        return sum;
    }
}

油漆

public class YQCosts implements ICosts<BigDecimal, String, BigDecimal> {

    /*
    * 油漆成本
    *   1、基础成本：单价 * 数量；
    *   2、存放成本： 数量 * 存放费用；
    * */
    @Override
    public BigDecimal calculateCosts(BigDecimal amount, BigDecimal price, Map<String, BigDecimal> other) {

        BigDecimal sum = amount.multiply(price);

        if (other.containsKey("CF")) {
            sum = sum.add(amount.multiply(other.get("CF")));
        }

        return sum;
    }
}

人工

public class RGCosts implements ICosts<Long, String, BigDecimal> {

    /*
    * 人工成本：
    *   1、基础成本：人数 * 工资；
    *   2、公积金成本： 人数 * （工资 * 公积金比例）；
    *   3、五险成本：人数 * （工资 * 企业分摊比例）；
    * */
    @Override
    public BigDecimal calculateCosts(Long amount, BigDecimal price, Map<String, BigDecimal> other) {

        // 先计算每个人成本
        BigDecimal personCosts = price;

        // 加公积金
        if (other.containsKey("GJJ")) {
            personCosts = personCosts.add(price.multiply(other.get("GJJ")));
        }

        // 加五险
        if (other.containsKey("BX")) {
            personCosts = personCosts.add(price.multiply(other.get("BX")));
        }

        return personCosts.multiply(BigDecimal.valueOf(amount)); // 计算总成本
    }
}

运输

public class YSCosts implements ICosts<BigDecimal, String, BigDecimal> {

    /*
    * 运输成本：
    *   1、基础成本：运输数量 * 运输单价；
    *   2、损耗费用：例如换机油，养护等费用，固定金额（模拟）；
    * */
    @Override
    public BigDecimal calculateCosts(BigDecimal amount, BigDecimal price, Map<String, BigDecimal> other) {

        BigDecimal sum = amount.multiply(price);

        // 加上固定的养护费用
        if (other.containsKey("YH")) {
            sum = sum.add(other.get("YH"));
        }

        return sum;
    }
}

3.3.3、成本策略控制类

public class CostsContext<T, K, V>{

    private ICosts<T, K, V> iCosts;

    public CostsContext(ICosts<T, K, V> iCosts) {
        this.iCosts = iCosts;
    }

    public BigDecimal calculateCosts(T amount, BigDecimal price, Map<K, V> other) {
        return iCosts.calculateCosts(amount, price, other);
    }
}

此类充当一个调度作用，不同的成本计算就选择不同的实现类，然后执行calculateCosts()方法。

3.3.4、测试

public static void main(String[] args) {

        // 钢板单价及数量
        BigDecimal plateAmount = new BigDecimal(15);
        BigDecimal platePrice = new BigDecimal("888.88");
        Map<String, BigDecimal> plateMap = new HashMap<String, BigDecimal>(){{
            put("CF", new BigDecimal("300"));
            put("CX", new BigDecimal("200"));
        }};

        // 油漆价格及数量
        BigDecimal paintsAmount = new BigDecimal(20);
        BigDecimal paintsPrice = new BigDecimal("490.00");
        Map<String, BigDecimal> paintsMap = new HashMap<String, BigDecimal>(){{
            put("CF", new BigDecimal("300"));
        }};

        // 雇员工资及人数
        // BigDecimal employerAmount = new BigDecimal(10);
        Long employerAmount = 10L;
        BigDecimal employerPrice = new BigDecimal("10000.00");
        Map<String, BigDecimal> employerMap = new HashMap<String, BigDecimal>(){{
            put("GJJ", new BigDecimal("0.12"));
            put("BX", new BigDecimal("0.06"));
        }};

        // 运输价格及吨数
        BigDecimal haulageAmount = new BigDecimal(25);
        BigDecimal haulagePrice = new BigDecimal("3000.00");
        Map<String, BigDecimal> haulageMap = new HashMap<String, BigDecimal>(){{
            put("YH", new BigDecimal("1500"));
        }};

        // 使用策略模式计算
        // 钢材
        CostsContext<BigDecimal, String, BigDecimal> plateCost = new CostsContext<>(new GCCosts());
        BigDecimal plate = plateCost.calculateCosts(plateAmount, platePrice, plateMap);

        // 油漆
        CostsContext<BigDecimal, String, BigDecimal> paintsCost = new CostsContext<>(new YQCosts());
        BigDecimal paints = paintsCost.calculateCosts(paintsAmount, paintsPrice, paintsMap);

        // 人工
        CostsContext<Long, String, BigDecimal> employerCost = new CostsContext<>(new RGCosts());
        BigDecimal employer = employerCost.calculateCosts(employerAmount, employerPrice, employerMap);

        // 运输
        CostsContext<BigDecimal, String, BigDecimal> haulageCost = new CostsContext<>(new YQCosts());
        BigDecimal haulage = haulageCost.calculateCosts(haulageAmount, haulagePrice, haulageMap);

        BigDecimal sum = plate.add(paints).add(employer).add(haulage);

        log.info("钢材:{}元,油漆{}元,雇员:{}元,运输:{}元,总计:{}元", plate, paints, employer, haulage, sum);

    }

上面是一些各种成本的数量和单价信息，同时使用Map存放了一些特有的成本信息，然后下面我们使用成本策略控制类CostsContext装载不同的实现类来进行成本的计算。

可以看到成功的计算出了结果。我们不仅实现了功能，也为后续的扩展打下了良好的基础，例如后续加新的成本类型进行计算，我们只需要继承ICosts实现不同的方法即可，如果某一个类型的计算加了一些条件，这样我们直接在Map里面添加条件，并在相应的实体类进行处理即可，这样让我们的代码结构变得清晰。

四、小结

if-else是我们初学代码时就接触到的代码，伴随我们走过了很多年，但是我们的思维一定要有提高，不要一有需求，就想到if-else，确实，if-else可以快速地实现功能，我们不可否认，但是随着项目复杂度的提高，if-else已经无法满足我们的需求，它会让我们的代码变得臃肿、复杂、结构混乱，所以我们就想到要用其它方法来替代if-else，让我们整个代码结构变得清晰。

上面我们介绍了使用策略模式来代替一堆的if-else，策略模式的优点远不止如此：

1、具有强的扩展性，例如我们扩充新的成本计算，只需要新建类并实现接口即可；

2、易维护和理解，如何我么要看或者修改钢材的成本计算，我们不用到一堆if-else里面去找具体的代码块，我们只需要进去相应的实现类去看就可以，代码清晰明了，没有其他代码的干扰（体现出隔离性），我们维护起来也得心应手；

3、使用策略模式可以让我们代码变得优美，思维能力和架构水平得到提升。