花名册树,同级情况下,部门展示在分部前面

pull/264/head
dxfeng 10 months ago
parent 4d88d2f063
commit 6e6c66687b

@ -116,16 +116,13 @@ public class SearchTreeUtil {
Set<String> leafIds = new HashSet<>(); Set<String> leafIds = new HashSet<>();
List<SearchTree> collect = treeList.stream().peek(e -> { List<SearchTree> collect = treeList.stream().peek(e -> {
Set<TreeNode> treeNodes = new LinkedHashSet<>(); Set<TreeNode> treeNodes = new LinkedHashSet<>();
List<TreeNode> nodes = parentMap.get(e.getId());
if (CollectionUtils.isNotEmpty(nodes)) {
treeNodes.addAll(nodes);
}
if (isAdd && CollectionUtils.isNotEmpty(childMap.get(e.getId()))) { if (isAdd && CollectionUtils.isNotEmpty(childMap.get(e.getId()))) {
List<SearchTree> searchTrees = childMap.get(e.getId()).stream().sorted(Comparator.comparing(SearchTree::getOrderNum)).collect(Collectors.toList()); List<SearchTree> searchTrees = childMap.get(e.getId()).stream().sorted(Comparator.comparing(SearchTree::getOrderNum)).collect(Collectors.toList());
treeNodes.addAll(searchTrees); treeNodes.addAll(searchTrees);
} }
if(CollectionUtils.isNotEmpty(treeNodes)){ List<TreeNode> nodes = parentMap.get(e.getId());
treeNodes = treeNodes.stream().sorted(Comparator.comparing(TreeNode::getType).reversed()).collect(Collectors.toCollection(LinkedHashSet::new)); if (CollectionUtils.isNotEmpty(nodes)) {
treeNodes.addAll(nodes);
} }
e.setSubs(new ArrayList<>(treeNodes)); e.setSubs(new ArrayList<>(treeNodes));
leafIds.add(e.getId()); leafIds.add(e.getId());

Loading…
Cancel
Save