linwe
2023-12-01 5d5cd3ad979a0b85ab8c7034dcad1663fdcf71be
src/main/java/org/springblade/modules/label/service/impl/LabelServiceImpl.java
@@ -16,17 +16,18 @@
 */
package org.springblade.modules.label.service.impl;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springblade.common.node.TreeIntegerNode;
import org.springblade.common.utils.NodeTreeUtil;
import org.springblade.modules.label.entity.LabelEntity;
import org.springblade.modules.label.vo.LabelVO;
import org.springblade.modules.label.mapper.LabelMapper;
import org.springblade.modules.label.service.ILabelService;
import org.springblade.core.mp.base.BaseServiceImpl;
import org.springblade.modules.label.vo.LabelVO;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.core.metadata.IPage;
import java.util.List;
import java.util.Map;
/**
 * 标签管理 服务实现类
@@ -53,7 +54,18 @@
   }
   @Override
   public List<LabelVO> tree(LabelEntity label) {
      return null;
   public List<TreeIntegerNode> tree(LabelVO label) {
      Map<Integer, TreeIntegerNode> labelTreeList = baseMapper.getLabelTreeList(label);
      List<TreeIntegerNode> nodeTree = NodeTreeUtil.getNodeTree(labelTreeList);
      nodeTree.forEach(node -> recursion(node));
      return nodeTree;
   }
   private void recursion(TreeIntegerNode node) {
      if (node.getChildren() != null && node.getChildren().size() > 0) {
         node.getChildren().forEach(node2 -> recursion(node2));
      } else {
         node.setChildren(null);
      }
   }
}