二维数据转树形结构

最近做一个后台管理项目，前端需要多级分类级联展示，后端需要把二维的配置数据转换成树形结构。琢磨了半天找到了一个简单的方法。

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));
            }
        }

4、前端展示效果