java switch重构

2019-04-22  本文已影响0人  Ray昱成

背景

原代码中存在一个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里面,引发错误

分析解决方法

使用多态、策略模式

public class ConvertA<A>{
VO convert(A model){ ... 省略};
}

使用函数式接口Function

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);
    }
}
AbstractModel model;  // 抽象模型定义省略
VO vo=Convert.dataMappingVO(model);

参考:https://codeday.me/bug/20190109/489447.html

上一篇 下一篇

猜你喜欢

热点阅读