xieb
2023-10-19 1dd5d8f8a616f5cf3caa7828d46989c7d3dcafc4
skjcmanager/skjcmanager-service/skjcmanager-system/src/main/java/cn/gistack/system/service/impl/DeptServiceImpl.java
@@ -16,6 +16,9 @@
 */
package cn.gistack.system.service.impl;
import cn.gistack.common.node.DeptUserTreeNode;
import cn.gistack.common.node.TreeNode;
import cn.gistack.common.utils.NodeTreeUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
@@ -81,6 +84,14 @@
   @Override
   public List<DeptVO> tree(String tenantId) {
      return ForestNodeMerger.merge(baseMapper.tree(tenantId));
   }
   @Override
   public List<DeptVO> treeByParentId(String parentId) {
      if (AuthUtil.isAdministrator() || AuthUtil.isAdmin()) {
         return ForestNodeMerger.merge(baseMapper.tree(StringPool.EMPTY));
      }
      return ForestNodeMerger.merge(baseMapper.treeByParentId(parentId));
   }
   @Override
@@ -170,4 +181,18 @@
      return DeptWrapper.build().listNodeVO(deptList);
   }
   /**
    * 查询组织机构树数据(下级包含人员信息)
    * @param treeNode
    * @return
    */
   @Override
   public List<TreeNode> getDeptAndUserTree(DeptUserTreeNode treeNode) {
      // 查询数据
      Map<String, DeptUserTreeNode> deptMap = baseMapper.getDeptMap(treeNode);
      Map<String, DeptUserTreeNode> userMap = baseMapper.getUserMap(treeNode);
      deptMap.putAll(userMap);
      // 处理并返回
      return NodeTreeUtil.getDeptAndUserNodeTree(deptMap);
   }
}