From ad2ec97fe0c60eee4bb7c8cdacff7c876f280ea7 Mon Sep 17 00:00:00 2001
From: linwei <872216696@qq.com>
Date: Mon, 08 Jan 2024 11:30:18 +0800
Subject: [PATCH] 公安安全监管

---
 src/main/java/org/springblade/modules/district/service/impl/DistrictServiceImpl.java |   57 +++++++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 55 insertions(+), 2 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 f6dc98e..8c8e791 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
@@ -20,15 +20,24 @@
 import org.springblade.common.node.TreeIntegerNode;
 import org.springblade.common.node.TreeStringNode;
 import org.springblade.common.utils.NodeTreeUtil;
+import org.springblade.common.utils.SpringUtils;
+import org.springblade.core.secure.utils.AuthUtil;
 import org.springblade.modules.district.entity.DistrictEntity;
 import org.springblade.modules.district.vo.DistrictVO;
 import org.springblade.modules.district.mapper.DistrictMapper;
 import org.springblade.modules.district.service.IDistrictService;
 import org.springblade.core.mp.base.BaseServiceImpl;
+import org.springblade.modules.grid.service.IGridService;
+import org.springblade.modules.grid.vo.GridVO;
+import org.springblade.modules.property.service.IPropertyDistrictUserService;
+import org.springblade.modules.system.entity.Dept;
+import org.springblade.modules.system.service.IDeptService;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 
 import java.util.List;
+import java.util.Map;
 
 /**
  * 小区表 服务实现类
@@ -39,9 +48,35 @@
 @Service
 public class DistrictServiceImpl extends ServiceImpl<DistrictMapper, DistrictEntity> implements IDistrictService {
 
+	@Autowired
+	private IGridService gridService;
+
+	@Autowired
+	private IDeptService deptService;
+
 	@Override
 	public IPage<DistrictVO> selectDistrictPage(IPage<DistrictVO> page, DistrictVO district) {
-		return page.setRecords(baseMapper.selectDistrictPage(page, district));
+		Dept dept = deptService.getById(AuthUtil.getDeptId());
+		if (null!=dept){
+			district.setRegionCode(dept.getRegionCode());
+		}
+		List<DistrictVO> districtVOS = baseMapper.selectDistrictPage(page, district);
+		// 遍历
+		for (DistrictVO districtVO : districtVOS) {
+			// 设置对应的网格名称
+			List<GridVO> gridVO = gridService.getGridListByAoiCode(districtVO.getAoiCode());
+			if (gridVO.size()>0) {
+				StringBuilder builder = new StringBuilder();
+				for (GridVO vo : gridVO) {
+					builder.append(vo.getGridName()).append(",");
+				}
+				String bui = builder.toString();
+				String substring = bui.substring(0, bui.length() - 1);
+				districtVO.setGridName(substring);
+			}
+		}
+		// 返回
+		return page.setRecords(districtVOS);
 	}
 
 	/**
@@ -51,7 +86,14 @@
 	 */
 	@Override
 	public Object getDistrictTree(DistrictVO district) {
-		List<TreeStringNode> stringNodeTree = NodeTreeUtil.getStringNodeTree(baseMapper.getDistrictTree(district));
+		// String userRole = AuthUtil.getUserRole();
+		// if (userRole.equals("wygly")) {
+		// 	IPropertyDistrictUserService bean = SpringUtils.getBean(IPropertyDistrictUserService.class);
+		// 	List<String> strings = bean.selectPropertyDistrictByUserId(AuthUtil.getUserId());
+		// 	district.setDistrictIdList(strings);
+		// }
+		Map<String, TreeStringNode> districtTree = baseMapper.getDistrictTree(district);
+		List<TreeStringNode> stringNodeTree = NodeTreeUtil.getStringNodeTree(districtTree);
 		stringNodeTree.forEach(node -> recursion(node));
 		return stringNodeTree;
 	}
@@ -68,4 +110,15 @@
 			node.setChildren(null);
 		}
 	}
+
+	/**
+	 * 小区表 自定义获取详情
+	 * @param district
+	 * @return
+	 */
+	@Override
+	public DistrictVO getDetail(DistrictVO district) {
+		// 小区自定义获取详情查询并返回
+		return baseMapper.getDetail(district);
+	}
 }

--
Gitblit v1.9.3