From b7df6735bcb2d447cae7fcfba2d12accf3dd2fcc Mon Sep 17 00:00:00 2001
From: linwei <872216696@qq.com>
Date: Sat, 03 Feb 2024 17:27:21 +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