拿不到地址总表数据，不设置网格信息

linwei

2024-01-31 b3b566ebdfed4005aaa513da3d5d2fd3924903cc

 src/main/java/org/springblade/modules/category/service/impl/CategoryServiceImpl.java

@@ -17,6 +17,8 @@
package org.springblade.modules.category.service.impl;

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springblade.common.node.TreeIntegerNode;
import org.springblade.common.utils.NodeTreeUtil;
import org.springblade.modules.category.dto.CategoryDTO;
import org.springblade.modules.category.entity.CategoryEntity;
import org.springblade.modules.category.vo.CategoryVO;
@@ -27,6 +29,7 @@
import com.baomidou.mybatisplus.core.metadata.IPage;

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

/**
 * 天地图poi 分类表 服务实现类
@@ -58,4 +61,20 @@
   public List<CategoryDTO> selectCategoryLabelList(CategoryDTO categoryDTO) {
      return this.baseMapper.selectCategoryList(categoryDTO);
   }

   @Override
   public List<TreeIntegerNode> tree(CategoryVO category) {
      Map<Integer, TreeIntegerNode> labelTreeList = baseMapper.getTreeList(category);
      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);
      }
   }
}