| | |
|---|
| | | */ |
|---|
| | | @Override |
|---|
| | | public List<TreeNode> getSecurityApplyTree(UserVO user) { |
|---|
| | | Map<Long, TreeNode> map = baseMapper.getSecurityApplyTree(user); |
|---|
| | | List<TreeNode> tree = new ArrayList<>(); |
|---|
| | | if (map.size() > 1) { |
|---|
| | | map.forEach((id, treeNode) -> { |
|---|
| | | if (map.containsKey(treeNode.getParentId())) { |
|---|
| | | map.get(treeNode.getParentId()).getChildren().add(treeNode); |
|---|
| | | } else { |
|---|
| | | tree.add(treeNode); |
|---|
| | | // 总的 |
|---|
| | | List<TreeNode> allList = new ArrayList<>(); |
|---|
| | | // 考试未通过 |
|---|
| | | List<TreeNode> notList = new ArrayList<>(); |
|---|
| | | // 新申报报名的 |
|---|
| | | List<TreeNode> newList = new ArrayList<>(); |
|---|
| | | // 查询 |
|---|
| | | List<TreeNode> treeNodeList = baseMapper.getSecurityApplyTree(user); |
|---|
| | | // 遍历 |
|---|
| | | if (treeNodeList.size()>0){ |
|---|
| | | for (TreeNode treeNode : treeNodeList) { |
|---|
| | | if (treeNode.getType()==1){ |
|---|
| | | notList.add(treeNode); |
|---|
| | | } |
|---|
| | | }); |
|---|
| | | if (treeNode.getType()==2){ |
|---|
| | | newList.add(treeNode); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | } |
|---|
| | | return tree; |
|---|
| | | if (newList.size()>0){ |
|---|
| | | // 新申报报名的 |
|---|
| | | TreeNode treeNode = new TreeNode(); |
|---|
| | | treeNode.setName("新申报名单"); |
|---|
| | | treeNode.setChildren(newList); |
|---|
| | | allList.add(treeNode); |
|---|
| | | } |
|---|
| | | if (notList.size()>0){ |
|---|
| | | // 考试未通过 |
|---|
| | | TreeNode treeNode = new TreeNode(); |
|---|
| | | treeNode.setName("未通过考试名单"); |
|---|
| | | treeNode.setChildren(notList); |
|---|
| | | allList.add(treeNode); |
|---|
| | | } |
|---|
| | | // 返回 |
|---|
| | | return allList; |
|---|
| | | } |
|---|
| | | |
|---|
| | | /** |
|---|
