今天改同事的代码发现同事的代码中有大量的if else语句。很不美观而且可读性太差。
因为需要给前端一个key,value的列表。这个列表写死。然而同事的代码。
放几张图，大家自己感受。我都不知道以后这个api咋维护，新增一个工站，减少一个工站。
果然新的任务新增4个工站，好吧，开始给代码装扮一下。
首先，写一个枚举类。
package com.smartflow.common.enumpackage;
/**
* @author ：tao
* @date ：Created in 2020/8/1 22:51
* @description：${description}
*/
public enum StationEnum {
/**
*
*/
STATION_PRODUCTION(1,"生产工站"),
STATION_TEST(2,"测试工站"),
STATION_FIX(3,"维修工站"),
STATION_VIRTUAL(4,"虚拟工站"),
STATION_MATERIAL(5,"上料工站"),
STATION_CRASH(6,"清洗工站"),
STATION_PAK(7,"配送工站");
private final int key;
private final String value;
StationEnum(int key, String value) {
this.key = key;
this.value = value;
}
public int getKey() {
return key;
}
public String getValue() {
return value;
}
}
然后这个是获取工站的列表，看吧你几乎看不到重复的new map,或者是魔法值。对比一下上边的是不是很美观。
/**
*获取工站类型列表
* @return 工站类型列表
*/
private List<Map<String,Object>>getStationTypeList()
{
List<Map<String, Object>> stationTypeList = new ArrayList<>();
for (StationEnum stationEnum:StationEnum.values())
{
Map<String, Object> stationType = new HashMap<>(2);
stationType.put("key", stationEnum.getKey());
stationType.put("label", stationEnum.getValue());
stationTypeList.add(stationType);
}
return stationTypeList;
}
接着根据工站id也就是key查找工站名
/**
* 根据key值查找工站名
* @param key key值
* @return 返回攻占名
*/
private String getStationLabel(int key)
{
for (StationEnum stationEnum:StationEnum.values())
{
if (stationEnum.getKey()==key)
{
return stationEnum.getValue();
}
}
return "测试工站";
}
可以看到几行代码就搞定了。而且代码重用性高，可维护性好。想添加工站或者减少工站直接在枚举类型中改变就可以了。不用修改大量的代码。
