From b08e66572bfcefea2e10c1213030b7d533c3ae46 Mon Sep 17 00:00:00 2001
From: liyj <1003249715@qq.com>
Date: Fri, 13 Sep 2024 17:07:05 +0800
Subject: [PATCH] 1、网格化平台单点登录功能

---
 dyh-service/dyh-cust/src/main/java/cn/huge/module/ctuser/service/CtUserService.java |  121 ++++++++++++----------------------------
 1 files changed, 37 insertions(+), 84 deletions(-)

diff --git a/dyh-service/dyh-cust/src/main/java/cn/huge/module/ctuser/service/CtUserService.java b/dyh-service/dyh-cust/src/main/java/cn/huge/module/ctuser/service/CtUserService.java
index 08bd4a3..1055308 100644
--- a/dyh-service/dyh-cust/src/main/java/cn/huge/module/ctuser/service/CtUserService.java
+++ b/dyh-service/dyh-cust/src/main/java/cn/huge/module/ctuser/service/CtUserService.java
@@ -594,99 +594,52 @@
 
     /**
      * 根据部门和角色查询
-     * @param deptIdList 部门编号
+     * @param unitId 组织编号
      * @param roleCode 角色代码
      * @return List<SelectTermDTO>
      */
-    public List<CtUser> listUserByDeptRoleList(List<String> deptIdList, String roleCode){
-        return mapper.listUserByDeptRoleList(deptIdList, roleCode);
+    public List<CtUser> listUserByDeptRoleList(String unitId, String roleCode){
+        return mapper.listUserByUnitRoleList(unitId, roleCode);
     }
 
     /**
-     * 按条件分页查询
-     *
-     * @param userId  分页对象
-     * @param roleCode 条件
-     * @return Page
+     * 选择人员
+     * @param userId
+     * @param roleCode
+     * @return
      */
     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);
+        try {
+            // 获取当前登录用户
+            CtUserDTO loginUser = clientGetUserAll(userId);
+            List<SelectTermDTO> result = new ArrayList<>();
+            // 人员条件
+            List<CtUser> ctUsers = null;
+            if (StringUtils.isNotEmpty(roleCode)){
+                ctUsers = listUserByDeptRoleList(loginUser.getUnitId(), roleCode);
+            }else {
+                QueryWrapper<CtUser> ctUserQueryWrapper = new QueryWrapper<>();
+                ctUserQueryWrapper.in("unit_id", loginUser.getUnitId()).select("id", "true_name");
+                ctUsers = this.list(ctUserQueryWrapper);
             }
-            unitSelectTerms.add(unitSelectTerm);
+            SelectTermDTO unitSelectTerm = new SelectTermDTO();
+            unitSelectTerm.setValue(loginUser.getUnitId());
+            unitSelectTerm.setLabel(loginUser.getUnitName());
+            unitSelectTerm.setCheckable(false);
+            List<SelectTermDTO> userSelectTerms = new ArrayList<>();
+            for(CtUser ctUser : ctUsers){
+                SelectTermDTO userSelectTerm = new SelectTermDTO();
+                userSelectTerm.setValue(ctUser.getId());
+                userSelectTerm.setCheckable(true);
+                userSelectTerm.setLabel(ctUser.getTrueName());
+                userSelectTerms.add(userSelectTerm);
+            }
+            unitSelectTerm.setChildren(userSelectTerms);
+            result.add(unitSelectTerm);
+            return result;
+        }catch (Exception e) {
+            log.error("[CtUserService.userChoose]调用失败,异常信息:" + e, e);
+            throw new ServiceException("CtUserService.userChoose", e);
         }
-        return SelectTermUtils.createTreeByFirst(unitSelectTerms, loginUser.getUnitId());
     }
 }

--
Gitblit v1.8.0