公安安全监管

linwei

2024-01-08 ad2ec97fe0c60eee4bb7c8cdacff7c876f280ea7

 src/main/java/org/springblade/modules/label/service/impl/LabelServiceImpl.java

@@ -26,6 +26,7 @@
import org.springblade.modules.label.vo.LabelVO;
import org.springframework.stereotype.Service;

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

@@ -50,7 +51,22 @@
    */
   @Override
   public Object getLabelList(LabelVO label) {
      return NodeTreeUtil.getNodeTree(baseMapper.getLabelList(label));
      Map<Integer, TreeIntegerNode> labelList = baseMapper.getLabelList(label);
      List<Integer> list = new ArrayList<>();
      // 遍历
      labelList.forEach((id, treeNode) -> {
         if (treeNode.getHasChildren()){
            list.add(id);
         }
      });
      if (list.size()>0) {
         // 查询子集
         Map<Integer, TreeIntegerNode> childrenLabelList = baseMapper.getChildrenLabelList(list);
         // 合并集合
         labelList.putAll(childrenLabelList);
      }
      // 处理并返回
      return NodeTreeUtil.getNodeTree(labelList);
   }

   @Override