设计原则（1）| 开闭原则

蝉沐风

发布于 2022-08-22 13:35:42

5840

发布于 2022-08-22 13:35:42

文章被收录于专栏：蝉沐风的码场

开闭原则是指一个软件实体（模块、类、方法等）应该对扩展开放，对修改关闭

我举一个例子，陀螺是个程序喵，创办了一个生产猫粮的公司——跑码场，手下有个小徒弟叫招财，写了一个下单的逻辑。

/**
 * @author 蝉沐风
 * @description 原始代码
 * @date 2022/2/8
 */
public class PaoMaChangV1 {

    public void order(String flavor) {

        if (flavor.equals("毛血旺")) {
            orderMaoXueWangCatFood();
        } else if (flavor.equals("鱼香肉丝")) {
            orderFishCatFood();
        }
    }

    private void orderMaoXueWangCatFood() {
        System.out.println("售卖一袋「毛血旺」风味猫粮");
    }

    private void orderFishCatFood() {
        System.out.println("售卖一袋「鱼香肉丝」风味猫粮");
    }

}

逻辑本身很简单，核心业务逻辑主要是order()函数，客户需要传入相应的猫粮口味flavor进行下单。

现在跑码场扩展了业务，新增了一种「大肠刺身」口味的猫粮，而且支持用户自定义猫粮购买数量（毕竟这种口味可能会供不应求）。在以上代码的基础上，招财做了如下修改：

/**
 * @author 蝉沐风
 * @description 原始代码功能扩展
 * @date 2022/2/8
 */
public class PaoMaChangV1Expand {

    public void order(String flavor, Integer count) {

        if (flavor.equals("毛血旺")) {
            orderMaoXueWangCatFood(count);
        } else if (flavor.equals("鱼香肉丝")) {
            orderFishCatFood(count);
        }
        // 更改1：添加口味的逻辑判断
        else if (flavor.equals("大肠刺身")) {
            orderDaChangFood(count);
        }
    }

    private void orderMaoXueWangCatFood(Integer count) {
        System.out.println("售卖" + count + "袋「毛血旺」风味猫粮");
    }

    private void orderFishCatFood(Integer count) {
        System.out.println("售卖" + count + "袋「鱼香肉丝」风味猫粮");
    }

    // 更改2：添加售卖逻辑
    private void orderDaChangFood(Integer count) {
        System.out.println("售卖" + count + "一袋「大肠刺身」风味猫粮");
    }
}

这种修改方式确实能解决目前的业务问题，但同时也存在很多问题。

首先，修改了order()方法，添加了一个参数，相应的客户端调用必须修改；其次，每当有新的口味猫粮产品诞生时，都必须在order()方法中添加口味的判断，同时需要添加该产品的售卖逻辑。这些操作都是通过「修改」来实现新功能的，不符合「开闭原则」。

如果我们要遵循「开闭原则」，必须对修改关闭，对扩展开放。

我们重构一下初始代码，主要做以下两方面的修改：

创建CatFood基类，然后创建对应口味的猫粮继承基类；
将每种口味猫粮的售卖逻辑写在具体类中。
修改客户调用的order方法

/**
 * @author 蝉沐风
 * @description 猫粮基类
 * @date 2022/2/8
 */
public abstract class CatFood {
   
    public abstract void order();

}

/**
 * @author 蝉沐风
 * @description 「毛血旺」猫粮
 * @date 2022/2/8
 */
public class MaoXueWangCatFood extends CatFood {

    @Override
    public void order() {
        System.out.println("售卖一袋「毛血旺」风味猫粮");
    }
}


/**
 * @author 蝉沐风
 * @description 「鱼香肉丝」猫粮
 * @date 2022/2/8
 */
public class FishCatFood extends CatFood {

    @Override
    public void order() {
        System.out.println("售卖一袋「鱼香肉丝」风味猫粮");
    }

}

order()方法修改如下

/**
 * @author 蝉沐风
 * @description 遵循「开闭原则」之后的代码
 * @date 2022/2/8
 */
public class PaoMaChangV2 {

    public void order(CatFood catFood) {
       catFood.order();
    }

}

重构之后的客户端调用方式如下

/**
 * @author 蝉沐风
 * @description 客户端调用
 * @date 2022/2/8
 */
public class ClientV2 {
    public static void main(String[] args) {
        PaoMaChangV2 paoMaChang  = new PaoMaChangV2();

        // 创建对应口味的猫粮
        FishCatFood fish = new FishCatFood();
        paoMaChang.order(fish);
    }
}

现在我们再来看，基于重构之后的代码，我们要实现刚才讲到的业务需求，我们需要进行怎样的改动。主要的修改内容有如下：

CatFood基类中添加属性count，为子类添加构造函数；
添加新类DaChangCatFood；

扩展之后的代码如下

/**
 * @author 蝉沐风
 * @description 猫粮类
 * @date 2022/2/8
 */
public abstract class CatFood {
    
    //订购数量
    private Integer count;

    public abstract void order();

    public Integer getCount() {
        return count;
    }

    public void setCount(Integer count) {
        this.count = count;
    }

    public CatFood(Integer count) {
        this.count = count;
    }

    public CatFood() {
    }
}

/**
 * @author 蝉沐风
 * @description 「毛血旺」猫粮
 * @date 2022/2/8
 */
public class MaoXueWangCatFood extends CatFood {

    public MaoXueWangCatFood(Integer count) {
        this.setCount(count);
    }

    @Override
    public void order() {
        System.out.println("售卖" + this.getCount() + "袋「毛血旺」风味猫粮");
    }
}

/**
 * @author 蝉沐风
 * @description 「鱼香肉丝」猫粮
 * @date 2022/2/8
 */
public class FishCatFood extends CatFood {

    public FishCatFood(Integer count) {
        this.setCount(count);
    }

    @Override
    public void order() {
        System.out.println("售卖" + this.getCount() + "袋「鱼香肉丝」风味猫粮");
    }

}

/**
 * @author 蝉沐风
 * @description 「大肠刺身」猫粮
 * @date 2022/2/8
 */
public class DaChangCatFood extends CatFood {

    public DaChangCatFood(Integer count) {
        this.setCount(count);
    }

    @Override
    public void order() {
        System.out.println("售卖" + this.getCount() + "袋「大肠刺身」风味猫粮");
    }

}

客户端调用方式变为

public class ClientV2 {
    public static void main(String[] args) {
        PaoMaChangV2 paoMaChang  = new PaoMaChangV2();

        // 创建对应口味的猫粮
        DaChangCatFood dachang = new DaChangCatFood(2);
        paoMaChang.order(dachang);
    }
}