forked from gzzfw/backEnd/gz-dyh

dyh-service/dyh-cust/src/main/java/cn/huge/module/ctuser/service/CtUserService.java
@@ -600,4 +600,102 @@
            ctDeptService.updateCtDept(ctDept);
        }
    }
    /**
     * 根据部门和角色查询
     * @param deptIdList 部门编号
     * @param roleCode 角色代码
     * @return List<SelectTermDTO>
     */
    public List<CtUser> listUserByDeptRoleList(List<String> deptIdList, String roleCode){
        return mapper.listUserByDeptRoleList(deptIdList, roleCode);
    }
    /**
     * 按条件分页查询
     *
     * @param userId  分页对象
     * @param roleCode 条件
     * @return Page
     */
    public List<SelectTermDTO> userChoose(String userId, String roleCode) {
        // 获取当前登录用户
        CtUserDTO loginUser = clientGetUserAll(userId);
        // 组织条件
        QueryWrapper<CtUnit> ctUnitQueryWrapper = new QueryWrapper<>();
        if (RoleBaseEnum.checkAdminPower(loginUser)){
            ctUnitQueryWrapper.eq("cust_id", loginUser.getCustId()).select("id", "unit_name", "parent_id");
        }else {
            ctUnitQueryWrapper.eq("id", loginUser.getUnitId()).select("id", "unit_name", "parent_id");
        }
        List<CtUnit> ctUnitList = ctUnitService.list(ctUnitQueryWrapper);
        List<SelectTermDTO> unitSelectTerms = new ArrayList<>();
        List<String> unitIdList= ctUnitList.stream().map(CtUnit::getId).collect(Collectors.toList());
        // 部门条件
        QueryWrapper<CtDept> ctDeptQueryWrapper = new QueryWrapper<>();
        ctDeptQueryWrapper.in("unit_id", unitIdList).select("id", "unit_id", "name", "parent_id");
        List<CtDept> ctDeptList = ctDeptService.list(ctDeptQueryWrapper);
        List<String> deptIdList = ctDeptList.stream().map(CtDept::getId).collect(Collectors.toList());
        // 人员条件
        List<CtUser> ctUsers = null;
        if (StringUtils.isNotEmpty(roleCode)){
            ctUsers = listUserByDeptRoleList(deptIdList, roleCode);
        }else {
            QueryWrapper<CtUser> ctUserQueryWrapper = new QueryWrapper<>();
            ctUserQueryWrapper.in("dept_id", deptIdList).select("id", "true_name", "dept_id");
            ctUsers = this.list(ctUserQueryWrapper);
        }
        List<String> userIdList = ctUsers.stream().map(CtUser::getId).collect(Collectors.toList());
        QueryWrapper<CtUserole> ctUserQueryWrapper = new QueryWrapper<>();
        ctUserQueryWrapper.in("user_id", userIdList).select("user_id", "role_name");
        List<CtUserole> roleList = ctUseroleService.list(ctUserQueryWrapper);
        for(CtUnit ctUnit: ctUnitList){
            SelectTermDTO unitSelectTerm = new SelectTermDTO();
            unitSelectTerm.setValue(ctUnit.getId());
            unitSelectTerm.setLabel(ctUnit.getUnitName());
            unitSelectTerm.setParentId(ctUnit.getParentId());
            unitSelectTerm.setCheckable(false);
            List<SelectTermDTO> deptSelectTerms = new ArrayList<>();
            for(CtDept ctDept : ctDeptList){
                if(ctUnit.getId().equals(ctDept.getUnitId())){
                    SelectTermDTO deptSelectTerm = new SelectTermDTO();
                    deptSelectTerm.setValue(ctDept.getId());
                    deptSelectTerm.setLabel(ctDept.getName());
                    deptSelectTerm.setParentId(ctDept.getParentId());
                    deptSelectTerm.setCheckable(false);
                    if (CollectionUtils.isNotEmpty(ctUsers)) {
                        List<SelectTermDTO> userSelectTerms = new ArrayList<>();
                        for(CtUser ctUser : ctUsers){
                            if(ctUser.getDeptId().equals(ctDept.getId())){
                                List<CtUserole> ctUseroleList = new ArrayList<>();
                                for(CtUserole ctUserole: roleList){
                                    if(ctUser.getId().equals(ctUserole.getUserId())){
                                        ctUseroleList.add(ctUserole);
                                    }
                                }
                                SelectTermDTO userSelectTerm = new SelectTermDTO();
                                userSelectTerm.setValue(ctUser.getId());
                                userSelectTerm.setCheckable(true);
                                String roleName = ctUseroleList.stream().map(CtUserole::getRoleName).collect(Collectors.joining(","));
                                userSelectTerm.setLabel(ctUser.getTrueName()+"("+roleName+")");
                                userSelectTerms.add(userSelectTerm);
                            }
                        }
                        List<SelectTermDTO> users = SelectTermUtils.createTreeByRoot(userSelectTerms);
                        deptSelectTerm.setChildren(users);
                    }
                    deptSelectTerms.add(deptSelectTerm);
                }
                List<SelectTermDTO> depts = SelectTermUtils.createTreeByRoot(deptSelectTerms);
                unitSelectTerm.setChildren(depts);
            }
            unitSelectTerms.add(unitSelectTerm);
        }
        return SelectTermUtils.createTreeByFirst(unitSelectTerms, loginUser.getUnitId());
    }
}