背景
原代码中存在一个switch语句,用于model转VO操作，因为model是抽象类要转成不同的VO，所以用了大量的switch语句
AbstractModel model; // 抽象模型
String type=model.getType();
//根据不同类型转换成视图对象
switch(type){
case "A":
convertA(model);
case "B":
convertB(model);
.......
case "N":
convertN(model);
}
此类代码可读性较差，而且稍有不慎可能会走到下一个case里面，引发错误
分析解决方法
使用多态、策略模式
- 定义convert接口和策略实现
public interface BaseConvert<T extends AbstractModel>{
VO convert(T model);
}
public class ConvertA<A>{
VO convert(A model){ ... 省略};
}
调用具体策略
Map<String,BaseConvert> map=new HashMap(); // map初始化省略
VO vo=map.get(model.getType()).convert;这种方法的问题是我必须创建大量的类/匿名类
使用函数式接口Function
- 定义Convert类
public final class Convert{
/** 默认map容量*/
private final static int INIT_MAPPING_CAPACITY = 32;
/** type映射VO转换方法*/
private static final Map<String, Function<AbstractModel, VO>> map;
/**
* 初始化mapping
*/
static {
map=new HashMap<>(INIT_MAPPING_CAPACITY);
// 当类型是A 调用convertA
map.put("A", (model) ->convertA(model));
map.put("B", (model) ->convertB(model));
.......
map.put("N", (model) ->convertN(model));
}
// 声明convertA、convertB 略过
/**
* model转VO
* @param model
* @return VO
*/
public static VO dataMappingVO(AbstractModel model) {
return Optional.ofNullable(map.get(model.getType)).orElse((model) -> null).apply(model);
}
}
- 调用convert
AbstractModel model; // 抽象模型定义省略
VO vo=Convert.dataMappingVO(model);
- 此种方式既不用写太多的类，可读性也会更高。
