From f91e3a2507c6972fb15d7d91f0d0380dd4be0bef Mon Sep 17 00:00:00 2001 From: luxiaotao1123 <t1341870251@63.com> Date: 星期一, 19 九月 2022 08:37:56 +0800 Subject: [PATCH] # --- src/main/java/com/zy/crm/common/utils/TreeUtils.java | 21 ++++++++++++++++----- 1 files changed, 16 insertions(+), 5 deletions(-) diff --git a/src/main/java/com/zy/crm/common/utils/TreeUtils.java b/src/main/java/com/zy/crm/common/utils/TreeUtils.java index 7c534bb..3f5872a 100644 --- a/src/main/java/com/zy/crm/common/utils/TreeUtils.java +++ b/src/main/java/com/zy/crm/common/utils/TreeUtils.java @@ -7,8 +7,10 @@ import com.zy.crm.manager.service.NodeService; import com.zy.crm.manager.service.TagService; import com.zy.crm.system.entity.Dept; +import com.zy.crm.system.entity.Role; import com.zy.crm.system.entity.User; import com.zy.crm.system.service.DeptService; +import com.zy.crm.system.service.RoleService; import com.zy.crm.system.service.UserService; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.cache.annotation.Cacheable; @@ -31,6 +33,8 @@ private DeptService deptService; @Autowired private UserService userService; + @Autowired + private RoleService roleService; /******************************** 褰掔被鏍� *********************************/ @@ -176,7 +180,7 @@ * 鑾峰彇鏍戝浘鏁版嵁缁撴瀯 */ @Cacheable(cacheNames="deptUserTree",key="#id") - public ArrayList<Map<String, Object>> getDeptUserTree(String id, Long hostId){ + public ArrayList<Map<String, Object>> getDeptUserTree(String id, Long hostId, Long userId){ ArrayList<Map<String, Object>> result = new ArrayList<>(); Dept dept = deptService.selectById(id); // 涓昏妭鐐� @@ -188,12 +192,12 @@ // map.put("icon", "layui-icon layui-icon-star-fill"); List<Map<String, Object>> childrens = new ArrayList<>(); map.put("children", childrens); - dealUserDept(dept, childrens, hostId); + dealUserDept(dept, childrens, hostId, userId); result.add(map); return result; } - public void dealUserDept(Dept parent, List<Map<String, Object>> list, Long hostId) { + public void dealUserDept(Dept parent, List<Map<String, Object>> list, Long hostId, Long userId) { Wrapper<Dept> wrapper = new EntityWrapper<Dept>() .eq("parent_id", parent.getId()) .eq("status", 1); @@ -202,7 +206,14 @@ } List<Dept> depts = deptService.selectList(wrapper); - List<User> users = userService.getUserByDept(hostId, parent.getId()); + List<User> users = new ArrayList<>(); + Role role = roleService.selectById(userService.selectById(userId).getRoleId()); + if (role.getCode().equals("salesman")) { + users.add(userService.selectById(userId)); + } else { + users = userService.getUserByDept(hostId, parent.getId()); + } + for (User user : users) { Map<String, Object> map = new HashMap<>(); map.put("title", user.getNickname()); @@ -221,7 +232,7 @@ // map.put("icon", "layui-icon layui-icon-star-fill"); List<Map<String, Object>> childrens = new ArrayList<>(); map.put("children", childrens); - dealUserDept(dept, childrens, hostId); + dealUserDept(dept, childrens, hostId, userId); list.add(map); } } -- Gitblit v1.9.1