From b3b566ebdfed4005aaa513da3d5d2fd3924903cc Mon Sep 17 00:00:00 2001
From: linwei <872216696@qq.com>
Date: Wed, 31 Jan 2024 16:17:54 +0800
Subject: [PATCH] 拿不到地址总表数据,不设置网格信息

---
 src/main/java/org/springblade/modules/category/service/impl/CategoryServiceImpl.java |   37 +++++++++++++++++++++++++++++++++++++
 1 files changed, 37 insertions(+), 0 deletions(-)

diff --git a/src/main/java/org/springblade/modules/category/service/impl/CategoryServiceImpl.java b/src/main/java/org/springblade/modules/category/service/impl/CategoryServiceImpl.java
index 4798c61..8749263 100644
--- a/src/main/java/org/springblade/modules/category/service/impl/CategoryServiceImpl.java
+++ b/src/main/java/org/springblade/modules/category/service/impl/CategoryServiceImpl.java
@@ -17,6 +17,9 @@
 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;
 import org.springblade.modules.category.mapper.CategoryMapper;
@@ -24,6 +27,9 @@
 import org.springblade.core.mp.base.BaseServiceImpl;
 import org.springframework.stereotype.Service;
 import com.baomidou.mybatisplus.core.metadata.IPage;
+
+import java.util.List;
+import java.util.Map;
 
 /**
  * 天地图poi 分类表 服务实现类
@@ -39,5 +45,36 @@
 		return page.setRecords(baseMapper.selectCategoryPage(page, category));
 	}
 
+	/**
+	 * 天地图poi 分类获取
+	 */
+	@Override
+	public Object getCategory(CategoryVO category) {
+		return baseMapper.getCategory(category);
+	}
+	/**
+	 * 查询场所标签
+	 * @param categoryDTO
+	 * @return
+	 */
+	@Override
+	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);
+		}
+	}
 }

--
Gitblit v1.9.3