From 557c6dddaab2a2827d1eb55a1903f7099f6a9676 Mon Sep 17 00:00:00 2001
From: linwei <872216696@qq.com>
Date: Fri, 26 Jan 2024 17:56:07 +0800
Subject: [PATCH] 场所标签统计

---
 src/main/java/org/springblade/modules/category/service/impl/CategoryServiceImpl.java |   19 +++++++++++++++++++
 1 files changed, 19 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 55fa0c8..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,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);
+		}
+	}
 }

--
Gitblit v1.9.3