系统消息，我的消息

guoshilong

2024-02-02 36b7f863fda32d53b5cdadd6ee1514de4aea3905

 src/main/java/org/springblade/common/utils/NodeTreeUtil.java

@@ -6,8 +6,10 @@
import org.springblade.core.tool.node.TreeNode;
import org.springblade.modules.doorplateAddress.vo.DoorplateAddressVOTree;
import org.springblade.modules.house.vo.HouseTree;
import org.springblade.modules.system.node.DeptUserTreeNode;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;

@@ -144,4 +146,44 @@
      }
      return parentList;
   }

   /**
    * map 转 tree 组织机构(父子id不会重复的情况，数据查询来自同一个表)
    * @param treeMap
    * @return
    */
   public  static List<DeptUserTreeNode> getDeptAndUserNodeTree(Map<String, DeptUserTreeNode> treeMap){
      List<DeptUserTreeNode> tree = new ArrayList<>();
      if (treeMap.size() > 1) {
         treeMap.forEach((id, treeNode) -> {
            if (!treeNode.getHasChildren()){
               // 判断是否有绑定多个部门
               List<String> asList = Arrays.asList(treeNode.getParentId().split(","));
               if (asList.size()>1){
                  for (String parentId : asList) {
                     if (treeMap.containsKey(parentId)) {
                        treeMap.get(parentId).getChildren().add(treeNode);
                     } else {
                        tree.add(treeNode);
                     }
                  }
               }else {
                  if (treeMap.containsKey(treeNode.getParentId())) {
                     DeptUserTreeNode deptUserTreeNode =treeMap.get(treeNode.getParentId());
                     deptUserTreeNode.getChildren().add(treeNode);
                  } else {
                     tree.add(treeNode);
                  }
               }
            }else {
               if (treeMap.containsKey(treeNode.getParentId())) {
                  treeMap.get(treeNode.getParentId()).getChildren().add(treeNode);
               } else {
                  tree.add(treeNode);
               }
            }
         });
      }
      return tree;
   }
}