最近做一个后台管理项目,前端需要多级分类级联展示,后端需要把二维的配置数据转换成树形结构。琢磨了半天找到了一个简单的方法。
1、模型定义
@Data
public class Options{
private String value;
private String label;
private List<Options> children;
private Integer parentKey;
public Options(String value, String label,Integer parentKey) {
this.value = value;
this.label = label;
this.parentKey = parentKey;
}
}
2、模拟数据
二维数据转Map这里不再赘述。
Map<Integer, Options> optionsMap = new HashMap<>();
optionsMap.put(1,new Options("DT","电梯",0));
optionsMap.put(2,new Options("JT","扶梯",1));
optionsMap.put(3,new Options("KT","客梯",1));
optionsMap.put(4,new Options("FJ","风机",0));
optionsMap.put(5,new Options("CF","出风",4));
optionsMap.put(6,new Options("TF","通风",5));
optionsMap.put(7,new Options("PY","排烟",5));
3、数据转换
List<Options> optionsList =new ArrayList<>();
for (Integer key : optionsMap.keySet()) {
if(optionsMap.get(key).parentKey.equals(0)){
optionsList.add(optionsMap.get(key));
}else{
Integer parentKey = optionsMap.get(key).parentKey;
if(optionsMap.get(parentKey).getChildren() == null){
optionsMap.get(parentKey).setChildren(new ArrayList<>());
}
optionsMap.get(parentKey).getChildren().add(optionsMap.get(key));
}
}