策略模式+工厂模式+map 整理 项目中的switch case/if
转载:文章大佬写的文章 (如有侵权,请联系我,会立即删除)
接手新任务:接入第三家存证机构,看之前的代码使用了swith case判断使用哪家存证机构,每家存证机构的实现逻辑不一样 代码的坏味道:多层swich case。多层swich case不好维护,是时候应该重构了,
优化前的代码
为了便于理解,举个没有业务逻辑的例子,基于这个例子上进行优化。 现在是12:47,举个饭后吃水果的例子哈哈哈(逃 假设我们可以选择的水果有香蕉、西瓜和苹果。吃香蕉的话我们得先剥皮,吃西瓜的话得先用水果刀切一下,如果是苹果的话就直接吃了。将这个场景转化为代码:
public class EatFruit {
private static final String APPLE = "apple";
private static final String BANANA = "banana";
private static final String WATERMELON = "watermelon";
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
//选择的水果种类
String fruitType = scanner.nextLine();
switch (fruitType) {
case APPLE:
eatApple();
break;
case BANANA:
eatBanana();
break;
case WATERMELON:
eatWatermelon();
break;
}
}
private static void eatBanana() {
System.out.println("吃香蕉了,需要先剥下皮");
}
private static void eatApple() {
System.out.println("是苹果,可以直接吃");
}
private static void eatWatermelon() {
System.out.println("吃西瓜了,但是还得弄把水果刀切一下先");
}
}
这个例子代码量不是很大,但是实际项目中的场景肯定没有这么简单,多重swich case不好维护,并且万一又加了一样水果,还得继续加一层case…
赶紧优化
很容易想到策略模式(简单的理解就是多态),水果都有吃的一个动作,但是每种水果的吃法不一样
使用策略模式进行优化
Fruit.java
public interface Fruit {
void eat();
}
Apple.java
public class Apple implements Fruit {
@Override
public void eat() {
System.out.println("是苹果,可以直接吃");
}
}
Banana.java
public class Banana implements Fruit {
@Override
public void eat() {
System.out.println("吃香蕉了,需要先剥下皮");
}
}
Watermelon.java
public class Watermelon implements Fruit {
@Override
public void eat() {
System.out.println("吃西瓜了,但是还得弄把水果刀切一下先");
}
}
但是发现即使用了策略模式也难逃类型判断 EatFruit.java
public class EatFruit {
private static final String APPLE = "apple";
private static final String BANANA = "banana";
private static final String WATERMELON = "watermelon";
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
//选择的水果种类
String fruitType = scanner.nextLine();
Fruit fruit = null;
switch (fruitType) {
case APPLE:
fruit = new Apple();
break;
case BANANA:
fruit = new Banana();
break;
case WATERMELON:
fruit = new Watermelon();
break;
}
fruit.eat();
}
}
使用策略模式具有良好的扩展性,兵来将挡,再给我添加十种水果都不怕,水果种类太多会使水果的实现类暴增。现在来一种水果无需在原先的业务类(EatFruit)里修改很多的代码逻辑,只要实现接口并加个条件判断就好了。但是使用策略模式时,我们需要知道具体的实现类,具体的实现类需要对外暴露
使用工厂模式
将类型判断放到工厂类中 工厂模式:在创建对象时不会对客户端暴露创建逻辑,并且是通过使用一个共同的接口来指向新创建的对象,这里共同的接口就是Fruit。粗暴的讲,工厂模式就是把创建同一类型对象逻辑写在了一个方法里 FruitFactory .java
public class FruitFactory {
public static Fruit getFruit(String fruitType) {
if ("apple".equals(fruitType)) {
return new Apple();
}
if ("banana".equals(fruitType)) {
return new Banana();
}
return new Watermelon();
}
}
EatFruit .java
public class EatFruit {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
//选择的水果种类
String fruitType = scanner.nextLine();
Fruit fruit = FruitFactory.getFruit(fruitType);
fruit.eat();
}
}
到现在为止EatFruit中的业务代码已经很清晰了。 使用工厂模式具有良好的封装性,这下妈妈再也不用关系创建类的过程,甚至连创建的实际类的都无需关心,实现了解耦,实际类的修改变化都不会影响上层业务
但是工厂类中还是有很多if的,革命尚未成功,仍需要进行优化。
map奥特曼来了
FruitFactory.java
public class FruitFactory {
private static Map<String, Fruit> fruitMap = new HashMap<>();
static {
fruitMap.put("apple", new Apple());
fruitMap.put("banana", new Banana());
fruitMap.put("watermelon", new Watermelon());
}
public static Fruit getFruit(String fruitType) {
return fruitMap.get(fruitType);
}
}
通过使用map,fruitType与进行一一映射,和条件判断说byebye~,代码也很优雅 最终项目结构类图:
有网友评论可以用反射替代map,如果传进来的是Class类型的值就可以用反射。用反射的好处是在添加多种水果,工厂类的代码都无需进行变动 通过反射替代map: FruitFactory2.java
public class FruitFactory2 {
public static <T extends Fruit> T getFruit(Class<T> fruitClass) throws IllegalAccessException, InstantiationException {
return fruitClass.newInstance();
}
}