From 5d5cd3ad979a0b85ab8c7034dcad1663fdcf71be Mon Sep 17 00:00:00 2001
From: linwe <872216996@qq.com>
Date: Fri, 01 Dec 2023 14:18:53 +0800
Subject: [PATCH] 文章通过范围查询

---
 src/main/java/org/springblade/modules/district/service/impl/DistrictServiceImpl.java |   39 +++++++++++++++++++++++++++++++++++++++
 1 files changed, 39 insertions(+), 0 deletions(-)

diff --git a/src/main/java/org/springblade/modules/district/service/impl/DistrictServiceImpl.java b/src/main/java/org/springblade/modules/district/service/impl/DistrictServiceImpl.java
index 24d2178..2974ce4 100644
--- a/src/main/java/org/springblade/modules/district/service/impl/DistrictServiceImpl.java
+++ b/src/main/java/org/springblade/modules/district/service/impl/DistrictServiceImpl.java
@@ -17,6 +17,9 @@
 package org.springblade.modules.district.service.impl;
 
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import org.springblade.common.node.TreeIntegerNode;
+import org.springblade.common.node.TreeStringNode;
+import org.springblade.common.utils.NodeTreeUtil;
 import org.springblade.modules.district.entity.DistrictEntity;
 import org.springblade.modules.district.vo.DistrictVO;
 import org.springblade.modules.district.mapper.DistrictMapper;
@@ -24,6 +27,8 @@
 import org.springblade.core.mp.base.BaseServiceImpl;
 import org.springframework.stereotype.Service;
 import com.baomidou.mybatisplus.core.metadata.IPage;
+
+import java.util.List;
 
 /**
  * 小区表 服务实现类
@@ -39,5 +44,39 @@
 		return page.setRecords(baseMapper.selectDistrictPage(page, district));
 	}
 
+	/**
+	 * 获取小区树
+	 * @param district
+	 * @return
+	 */
+	@Override
+	public Object getDistrictTree(DistrictVO district) {
+		List<TreeStringNode> stringNodeTree = NodeTreeUtil.getStringNodeTree(baseMapper.getDistrictTree(district));
+		stringNodeTree.forEach(node -> recursion(node));
+		return stringNodeTree;
+	}
 
+
+	/**
+	 * 去除空的数据组
+	 * @param node
+	 */
+	private void recursion(TreeStringNode node) {
+		if (node.getChildren() != null && node.getChildren().size() > 0) {
+			node.getChildren().forEach(node2 -> recursion(node2));
+		} else {
+			node.setChildren(null);
+		}
+	}
+
+	/**
+	 * 小区表 自定义获取详情
+	 * @param district
+	 * @return
+	 */
+	@Override
+	public DistrictVO getDetail(DistrictVO district) {
+		// 小区自定义获取详情查询并返回
+		return baseMapper.getDetail(district);
+	}
 }

--
Gitblit v1.9.3