|
@@ -801,21 +801,19 @@ public class DataResourceFacade implements IDataResourceFacade {
|
|
|
}
|
|
|
} else {
|
|
|
// 如果为叶子节点,查找对应的表和列
|
|
|
+ // 为字段分类添加表
|
|
|
// 如果查找不到表和列则移除
|
|
|
String id = childVo.getId();
|
|
|
String classifyCode = DataResourceEnum.DATA_RESOURCE_CLASSIFY.getCode();
|
|
|
- if (id.startsWith(classifyCode)) {
|
|
|
- String replace = id.replace(classifyCode + CommonCons.ID_SEPARATOR, "");
|
|
|
- List<AppDataResourceInfo> dataTabInfo = tabInfosMap.get(replace);
|
|
|
- if (dataTabInfo == null) {
|
|
|
- iterator.remove();
|
|
|
- } else {
|
|
|
- List<DataResourceTreeVo> tableVos = convertTableToTreeVo(childVo.getId(), dataTabInfo, null, dataResourceEnum);
|
|
|
- childVo.setTreeNode(true);
|
|
|
- childVo.setChild(tableVos);
|
|
|
- }
|
|
|
+ String replace = id.replace(classifyCode + CommonCons.ID_SEPARATOR, "");
|
|
|
+ List<AppDataResourceInfo> dataTabInfo = tabInfosMap.get(replace);
|
|
|
+ if (dataTabInfo == null) {
|
|
|
+ iterator.remove();
|
|
|
+ } else {
|
|
|
+ List<DataResourceTreeVo> tableVos = convertTableToTreeVo(childVo.getId(), dataTabInfo, null, dataResourceEnum);
|
|
|
+ childVo.setTreeNode(true);
|
|
|
+ childVo.setChild(tableVos);
|
|
|
}
|
|
|
-
|
|
|
}
|
|
|
}
|
|
|
return CollectionUtils.isEmpty(child);
|
|
@@ -824,7 +822,8 @@ public class DataResourceFacade implements IDataResourceFacade {
|
|
|
private List<DataResourceTreeVo> convertTableToTreeVo(String pid, List<AppDataResourceInfo> dataTabInfo, Map<String, List<AppDataItemInfo>> colInfosMap, DataResourceEnum resourceEnum) {
|
|
|
return dataTabInfo.stream().map(e -> {
|
|
|
DataResourceTreeVo vo = new DataResourceTreeVo();
|
|
|
- vo.setId(e.getId());
|
|
|
+ // 因为表可能属于多个字段分类,前端树结构需要唯一id
|
|
|
+ vo.setId(pid + CommonCons.ID_SEPARATOR + e.getId());
|
|
|
vo.setPId(pid);
|
|
|
vo.setLabel(e.getTableName());
|
|
|
vo.setClassifyCode(resourceEnum.getCode());
|
|
@@ -838,6 +837,9 @@ public class DataResourceFacade implements IDataResourceFacade {
|
|
|
vo.setChild(Collections.emptyList());
|
|
|
vo.setTreeNode(false);
|
|
|
}
|
|
|
+ } else {
|
|
|
+ vo.setChild(Collections.emptyList());
|
|
|
+ vo.setTreeNode(false);
|
|
|
}
|
|
|
return vo;
|
|
|
}).collect(Collectors.toList());
|
|
@@ -850,6 +852,7 @@ public class DataResourceFacade implements IDataResourceFacade {
|
|
|
vo.setPId(pid);
|
|
|
vo.setLabel(e.getColumnName());
|
|
|
vo.setClassifyCode(resourceEnum.getCode());
|
|
|
+ vo.setTreeNode(false);
|
|
|
return vo;
|
|
|
}).collect(Collectors.toList());
|
|
|
}
|