From 8b7258c9427882bb1798f1502eaa35184c6e374e Mon Sep 17 00:00:00 2001
From: linwe <872216996@qq.com>
Date: Fri, 09 Aug 2024 14:29:18 +0800
Subject: [PATCH] 短信指定楼栋发送

---
 src/main/java/org/springblade/modules/district/service/impl/DistrictServiceImpl.java |   70 ++++++++++++++++++++++++++--------
 1 files changed, 53 insertions(+), 17 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 b25ec07..aef457b 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
@@ -16,29 +16,36 @@
  */
 package org.springblade.modules.district.service.impl;
 
+import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import org.springblade.common.cache.SysCache;
-import org.springblade.common.node.TreeIntegerNode;
 import org.springblade.common.node.TreeStringNode;
+import org.springblade.common.param.CommonParamSet;
 import org.springblade.common.utils.NodeTreeUtil;
+import org.springblade.common.utils.RoleUtil;
 import org.springblade.common.utils.SpringUtils;
 import org.springblade.core.secure.utils.AuthUtil;
+import org.springblade.core.tool.utils.SpringUtil;
 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.district.vo.DistrictVO;
 import org.springblade.modules.grid.service.IGridService;
 import org.springblade.modules.grid.vo.GridVO;
+import org.springblade.modules.property.entity.PropertyCompanyDistrictEntity;
+import org.springblade.modules.property.entity.PropertyCompanyEntity;
+import org.springblade.modules.property.service.IPropertyCompanyDistrictService;
+import org.springblade.modules.property.service.IPropertyCompanyService;
 import org.springblade.modules.property.service.IPropertyDistrictUserService;
-import org.springblade.modules.system.entity.Dept;
-import org.springblade.modules.system.service.IDeptService;
+import org.springblade.modules.task.vo.TaskReportForRepairsVO;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
-import com.baomidou.mybatisplus.core.metadata.IPage;
 
+import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
+import java.util.stream.Collectors;
 
 /**
  * 小区表 服务实现类
@@ -54,14 +61,16 @@
 
 	@Override
 	public IPage<DistrictVO> selectDistrictPage(IPage<DistrictVO> page, DistrictVO district) {
-		List<String> regionChildCodesList = SysCache.getRegionChildCodesByDeptId(AuthUtil.getDeptId());
-		Integer isAdministrator = AuthUtil.isAdministrator()==true?1:2;
-		List<DistrictVO> districtVOS = baseMapper.selectDistrictPage(page, district,regionChildCodesList,isAdministrator);
+		CommonParamSet invoke = new CommonParamSet<>().invoke(DistrictVO.class, district);
+		List<DistrictVO> districtVOS = baseMapper.selectDistrictPage(page,
+			district,
+			invoke.getRegionChildCodesList(),
+			invoke.getIsAdministrator());
 		// 遍历
 		for (DistrictVO districtVO : districtVOS) {
 			// 设置对应的网格名称
 			List<GridVO> gridVO = gridService.getGridListByAoiCode(districtVO.getAoiCode());
-			if (gridVO.size()>0) {
+			if (gridVO.size() > 0) {
 				StringBuilder builder = new StringBuilder();
 				for (GridVO vo : gridVO) {
 					builder.append(vo.getGridName()).append(",");
@@ -77,18 +86,43 @@
 
 	/**
 	 * 获取小区树
+	 *
 	 * @param district
 	 * @return
 	 */
 	@Override
 	public Object getDistrictTree(DistrictVO 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);
+		// 判断角色,物业角色只能查询当前小区的
+		String userRole = AuthUtil.getUserRole();
+		CommonParamSet invoke = new CommonParamSet<>().invoke(DistrictVO.class, district);
+
+		if (RoleUtil.isProperty(userRole)) {
+			if (district.getFilterFlag().equals(1)) {
+				// 查询小区id
+				IPropertyDistrictUserService propertyDistrictUserService = SpringUtils.getBean(IPropertyDistrictUserService.class);
+				List<String> districtIds = propertyDistrictUserService.selectPropertyDistrictByUserId(AuthUtil.getUserId());
+				// 通过用户机构查询用户的物业公司
+				// 通过用户机构查询用户的物业公司
+				IPropertyCompanyService bean = SpringUtil.getBean(IPropertyCompanyService.class);
+				PropertyCompanyEntity one = bean.getOne(Wrappers.<PropertyCompanyEntity>lambdaQuery()
+					.eq(PropertyCompanyEntity::getDeptId, AuthUtil.getDeptId()).last("limit 1"));
+				if (one != null) {
+					IPropertyCompanyDistrictService bean2 = SpringUtils.getBean(IPropertyCompanyDistrictService.class);
+					// 通过物业公司,查询小区
+					List<PropertyCompanyDistrictEntity> list = bean2.list(Wrappers.<PropertyCompanyDistrictEntity>lambdaQuery()
+						.eq(PropertyCompanyDistrictEntity::getPropertyCompanyId, one.getId()));
+					if (list.size() > 0) {
+						List<String> collect = list.stream().map(i -> i.getDistrictId()).collect(Collectors.toList());
+						districtIds.addAll(collect);
+					}
+				}
+				district.setDistrictIdList(districtIds);
+				if (districtIds.size() == 0) {
+					return new ArrayList<>();
+				}
+			}
+		}
+		Map<String, TreeStringNode> districtTree = baseMapper.getDistrictTree(district,invoke.getRegionChildCodesList(),invoke.getIsAdministrator());
 		List<TreeStringNode> stringNodeTree = NodeTreeUtil.getStringNodeTree(districtTree);
 		stringNodeTree.forEach(node -> recursion(node));
 		return stringNodeTree;
@@ -97,6 +131,7 @@
 
 	/**
 	 * 去除空的数据组
+	 *
 	 * @param node
 	 */
 	private void recursion(TreeStringNode node) {
@@ -109,6 +144,7 @@
 
 	/**
 	 * 小区表 自定义获取详情
+	 *
 	 * @param district
 	 * @return
 	 */

--
Gitblit v1.9.3