package com.ydtech.modules.admin.service.impl; import cn.hutool.core.bean.BeanUtil; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.ydtech.core.page.HttpResult; import com.ydtech.core.page.PageRequest; import com.ydtech.core.page.PageResult; import com.ydtech.exception.SystemException; import com.ydtech.modules.admin.dao.SysAreaMapper; import com.ydtech.modules.admin.dao.SysDeptMapper; import com.ydtech.modules.admin.dao.SysUserDeptRelationshipMapper; import com.ydtech.modules.admin.dao.SysUserMapper; import com.ydtech.modules.admin.model.SysArea; import com.ydtech.modules.admin.model.SysDept; import com.ydtech.modules.admin.model.dto.QxFrontlineAgentDto; import com.ydtech.modules.admin.model.po.SysUserDeptRelationship; import com.ydtech.modules.admin.model.report.financialReceivables.FinancialReceivablesMonthVo; import com.ydtech.modules.admin.model.vo.*; import com.ydtech.modules.admin.model.SysUser; import com.ydtech.modules.admin.model.po.SysUserTreeShowLevel; import com.ydtech.modules.admin.service.SysAreaService; import com.ydtech.modules.admin.service.SysDeptService; import com.ydtech.modules.admin.service.SysUserTreeShowLevelService; import com.ydtech.modules.base.controller.BaseController; import com.ydtech.modules.protocol.utils.AssertionUtils; import com.ydtech.utils.StringUtils; import io.lettuce.core.ScriptOutputType; import lombok.extern.slf4j.Slf4j; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; import org.springframework.util.ObjectUtils; import java.util.*; import java.util.function.Function; import java.util.stream.Collectors; @Service @Slf4j public class SysDeptServiceImpl extends ServiceImpl implements SysDeptService { @Autowired private SysDeptMapper sysDeptMapper; @Autowired private SysAreaService sysAreaService; @Autowired private SysAreaMapper sysAreaMapper; @Autowired private SysUserTreeShowLevelService sysUserTreeShowLevelService; @Autowired private SysUserMapper sysUserMapper; @Autowired private SysUserDeptRelationshipMapper sysUserDeptRelationshipMapper; /** * 根据用户权限和指定部门ID，查询并构建部门树。 * * @param deptId 部门ID，用于查询特定部门及其子部门，如果为空或特定值，则查询用户所在部门及其子部门。 * @param managementSource 管理范围来源标识，用于不同场景下的部门树构建。 * @return 返回符合用户权限和管理范围的部门树列表。 */ @Override public List findTree(String deptId, String managementSource) { // 获取当前操作用户 SysUser user = BaseController.getUser(); String userId = user.getId(); // 根据用户ID查询用户的部门树展示级别 LambdaQueryWrapper lambdaQueryWrapper = new LambdaQueryWrapper<>(); lambdaQueryWrapper.eq(SysUserTreeShowLevel::getUserId, userId); SysUserTreeShowLevel treeShowLevel = sysUserTreeShowLevelService.getOne(lambdaQueryWrapper); List treeByShowLevel = new ArrayList<>(); //如果当前用户未配置树展示等级，则默认为查询全部 if (Objects.isNull(treeShowLevel)) { treeShowLevel = new SysUserTreeShowLevel(); treeShowLevel.setShowLevel("subordinate"); } // 如果指定的部门ID为空或特定值，则使用当前用户所在部门ID if (null == deptId || "".equals(deptId) || "undefined".equals(deptId)) { if (!treeShowLevel.getShowLevel().equals("all")) { deptId = user.getDeptId(); } else { deptId = "9"; } } // 根据用户的部门树展示级别查询相应的部门树 switch (treeShowLevel.getShowLevel()) { case "all": treeByShowLevel = findTreeByShowLevel(deptId, managementSource, "all"); break; case "same": treeByShowLevel = findTreeByShowLevel(deptId, managementSource, "same"); break; case "subordinate": treeByShowLevel = findTreeByShowLevel(deptId, managementSource, "subordinate"); break; } return treeByShowLevel; } private List findTreeByShowLevel(String deptId, String managementSource, String showLevel) { SysDept sysDept = getById(deptId); LambdaQueryWrapper lqw = new LambdaQueryWrapper<>(); lqw.eq(SysDept::getDelFlag, 0); lqw.likeRight(SysDept::getId, deptId); lqw.orderByAsc(SysDept::getOrderNum); List sysDeptList = list(lqw); //判断机构来源条件 进行遍历 if (null != managementSource && !"undefined".equals(managementSource) && !"".equals(managementSource)) { lqw.eq(SysDept::getManagementSource, managementSource); //条件 List conditionList = list(lqw); Map> conditionMap = conditionList.stream().collect(Collectors.groupingBy(SysDept::getParentId)); Map> map = conditionList.stream().collect(Collectors.groupingBy(SysDept::getParentId)); //先查第6层 conditionList.forEach(i -> i.setChildren(map.getOrDefault(i.getId(), new ArrayList<>())) ); List result = new ArrayList<>(); //再查前4层 int level = 5; while (level > 0) { List resultTmp = recursion(sysDeptList, conditionMap); if (resultTmp.size() == 0) { break; } else { int finalLevel = level; resultTmp = resultTmp.stream().filter(x -> x.getComlevel().equals(String.valueOf(finalLevel))).collect(Collectors.toList()); result.clear(); result.addAll(resultTmp); } conditionMap = result.stream().collect(Collectors.groupingBy(SysDept::getParentId)); level--; } sysDeptList = result; } else { Map> map = sysDeptList.stream().collect(Collectors.groupingBy(SysDept::getParentId)); sysDeptList.forEach(i -> { i.setChildren(map.getOrDefault(i.getId(), new ArrayList<>())); i.setDeptCount(i.getChildren().size()); }); sysDeptList = map.get(sysDept.getParentId()); } if (showLevel.equals("same")) { if (null != managementSource && !"undefined".equals(managementSource) && !"".equals(managementSource)) { if (sysDept.getManagementSource().equals(managementSource)) { sysDeptList.clear(); sysDeptList.add(sysDept); } else { return new ArrayList<>(); } } else { sysDeptList.clear(); sysDeptList.add(sysDept); } } return sysDeptList; } private List recursion(List sysDeptList, Map> conditionMap) { List result = new ArrayList<>(); sysDeptList.forEach(i -> conditionMap.forEach((key, value) -> { if (i.getId().equals(key)) { if (i.getChildren() == null) { i.setChildren(new ArrayList<>()); } i.getChildren().addAll(value); //处理总部的排序为空造成的空指针报错处理 add by hxl i.setOrderNum(i.getOrderNum() == null ? 0 : i.getOrderNum()); result.add(i); } }) ); result.sort(Comparator.comparing(SysDept::getOrderNum)); return result; } @Override public List getUserList(String deptId) { SysDept sysDept = getById(deptId); LambdaQueryWrapper lqw = new LambdaQueryWrapper<>(); lqw.eq(SysDept::getDelFlag, 0); lqw.likeRight(SysDept::getId, deptId); List sysDeptList = list(lqw); Map> map = sysDeptList.stream().collect(Collectors.groupingBy(SysDept::getParentId)); return map.get(sysDept.getParentId()); } @Override @Transactional(rollbackFor = Exception.class) public SysDeptVo addDept(SysDeptVo sysDeptVo) { SysDept sysDept = new SysDept(); BeanUtils.copyProperties(sysDeptVo, sysDept); String comlevel = ObjectUtils.isEmpty(sysDept.getComlevel()) ? "" : sysDept.getComlevel(); String parentId = sysDept.getParentId(); String number; if (comlevel.equals("2")) { AssertionUtils.isStringEmpty(sysDept.getProvince(), "二级机构份必填"); number = sysDept.getProvince().substring(0, 2); } else if (comlevel.equals("3")) { AssertionUtils.isStringEmpty(sysDept.getCity(), "三级机构市必填"); number = sysDept.getCity().substring(0, 4); String maxId = baseMapper.getMaxId(parentId, sysDeptVo.getDeptType(), comlevel); if (!maxId.equals("0")) { String substring = maxId.substring(sysDept.getParentId().length() + 1); int i = Integer.parseInt("1" + substring) + 1; number = String.valueOf(i); } else { number = "101"; } } else if (comlevel.equals("4")) { AssertionUtils.isStringEmpty(sysDept.getArea(), "四级机构区必填"); number = sysDept.getArea().substring(0, 6); String maxId = baseMapper.getMaxId(parentId, sysDeptVo.getDeptType(), comlevel); if (!maxId.equals("0")) { String substring = maxId.substring(sysDept.getParentId().length() + 1); int i = Integer.parseInt("1" + substring) + 1; number = String.valueOf(i); } else { number = "101"; } } else { String deptType = sysDeptVo.getDeptType(); if ("C".equals(sysDept.getDeptType()) && comlevel.equals("5")) { sysDeptVo.setDeptType("M"); } else if ("D".equals(sysDept.getDeptType())) { sysDeptVo.setDeptType("D"); deptType = "M"; } String maxId = baseMapper.getMaxId(parentId, deptType, comlevel); if (!maxId.equals("0")) { String substring = maxId.substring(sysDept.getParentId().length() + 1); int i = Integer.parseInt("1" + substring) + 1; number = String.valueOf(i); } else { number = "101"; } } String substring = number.substring(number.length() - 2); String serialNumber; if ("M,D".contains(sysDeptVo.getDeptType())) { serialNumber = parentId + sysDeptVo.getDeptType() + substring; } else { serialNumber = parentId + substring; } sysDept.setId(serialNumber); SysDept dept = getById(sysDept.getId()); if (!ObjectUtils.isEmpty(dept)) { throw new SystemException("机构编号为:" + dept.getId() + " " + dept.getName() + "已被占用"); } SysDept parentDept = baseMapper.selectById(sysDept.getParentId()); if (parentDept.getComlevel().equals("5")) { sysDept.setManagementSource(parentDept.getManagementSource()); } // 查询当前部门序号是否存在 LambdaQueryWrapper qw = new LambdaQueryWrapper<>(); qw.eq(SysDept::getDelFlag, 0); // 删除标志为未删除 qw.eq(SysDept::getParentId, sysDept.getParentId()); // 父级部门ID相同 qw.eq(SysDept::getOrderNum, sysDept.getOrderNum()); // 排序号相同 boolean exists = baseMapper.exists(qw); if (exists) { // 如果存在，查询该父级部门下排序号大于当前部门的部门列表 LambdaQueryWrapper lqw = new LambdaQueryWrapper<>(); lqw.eq(SysDept::getDelFlag, 0); // 删除标志为未删除 lqw.eq(SysDept::getParentId, sysDept.getParentId()); // 父级部门ID相同 lqw.ge(SysDept::getOrderNum, sysDept.getOrderNum()); // 排序号大于等于当前部门 List sysDeptList = list(lqw); // 更新这些部门的排序号 sysDeptList.forEach(i -> { i.setOrderNum(i.getOrderNum() + 1); updateById(i); }); } boolean b = save(sysDept); log.info("------------------添加部门成功-----------------------"); AssertionUtils.isSucceed(b, "删除失败"); sysDeptVo.setId(sysDept.getId()); return sysDeptVo; } @Override @Transactional public HttpResult updateDept(SysDept record) { // 查询修改的序号是否存在 LambdaQueryWrapper qw = new LambdaQueryWrapper<>(); qw.eq(SysDept::getDelFlag, 0); // 删除标志为未删除 qw.eq(SysDept::getParentId, record.getParentId()); // 父级部门ID相同 qw.eq(SysDept::getOrderNum, record.getOrderNum()); // 排序号相同 boolean exists = baseMapper.exists(qw); //不存在则直接修改 if (!exists) { baseMapper.updateById(record); return HttpResult.ok(); } //查询机构 SysDept sysDept = getById(record.getId()); //原本的序号 Integer orderNum = sysDept.getOrderNum(); //本次修改的序号 Integer newOrderNum = record.getOrderNum(); // 向上换位 if (newOrderNum < orderNum) { // 查询该父级部门下排序号大于当前部门的部门列表 LambdaQueryWrapper lqw = new LambdaQueryWrapper<>(); lqw.eq(SysDept::getDelFlag, 0); // 删除标志为未删除 lqw.eq(SysDept::getParentId, record.getParentId()); // 父级部门ID相同 lqw.ge(SysDept::getOrderNum, record.getOrderNum()); // 排序号大于等于当前部门 List sysDeptList = list(lqw); sysDeptList.sort(Comparator.comparing(SysDept::getOrderNum)); for (SysDept dept : sysDeptList) { dept.setOrderNum(dept.getOrderNum() + 1); updateById(dept); if (dept.getOrderNum().equals(orderNum)) { break; } } baseMapper.updateById(record); } //向下换位 if (newOrderNum > orderNum) { // 查询该父级部门下排序号小于当前部门的部门列表 LambdaQueryWrapper lqw = new LambdaQueryWrapper<>(); lqw.eq(SysDept::getDelFlag, 0); // 删除标志为未删除 lqw.eq(SysDept::getParentId, record.getParentId()); // 父级部门ID相同 lqw.le(SysDept::getOrderNum, record.getOrderNum()); // 排序号大于等于当前部门 List sysDeptList = list(lqw); sysDeptList.sort(Comparator.comparing(SysDept::getOrderNum).reversed()); for (SysDept dept : sysDeptList) { dept.setOrderNum(dept.getOrderNum() - 1); updateById(dept); if (dept.getOrderNum().equals(orderNum)) { break; } } baseMapper.updateById(record); } if (newOrderNum.equals(orderNum)) { baseMapper.updateById(record); } return HttpResult.ok(); } @Override public void deleteDept(SysDeptRemoveVo sysDeptRemoveVo) { SysDept sysDept = getById(sysDeptRemoveVo.getDeptId()); List list = lambdaQuery().eq(SysDept::getParentId, sysDept.getId()).list(); if (list.isEmpty()) { boolean b = removeById(sysDeptRemoveVo.getDeptId()); AssertionUtils.isSucceed(b, "删除失败"); } else { throw new SystemException(sysDept.getName() + "下还有其他隶属机构"); } } @Override public PageResult findSubDepts(PageRequest pageRequest) { String id = pageRequest.getColumnFilterValue(pageRequest, "id"); String managementSource = pageRequest.getColumnFilterValue(pageRequest, "managementSource"); Page p = PageRequest.buildPageRequest(pageRequest); LambdaQueryWrapper lqw = new LambdaQueryWrapper<>(); lqw.eq(StringUtils.isNotEmpty(id), SysDept::getParentId, id); lqw.eq(StringUtils.isNotEmpty(managementSource), SysDept::getManagementSource, managementSource); IPage pResult = page(p, lqw); //传入查询条件 return PageResult.buildPageResult(pResult); } public String findMaxDeptId(String deptId) { if (deptId == null) return ""; String result = "00"; QueryWrapper qw = new QueryWrapper<>(); qw.likeRight(StringUtils.isNotEmpty(deptId), "id", deptId + "D%"); qw.select("IFNULL(MAX(id),0) maxid"); Map m = getMap(qw); String maxId = m.get("maxid").toString(); if (!maxId.equalsIgnoreCase("0") && !maxId.equalsIgnoreCase(deptId)) { result = maxId.substring(deptId.length() + 1); } return result; } public String findMaxHouseId(String deptId) { if (deptId == null) { return ""; } String result = "00"; QueryWrapper qw = new QueryWrapper<>(); qw.like(StringUtils.isNotEmpty(deptId), "id", deptId + "M%"); qw.select("IFNULL(MAX(id),0) maxid"); Map m = getMap(qw); String maxId = m.get("maxid").toString(); if (!maxId.equalsIgnoreCase("0") && !maxId.equalsIgnoreCase(deptId)) { result = maxId.substring(deptId.length() + 1); } return result; } // @Override public List selectListWhere(String where, Object[] paramValues) { String sql = "select t1.id, t1.name, t1.parent_id, t1.dept_code, t1.province, t1.city, t1.area, t1.comlevel from sys_dept t1 " + "where del_flag='0' " + where; QueryWrapper qw = new QueryWrapper<>(); qw.select(sql); List sysDeptList = list(qw); List recordList = new ArrayList(); sysDeptList.forEach(a -> { TreeSysDeptVO model = new TreeSysDeptVO(); model.setId(a.getId()); model.setName(a.getName()); model.setParentId(a.getParentId()); model.setDeptCode(a.getDeptCode()); model.setProvince(a.getProvince()); model.setCity(a.getCity()); model.setArea(a.getArea()); model.setComlevel(a.getComlevel()); recordList.add(model); }); return recordList; } /** * 获取机构树(不包含团队 dept_type=C) */ @Override public List findCompanyTree(String deptId) { SysDept sysDept = getById(deptId); LambdaQueryWrapper lqw = new LambdaQueryWrapper<>(); lqw.eq(SysDept::getDelFlag, 0); lqw.likeRight(SysDept::getId, deptId); lqw.and((wrapper) -> { wrapper.eq(SysDept::getDeptType, 'C').or().eq(SysDept::getId, '9'); }); List sysDeptList = list(lqw); Map> map = sysDeptList.stream().collect(Collectors.groupingBy(SysDept::getParentId)); sysDeptList.forEach(i -> i.setChildren(map.getOrDefault(i.getId(), new ArrayList<>()))); return map.get(sysDept.getParentId()); } /** * 根据传入机构，查询归属所有团队 */ @Override public List findTeamList(String deptId) { SysDept sysDept = getById(deptId); LambdaQueryWrapper lqw = new LambdaQueryWrapper<>(); lqw.eq(SysDept::getDelFlag, 0); lqw.likeRight(SysDept::getId, deptId); lqw.eq(SysDept::getDeptType, 'D'); lqw.apply(" exists (SELECT 1 FROM `sys_dept` b WHERE sys_dept.parent_id = b.`id` AND b.`comlevel`='5') "); List sysDeptList = list(lqw); return sysDeptList; } /** * @version * @author: hxl * @Date: 2024/4/28 10:24 * @Description: 查询省、市、县 、门店的列表 */ @Override public List queryListByParentId(String level, String parentId) { return sysDeptMapper.queryListByParentId(level, parentId); } @Override public String queryBelongArea(String deptId) { //长度大于8的话 进行切割 StringBuilder result = new StringBuilder(); if (deptId.length() >= 8) { String area = deptId.substring(2, 8); SysArea sysArea = sysAreaMapper.selectOne(new LambdaQueryWrapper().eq(SysArea::getAreaCode, area)); if (sysArea == null) { return ""; } List sysAreas = sysAreaService.queryAllParent(sysArea); result.append(sysAreas.get(2).getAreaCname()); result.append(sysAreas.get(1).getAreaCname()); result.append(sysAreas.get(0).getAreaCname()); } return result.toString(); } @Override public List getListQueryBelongArea(String deptId) { //长度大于8的话 进行切割 List sysAreas = new ArrayList<>(); if (deptId.length() >= 8) { String area = deptId.substring(2, 8); SysArea sysArea = sysAreaMapper.selectOne(new LambdaQueryWrapper().eq(SysArea::getAreaCode, area)); if (sysArea == null) { return null; } sysAreas = sysAreaService.queryAllParent(sysArea); } return sysAreas; } /** * @version * @author: hxl * @Date: 2024/6/28 8:43 * @Description: 查询后线机构树 */ @Override public List queryHXDeptTree() { //所有的后线机构 List sysDeptsHX = sysDeptMapper.queryHXDeptTree(); //查询后线中顶级的数据 List resultList = new ArrayList(); List sysDeptTop = sysDeptsHX.stream().filter(item -> "99".equals(item.getId())).collect(Collectors.toList()); for (SysDept dept : sysDeptTop) { resultList.add(buildDeptTree(dept, sysDeptsHX)); } return resultList; } /** * 递归构建树形结构 */ private SysDept buildDeptTree(SysDept dept, List deptList) { for (SysDept d : deptList) { if (dept.getId().equals(d.getParentId())) { if (null == dept.getChildren()) { dept.setChildren(new ArrayList()); } //递归调用 dept.getChildren().add(buildDeptTree(d, deptList)); } } return dept; } /** * @version * @author: hxl * @Date: 2024/6/28 8:42 * @Description: 查询前线机构树 */ @Override public List queryQXDeptTree() { List sysDeptAll = sysDeptMapper.queryListByQXDept(); //查询后线中顶级的数据 List resultList = new ArrayList(); List sysDeptTop = sysDeptAll.stream().filter(item -> "3".equals(item.getComlevel())).collect(Collectors.toList()); for (SysDept dept : sysDeptTop) { resultList.add(buildDeptTree(dept, sysDeptAll)); } return sysDeptMapper.queryListByQXDept(); } @Override public List getSysDeptTree() { SysDept sysDept = getById("9"); LambdaQueryWrapper lqw = new LambdaQueryWrapper<>(); lqw.eq(SysDept::getDelFlag, 0); lqw.isNotNull(SysDept::getComlevel); lqw.likeRight(SysDept::getId, "9"); lqw.orderByAsc(SysDept::getOrderNum); List sysDeptList = list(lqw); SysDeptTreeVo sysDeptTreeVo = BeanUtil.copyProperties(sysDept, SysDeptTreeVo.class); List sysDeptTreeVos = BeanUtil.copyToList(sysDeptList, SysDeptTreeVo.class); sysDeptTreeVos.add(sysDeptTreeVo); Map> map = sysDeptTreeVos.stream().sorted(Comparator.comparing(SysDeptTreeVo::getId)).collect(Collectors.groupingBy(SysDeptTreeVo::getParentId)); sysDeptTreeVos.forEach(i -> { i.setChildren(map.getOrDefault(i.getId(), new ArrayList<>())); i.setDeptCount(i.getChildren().size()); }); sysDeptTreeVos = map.get(sysDept.getParentId()); return sysDeptTreeVos; } @Override public List getPartnerDeptTree() { SysDept sysDept = getById("9"); LambdaQueryWrapper lqw = new LambdaQueryWrapper<>(); lqw.eq(SysDept::getDelFlag, 0); lqw.likeRight(SysDept::getId, "9"); lqw.orderByAsc(SysDept::getOrderNum); List sysDeptList = list(lqw); List sysDeptTreeVos = BeanUtil.copyToList(sysDeptList, SysDeptTreeVo.class); Map> map = sysDeptTreeVos.stream().sorted(Comparator.comparing(SysDeptTreeVo::getId)) .collect(Collectors.groupingBy(SysDeptTreeVo::getParentId)); sysDeptTreeVos.forEach(i -> { i.setChildren(map.getOrDefault(i.getId(), new ArrayList<>())); i.setDeptCount(i.getChildren().size()); }); sysDeptTreeVos = map.get(sysDept.getParentId()); //合伙人机构树 for (SysDeptTreeVo deptTreeVo : sysDeptTreeVos) {//1 List children2 = deptTreeVo.getChildren(); for (SysDeptTreeVo children : children2) {//2 List children3 = children.getChildren(); if (children3.size()>0) { children3=children3.stream().filter(SysDeptTreeVo -> SysDeptTreeVo.getComlevel() != null).collect(Collectors.toList()); children.setChildren(children3); for (SysDeptTreeVo child4 : children3) {//3 List children4 = child4.getChildren(); if (children4.size()>0) { children4=children4.stream().filter(SysDeptTreeVo -> SysDeptTreeVo.getComlevel() != null).collect(Collectors.toList()); child4.setChildren(children4); for (SysDeptTreeVo child : children4) {//4 List children5 = child.getChildren(); children5 = children5.stream().filter(SysDeptTreeVo ->SysDeptTreeVo.getComlevel() != null).collect(Collectors.toList()); child.setChildren(children5); if (children5.size()>0) { for (SysDeptTreeVo treeVo : children5) { List childrenEnd = treeVo.getChildren(); if (childrenEnd.size()>0) { childrenEnd=childrenEnd.stream().filter(SysDeptTreeVo ->SysDeptTreeVo.getComlevel() == null).collect(Collectors.toList()); treeVo.setChildren(childrenEnd); } } } } } } } } } return sysDeptTreeVos; } @Override public boolean contrastiveManagementSource(String orderNo, String managementSource) { String managementSourceDept = sysDeptMapper.getByOrderNoQueryManagementSource(orderNo); return managementSource.equals(managementSourceDept); } /** * 数据权限机构数 * @return */ @Override public List getDataDeptTree(List deptIds) { String name = BaseController.getUser().getName(); List result = new ArrayList<>(); if(!deptIds.isEmpty()){ List deptList = sysDeptMapper.getListByIdsByKzt(deptIds); // 使用一个Map来快速查找部门节点 Map deptMap = new HashMap<>(); for (SysDept dept : deptList) { deptMap.put(dept.getId(), dept); } // 找出顶级部门（即没有父部门的部门） for (SysDept dept : deptList) { if (dept.getParentId() == null || deptMap.get(dept.getParentId()) == null) { // 添加到结果列表并构建其子树 buildDeptTree(dept, deptMap); result.add(dept); } } } else if ("admin".equals(name)) { String deptId ="9"; if(deptId==null || "".equals(deptId)){ deptId="9"; }else{ //如果是平台下判断包含D的查询上级id if(deptId.contains("D")){ SysDept dept = baseMapper.selectById(deptId); deptId =dept==null?"":dept.getParentId(); } } LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>(); queryWrapper.eq(SysDept::getDelFlag, 0); // 获取当前操作用户 SysUser user = BaseController.getUser(); String userId = user.getId(); // 根据用户ID查询用户的部门树展示级别 LambdaQueryWrapper lambdaQueryWrapper = new LambdaQueryWrapper<>(); lambdaQueryWrapper.eq(SysUserTreeShowLevel::getUserId, userId); SysUserTreeShowLevel treeShowLevel = sysUserTreeShowLevelService.getOne(lambdaQueryWrapper); List treeByShowLevel = new ArrayList<>(); //如果当前用户未配置树展示等级，则默认为查询全部 if (Objects.isNull(treeShowLevel)) { treeShowLevel = new SysUserTreeShowLevel(); treeShowLevel.setShowLevel("subordinate"); } // 如果指定的部门ID为空或特定值，则使用当前用户所在部门ID if (null == deptId || "".equals(deptId) || "undefined".equals(deptId)) { if (!treeShowLevel.getShowLevel().equals("all")) { deptId = user.getDeptId(); } else { deptId = "9"; } } // 根据用户的部门树展示级别查询相应的部门树 switch (treeShowLevel.getShowLevel()) { case "all": treeByShowLevel = findTreeByShowLevel(deptId, null, "all"); break; case "same": treeByShowLevel = findTreeByShowLevel(deptId, null, "same"); break; case "subordinate": treeByShowLevel = findTreeByShowLevel(deptId, null, "subordinate"); break; } return treeByShowLevel; } return result; } @Override public List getFiveLevelDept(SysDeptVo sysDeptVo) { SysDept sysDept = getById("9"); List sysDeptList = baseMapper.getFiveLevelDept(sysDeptVo); sysDeptList.add(sysDept);//添加总部 总部没有等级机构 List sysDeptTreeVos = BeanUtil.copyToList(sysDeptList, SysDeptTreeVo.class); Map> map = sysDeptTreeVos.stream().sorted(Comparator.comparing(SysDeptTreeVo::getId)) .collect(Collectors.groupingBy(SysDeptTreeVo::getParentId)); sysDeptTreeVos.forEach(i -> { i.setChildren(map.getOrDefault(i.getId(), new ArrayList<>())); i.setDeptCount(i.getChildren().size()); }); return map.get(sysDept.getParentId()); } private void buildDeptTree(SysDept dept, Map deptMap) { List children = new ArrayList<>(); dept.setChildren(children); // 查找并添加子部门 for (SysDept potentialChild : deptMap.values()) { if (potentialChild.getParentId() != null && potentialChild.getParentId().equals(dept.getId())) { children.add(potentialChild); buildDeptTree(potentialChild, deptMap); // 递归构建子部门的子树 } } } /** * 前线只要机构 后线人员（按照领导人） 机构 * @param sysDeptVo * @return */ @Override public KztUserDeptTreeDto getUserDeptTree(SysDeptVo sysDeptVo) { String managementSource = sysDeptVo.getManagementSource(); KztUserDeptTreeDto kztUserDeptTreeDto = new KztUserDeptTreeDto(); String userId = BaseController.getUserId(); QueryWrapper queryWrapper = new QueryWrapper<>(); QueryWrapper leaderId = queryWrapper.eq("leader_id", userId); List sysDepts = sysDeptMapper.selectList(leaderId); if (!sysDepts.isEmpty()) { kztUserDeptTreeDto.setType("1"); //前线 } else { kztUserDeptTreeDto.setType("2"); //代理 } SysUser byUserId = sysUserMapper.getByUserId(userId); List resultDeptList = this.getFiveLevelDeptKzt(byUserId,managementSource); kztUserDeptTreeDto.setDeptList(resultDeptList); if (!userId.contains("M")) { List resultList =this.getUserTreeHx(byUserId,managementSource); kztUserDeptTreeDto.setUserList(resultList); } return kztUserDeptTreeDto; } private List getUserTreeHx(SysUser sysUser,String managementSource) { //获取领导字段不为空的人的ID QueryWrapper queryWrapper = new QueryWrapper<>(); queryWrapper.isNotNull("leader_id"); HashMap params = new HashMap<>(); params.put("managementSource",managementSource); List users = sysUserMapper.selectByParams(params); for (SysUser user : users) { this.appendName(user); } if (users == null || users.isEmpty()) { return new ArrayList<>(); } List allList = new ArrayList<>(); for (SysUser user : users) { this.addLeadUserList(user,allList); } List collect = allList.stream() .collect(Collectors.collectingAndThen( Collectors.toMap(SysUser::getId, u -> u, (u1, u2) -> u1), // 以id作为键来合并重复元素，保留第一个遇到的元素 Map::values)) .stream() .distinct() // 进一步确保去重，主要针对可能存在的相同id但不同对象引用情况（较极端情况） .collect(Collectors.toList()); List topUsers = new ArrayList<>(); //存放1级用户,即没有直接上级的用户 // 根据组织结构展示级别，筛选顶层节点并构建组织结构树 if("admin".equals(sysUser.getName())){ topUsers.add(sysUserMapper.selectById("99D1501")); }else { this.appendName(sysUser); topUsers.add(sysUser); } // 递归构建用户树结构 buildUserTree(topUsers, collect,managementSource); return topUsers; } private void addLeadUserList(SysUser user, List allList) { addLeadUserList(user, allList, new HashSet<>()); } private void addLeadUserList(SysUser user, List allList, Set processedIds) { if (user == null) { return; } // 先判断当前用户ID是否已经处理过，如果处理过则直接返回，避免循环引用导致的无限递归 if (processedIds.contains(user.getId())) { return; } allList.add(user); processedIds.add(user.getId()); String leaderId = user.getLeaderId(); if (StringUtils.isNotEmpty(leaderId)) { SysUser leaderUser = sysUserMapper.getByUserId(leaderId); addLeadUserList(leaderUser, allList, processedIds); } } /** * 拼接机构来源 * @param users */ private void appendName(SysUser user) { StringBuilder stringBuilder = new StringBuilder(); String deptId = user.getDeptId(); if (StringUtils.isNotEmpty(deptId)) { int length = deptId.length(); if (length > 0) { if (length <= 5) { String dept = deptId.replaceAll("[^\\d]*(\\d+).*", "$1"); SysDept cityName = baseMapper.selectById(dept); String name = cityName.getName(); stringBuilder.append("(").append(name).append(")"); } else if (length > 5) { String dept = deptId.replaceAll("[^\\d]*(\\d+).*", "$1"); if (dept.length() < 5) { SysDept dept1 = baseMapper.selectById(dept); SysDept cityName = baseMapper.selectById(deptId); stringBuilder.append("(").append(dept1.getName()).append(cityName.getName()).append(")"); } else { String deptIdTwo = dept.substring(0, 4); SysDept dept1 = baseMapper.selectById(deptIdTwo); SysDept dept2 = baseMapper.selectById(dept); stringBuilder.append("(").append(dept1.getName()).append(dept2.getName()).append(")"); } } user.setName(user.getName()+stringBuilder.toString()); } } } private List getFiveLevelDeptKzt(SysUser byUserId,String managementSource) { SysDept sysDept = getById("9"); SysDeptVo sysDeptVo = new SysDeptVo(); String id = byUserId.getId(); List deptIds = new ArrayList<>(); if("admin".equals(byUserId.getId())){ List sysDeptList = baseMapper.getFiveLevelDept(sysDeptVo); deptIds.addAll(sysDeptList.stream().map(SysDept::getId).collect(Collectors.toList())); }else { List relationships = sysUserDeptRelationshipMapper.selectList(new QueryWrapper().eq("user_id", id)); deptIds.addAll(relationships.stream().map(SysUserDeptRelationship::getDeptId).collect(Collectors.toList())); } if(!deptIds.isEmpty()){ sysDeptVo.setDeptIds(deptIds); List sysDeptList = baseMapper.getFiveLevelDept(sysDeptVo); if("admin".equals(byUserId.getId())){ sysDept.setComlevel("1"); sysDeptList.add(sysDept); List sysDeptTreeVos = BeanUtil.copyToList(sysDeptList, SysDeptTreeVo.class); List filteredSysDeptTreeVos = sysDeptTreeVos.stream() .filter(dept -> !(dept.getComlevel().equals("5") && !Objects.equals(dept.getManagementSource(), managementSource))) .collect(Collectors.toList()); Map> map = filteredSysDeptTreeVos.stream().sorted(Comparator.comparing(SysDeptTreeVo::getId)) .collect(Collectors.groupingBy(SysDeptTreeVo::getParentId)); filteredSysDeptTreeVos.forEach(i -> { if(Integer.parseInt(i.getComlevel())<=5){ i.setChildren(map.getOrDefault(i.getId(), new ArrayList<>())); i.setDeptCount(i.getChildren().size()); } }); return map.get(sysDept.getParentId()); } else { List result = new ArrayList<>(); List deptList = sysDeptMapper.getListByIds(deptIds); List filteredDeptList = deptList.stream() .filter(dept -> dept.getComlevel() != null) // 过滤条件：Comlevel 不为 null .collect(Collectors.toList()); List filteredSysDeptTreeVos = filteredDeptList.stream() .filter(dept -> !(dept.getComlevel().equals("5") && !Objects.equals(dept.getManagementSource(), managementSource))) .collect(Collectors.toList()); // 使用一个Map来快速查找部门节点 Map deptMap = new HashMap<>(); for (SysDept dept : filteredSysDeptTreeVos) { deptMap.put(dept.getId(), dept); } // 找出顶级部门（即没有父部门的部门） for (SysDept dept : filteredSysDeptTreeVos) { if (dept.getParentId() == null || deptMap.get(dept.getParentId()) == null) { // 添加到结果列表并构建其子树 buildDeptTreeKzt(dept, deptMap); result.add(dept); } } List sysDeptTreeVos = BeanUtil.copyToList(result, SysDeptTreeVo.class); return sysDeptTreeVos; } }else { return new ArrayList(); } } private void buildDeptTreeKzt(SysDept dept, Map deptMap) { List children = new ArrayList<>(); dept.setChildren(children); // 查找并添加子部门 for (SysDept potentialChild : deptMap.values()) { if (potentialChild.getParentId() != null && potentialChild.getParentId().equals(dept.getId())) { if(StringUtils.isNotEmpty(potentialChild.getComlevel()) && Integer.parseInt(potentialChild.getComlevel())<=5 ){ children.add(potentialChild); buildDeptTreeKzt(potentialChild, deptMap); // 递归构建子部门的子树 } } } } private void buildUserTree(List topUsers, List users, String managementSource) { for (SysUser user : topUsers) { List children = new ArrayList<>(); for (SysUser childUser : users) { if (user.getId().equals(childUser.getLeaderId())) { children.add(childUser); // 递归查找下一级 buildUserTree(children, users,managementSource); } } user.setChildren(children); } } @Override public List getLeadListDeptId(String deptId) { return sysUserMapper.getLeadListDeptId(deptId); } /** * @version * @author: hxl * @Date: 2024/9/26 18:31 * @Description: 设置领导人 */ @Override public void saveLeadId(DeptLeadVo deptLeadVo) { SysDept dept = baseMapper.selectById(deptLeadVo.getDeptId()); if(StringUtils.isBlank(deptLeadVo.getLeadId())){ dept.setLeaderId(""); dept.setLeadername(""); baseMapper.updateById(dept); }else{ SysUser sysUser = sysUserMapper.selectById(deptLeadVo.getLeadId()); dept.setLeaderId(deptLeadVo.getLeadId()); dept.setLeadername(sysUser.getName()); } baseMapper.updateById(dept); } }