java switch重构

背景

原代码中存在一个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{
    VO convert(T model);
    }

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

  • 调用具体策略
    Map 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> 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);
  • 此种方式既不用写太多的类,可读性也会更高。

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

你可能感兴趣的:(java switch重构)