From 5745a87f3a139ea2bbdaf4704978e4c151ecc45a Mon Sep 17 00:00:00 2001
From: linwei <872216696@qq.com>
Date: Tue, 23 Jan 2024 17:39:48 +0800
Subject: [PATCH] 物业管理管和项目经理数据过滤

---
 src/main/java/org/springblade/modules/property/service/impl/PropertyCompanyDistrictServiceImpl.java |   35 +++++++++++++++++++++++++++++++----
 1 files changed, 31 insertions(+), 4 deletions(-)

diff --git a/src/main/java/org/springblade/modules/property/service/impl/PropertyCompanyDistrictServiceImpl.java b/src/main/java/org/springblade/modules/property/service/impl/PropertyCompanyDistrictServiceImpl.java
index e64eb61..cc4f04d 100644
--- a/src/main/java/org/springblade/modules/property/service/impl/PropertyCompanyDistrictServiceImpl.java
+++ b/src/main/java/org/springblade/modules/property/service/impl/PropertyCompanyDistrictServiceImpl.java
@@ -1,27 +1,31 @@
 package org.springblade.modules.property.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.apache.commons.lang3.StringUtils;
 import org.flowable.idm.engine.impl.persistence.entity.UserEntity;
+import org.springblade.common.utils.SpringUtils;
+import org.springblade.core.secure.utils.AuthUtil;
 import org.springblade.core.tool.utils.Func;
 import org.springblade.core.tool.utils.SpringUtil;
 import org.springblade.modules.property.entity.PropertyCompanyDistrictEntity;
+import org.springblade.modules.property.entity.PropertyCompanyEntity;
 import org.springblade.modules.property.entity.PropertyDistrictUserEntity;
-import org.springblade.modules.property.service.IPropertyDistrictUserService;
-import org.springblade.modules.property.vo.PropertyCompanyDistrictVO;
 import org.springblade.modules.property.mapper.PropertyCompanyDistrictMapper;
 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.property.vo.PropertyCompanyDistrictVO;
 import org.springblade.modules.system.entity.User;
-import org.springblade.modules.system.entity.UserDept;
 import org.springblade.modules.system.service.IUserService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
-import com.baomidou.mybatisplus.core.metadata.IPage;
 import org.springframework.transaction.annotation.Transactional;
 
 import java.util.ArrayList;
 import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * 物业派驻小区表 服务实现类
@@ -44,6 +48,27 @@
 	 */
 	@Override
 	public IPage<PropertyCompanyDistrictVO> selectPropertyCompanyDistrictPage(IPage<PropertyCompanyDistrictVO> page, PropertyCompanyDistrictVO propertyCompanyDistrict) {
+		// 物业 查询用户管理的小区
+		String userRole = AuthUtil.getUserRole();
+		if (userRole.contains("wygly") || userRole.contains("wyxmjl")) {
+			// 查询小区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()));
+			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);
+			}
+			propertyCompanyDistrict.setDistrictIds(districtIds);
+		}
+		// 街道
+
 		return page.setRecords(baseMapper.selectPropertyCompanyDistrictPage(page, propertyCompanyDistrict));
 	}
 
@@ -89,6 +114,8 @@
 	 */
 	private boolean submitPropertyDistrictUser(PropertyCompanyDistrictEntity propertyCompanyDistrict) {
 		List<Long> userIdList = Func.toLongList(propertyCompanyDistrict.getUserId());
+		// 把项目经理也加到物业用户关联表
+		userIdList.add(Long.valueOf(propertyCompanyDistrict.getPrincipal()));
 		List<PropertyDistrictUserEntity> propertyDistrictUserEntityList = new ArrayList<>();
 		userIdList.forEach(userId -> {
 			PropertyDistrictUserEntity propertyDistrictUserEntity = new PropertyDistrictUserEntity();

--
Gitblit v1.9.3