| | |
|---|
| | | public List<TreeNode> getSecurityApplyTree(UserVO user) { |
|---|
| | | Map<Long,TreeNode> map = baseMapper.getSecurityApplyTree(user); |
|---|
| | | List<TreeNode> tree = new ArrayList<>(); |
|---|
| | | map.forEach((id,treeNode) ->{ |
|---|
| | | if (map.containsKey(treeNode.getParentId())){ |
|---|
| | | map.get(treeNode.getParentId()).getChildren().add(treeNode); |
|---|
| | | }else { |
|---|
| | | tree.add(treeNode); |
|---|
| | | } |
|---|
| | | }); |
|---|
| | | if (map.size()>1){ |
|---|
| | | map.forEach((id,treeNode) ->{ |
|---|
| | | if (map.containsKey(treeNode.getParentId())){ |
|---|
| | | map.get(treeNode.getParentId()).getChildren().add(treeNode); |
|---|
| | | }else { |
|---|
| | | tree.add(treeNode); |
|---|
| | | } |
|---|
| | | }); |
|---|
| | | } |
|---|
| | | return tree; |
|---|
| | | } |
|---|
| | | |
|---|
