部门端历史数据模块、字段查看模块返回数据层级关系处理

mitao

2024-04-03 321d507a5670cb36c1550445c4795a76c9d7d226

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
package com.ruoyi.system.utils;
 
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.system.domain.TbBasicDataField;
import com.ruoyi.system.domain.TbField;
import com.ruoyi.system.vo.FieldsTreeVO;
 
import java.util.HashMap;
import java.util.List;
import java.util.Map;
 
/**
 * @author mitao
 * @date 2024/4/3
 */
public class FieldBuildUtil {
    /**
     * 构建基础数据字段树形层级
     *
     * @param basicDataFields 基础数据字段
     * @param fieldMap        字段Map
     * @param roots           返回的vo
     */
    public static void buildTreeStructure(List<TbBasicDataField> basicDataFields, Map<Long, TbField> fieldMap, List<FieldsTreeVO> roots) {
        Map<String, FieldsTreeVO> levelTwoMap = new HashMap<>();
        Map<String, FieldsTreeVO> levelThreeMap = new HashMap<>();
 
        for (TbBasicDataField item : basicDataFields) {
            TbField tbField = fieldMap.get(item.getFieldId());
            if (tbField != null) {
                String levelOneCategory = tbField.getLevelOneCategory();
                String levelTwoCategory = tbField.getLevelTwoCategory();
                String levelThreeCategory = tbField.getLevelThreeCategory();
                Long id = tbField.getId();
                String fieldName = tbField.getFieldName();
                String fieldValue = item.getFieldValue();
                FieldsTreeVO field = new FieldsTreeVO();
                field.setId(id);
                field.setName(fieldName);
                field.setValue(fieldValue);
                field.setCategory(Boolean.FALSE);
                FieldsTreeVO levelOneField = null;
                for (FieldsTreeVO root : roots) {
                    if (root.getName().equals(levelOneCategory)) {
                        levelOneField = root;
                        break;
                    }
                }
                if (levelOneField == null) {
                    levelOneField = new FieldsTreeVO();
                    levelOneField.setName(levelOneCategory);
                    roots.add(levelOneField);
                }
 
                if (StringUtils.isNotBlank(levelTwoCategory)) {
                    FieldsTreeVO levelTwoField = levelTwoMap.getOrDefault(levelTwoCategory, null);
                    if (levelTwoField == null) {
                        levelTwoField = new FieldsTreeVO();
                        levelTwoField.setName(levelTwoCategory);
                        levelTwoMap.put(levelTwoCategory, levelTwoField);
                        levelOneField.getChildren().add(levelTwoField);
                    }
 
                    if (StringUtils.isNotBlank(levelThreeCategory)) {
                        FieldsTreeVO levelThreeField = levelThreeMap.getOrDefault(levelThreeCategory, null);
                        if (levelThreeField == null) {
                            levelThreeField = new FieldsTreeVO();
                            levelThreeField.setName(levelThreeCategory);
                            levelThreeMap.put(levelThreeCategory, levelThreeField);
                            levelTwoField.getChildren().add(levelThreeField);
                        }
                        levelThreeField.getChildren().add(field);
                    } else {
                        levelTwoField.getChildren().add(field);
                    }
                } else {
                    levelOneField.getChildren().add(field);
                }
            }
        }
    }
 
    /**
     * 构建字段树形层级
     *
     * @param tbFieldList 字段列表
     */
    public static void buildTreeFromTbFieldList(List<TbField> tbFieldList, List<FieldsTreeVO> roots) {
        Map<String, FieldsTreeVO> levelOneMap = new HashMap<>();
        Map<String, FieldsTreeVO> levelTwoMap = new HashMap<>();
 
        for (TbField tbField : tbFieldList) {
            String levelOneCategory = tbField.getLevelOneCategory();
            String levelTwoCategory = tbField.getLevelTwoCategory();
            String levelThreeCategory = tbField.getLevelThreeCategory();
            String fieldName = tbField.getFieldName();
            Long id = tbField.getId();
            FieldsTreeVO field = new FieldsTreeVO();
            field.setId(id);
            field.setName(fieldName);
            field.setCategory(Boolean.FALSE);
            FieldsTreeVO levelOneField = null;
            for (FieldsTreeVO root : roots) {
                if (root.getName().equals(levelOneCategory)) {
                    levelOneField = root;
                    break;
                }
            }
            if (levelOneField == null) {
                levelOneField = new FieldsTreeVO();
                levelOneField.setName(levelOneCategory);
                roots.add(levelOneField);
            }
 
            if (StringUtils.isNotBlank(levelTwoCategory)) {
                FieldsTreeVO levelTwoField = levelTwoMap.getOrDefault(levelTwoCategory, null);
                if (levelTwoField == null) {
                    levelTwoField = new FieldsTreeVO();
                    levelTwoField.setName(levelTwoCategory);
                    levelOneField.getChildren().add(levelTwoField);
                    levelTwoMap.put(levelTwoCategory, levelTwoField);
                }
 
                if (StringUtils.isNotBlank(levelThreeCategory)) {
                    FieldsTreeVO levelThreeField = new FieldsTreeVO();
                    levelThreeField.setName(levelThreeCategory);
                    levelTwoField.getChildren().add(levelThreeField);
                    levelThreeField.getChildren().add(field);
                } else {
                    levelTwoField.getChildren().add(field);
                }
            } else {
                levelOneField.getChildren().add(field);
            }
        }
    }
}