From 26ac1c8712346522ea1c3fb59be34f4d96544310 Mon Sep 17 00:00:00 2001
From: lin <sbla5888@163.com>
Date: Wed, 28 Feb 2024 12:02:42 +0800
Subject: [PATCH] 议事管理-物业服务人员数据未过滤有问题

---
 src/main/java/org/springblade/modules/property/service/impl/PropertyCompanyServiceImpl.java         |    3 +
 src/main/java/org/springblade/modules/district/service/impl/DistrictServiceImpl.java                |   45 +++++++++++-----------
 src/main/java/org/springblade/modules/task/service/impl/TaskReportForRepairsServiceImpl.java        |    3 +
 src/main/java/org/springblade/modules/property/service/impl/PropertyCapitalApplyServiceImpl.java    |    3 +
 src/main/java/org/springblade/modules/property/service/impl/PropertyCompanyDistrictServiceImpl.java |    3 +
 src/main/java/org/springblade/common/utils/RoleUtil.java                                            |   19 +++++++++
 src/main/java/org/springblade/modules/article/service/impl/ArticleServiceImpl.java                  |    3 +
 7 files changed, 52 insertions(+), 27 deletions(-)

diff --git a/src/main/java/org/springblade/common/utils/RoleUtil.java b/src/main/java/org/springblade/common/utils/RoleUtil.java
new file mode 100644
index 0000000..3c82975
--- /dev/null
+++ b/src/main/java/org/springblade/common/utils/RoleUtil.java
@@ -0,0 +1,19 @@
+package org.springblade.common.utils;
+
+import java.util.UUID;
+
+public class RoleUtil {
+
+
+	/**
+	 * 判断是否物业人员
+	 *
+	 * @return
+	 */
+	public static boolean isProperty(String userRole) {
+		if (userRole.contains("wygly") || userRole.contains("wyxmjl") || userRole.contains("wyfwry")) {
+			return true;
+		}
+		return false;
+	}
+}
diff --git a/src/main/java/org/springblade/modules/article/service/impl/ArticleServiceImpl.java b/src/main/java/org/springblade/modules/article/service/impl/ArticleServiceImpl.java
index 283b240..369ee7f 100644
--- a/src/main/java/org/springblade/modules/article/service/impl/ArticleServiceImpl.java
+++ b/src/main/java/org/springblade/modules/article/service/impl/ArticleServiceImpl.java
@@ -4,6 +4,7 @@
 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.utils.RoleUtil;
 import org.springblade.common.utils.SpringUtils;
 import org.springblade.core.secure.utils.AuthUtil;
 import org.springblade.core.tool.utils.SpringUtil;
@@ -50,7 +51,7 @@
 		// 	List<String> collect = list.stream().map(PropertyCompanyDistrictEntity::getDistrictId).collect(Collectors.toList());
 		// 	article.setDistrictIdList(collect);
 		// }
-		if (userRole.contains("wygly") || userRole.contains("wyxmjl")) {
+		if (RoleUtil.isProperty(userRole)) {
 			// 查询小区id
 			IPropertyDistrictUserService propertyDistrictUserService = SpringUtils.getBean(IPropertyDistrictUserService.class);
 			List<String> districtIds = propertyDistrictUserService.selectPropertyDistrictByUserId(AuthUtil.getUserId());
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 54edf40..a0c630e 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
@@ -22,6 +22,7 @@
 import org.springblade.common.cache.SysCache;
 import org.springblade.common.node.TreeStringNode;
 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;
@@ -89,30 +90,30 @@
 	public Object getDistrictTree(DistrictVO district) {
 		// 判断角色,物业角色只能查询当前小区的
 		String userRole = AuthUtil.getUserRole();
-		if (userRole.contains("wygly") || userRole.contains("wyxmjl")) {
-			 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()));
-			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);
+		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()));
+				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<>();
 				}
 			}
-			district.setDistrictIdList(districtIds);
-			if (districtIds.size() == 0) {
-				return new ArrayList<>();
-			}
-			 }
 		}
 		Map<String, TreeStringNode> districtTree = baseMapper.getDistrictTree(district);
 		List<TreeStringNode> stringNodeTree = NodeTreeUtil.getStringNodeTree(districtTree);
diff --git a/src/main/java/org/springblade/modules/property/service/impl/PropertyCapitalApplyServiceImpl.java b/src/main/java/org/springblade/modules/property/service/impl/PropertyCapitalApplyServiceImpl.java
index 60bb760..79302e8 100644
--- a/src/main/java/org/springblade/modules/property/service/impl/PropertyCapitalApplyServiceImpl.java
+++ b/src/main/java/org/springblade/modules/property/service/impl/PropertyCapitalApplyServiceImpl.java
@@ -27,6 +27,7 @@
 import org.slf4j.LoggerFactory;
 import org.springblade.common.cache.SysCache;
 import org.springblade.common.constant.CommonConstant;
+import org.springblade.common.utils.RoleUtil;
 import org.springblade.common.utils.SpringUtils;
 import org.springblade.core.log.exception.ServiceException;
 import org.springblade.core.secure.utils.AuthUtil;
@@ -85,7 +86,7 @@
 		Integer isAdministrator = AuthUtil.isAdministrator() == true ? 1 : 2;
 		// 判断角色,物业角色只能查询当前小区的
 		String userRole = AuthUtil.getUserRole();
-		if (userRole.contains("wygly") || userRole.contains("wyxmjl")) {
+		if (RoleUtil.isProperty(userRole)) {
 			// 查询小区id
 			IPropertyDistrictUserService propertyDistrictUserService = SpringUtils.getBean(IPropertyDistrictUserService.class);
 			List<String> districtIds = propertyDistrictUserService.selectPropertyDistrictByUserId(AuthUtil.getUserId());
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 3fcdef0..cde90fa 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
@@ -4,6 +4,7 @@
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import org.flowable.idm.engine.impl.persistence.entity.UserEntity;
+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.Func;
@@ -47,7 +48,7 @@
 	public IPage<PropertyCompanyDistrictVO> selectPropertyCompanyDistrictPage(IPage<PropertyCompanyDistrictVO> page, PropertyCompanyDistrictVO propertyCompanyDistrict) {
 		// 物业 查询用户管理的小区
 		String userRole = AuthUtil.getUserRole();
-		if (userRole.contains("wygly") || userRole.contains("wyxmjl")) {
+		if (RoleUtil.isProperty(userRole)) {
 			propertyCompanyDistrict.setDeptId(AuthUtil.getDeptId());
 			// 查询小区id
 			IPropertyDistrictUserService propertyDistrictUserService = SpringUtils.getBean(IPropertyDistrictUserService.class);
diff --git a/src/main/java/org/springblade/modules/property/service/impl/PropertyCompanyServiceImpl.java b/src/main/java/org/springblade/modules/property/service/impl/PropertyCompanyServiceImpl.java
index 0c24ca5..b0f6810 100644
--- a/src/main/java/org/springblade/modules/property/service/impl/PropertyCompanyServiceImpl.java
+++ b/src/main/java/org/springblade/modules/property/service/impl/PropertyCompanyServiceImpl.java
@@ -22,6 +22,7 @@
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import org.springblade.common.cache.SysCache;
 import org.springblade.common.constant.CommonConstant;
+import org.springblade.common.utils.RoleUtil;
 import org.springblade.common.utils.SpringUtils;
 import org.springblade.core.mp.support.Condition;
 import org.springblade.core.secure.utils.AuthUtil;
@@ -232,7 +233,7 @@
 	@Override
 	public PropertyCompanyDetailVO getDetailByDeptId() {
 		String userRole = AuthUtil.getUserRole();
-		if (userRole.contains("wygly")) {
+		if (RoleUtil.isProperty(userRole)) {
 			// 通过用户机构查询用户的物业公司
 			IPropertyCompanyService bean = SpringUtil.getBean(IPropertyCompanyService.class);
 			PropertyCompanyEntity propertyCompanyEntity = bean.getOne(Wrappers.<PropertyCompanyEntity>lambdaQuery().eq(PropertyCompanyEntity::getDeptId, AuthUtil.getDeptId()));
diff --git a/src/main/java/org/springblade/modules/task/service/impl/TaskReportForRepairsServiceImpl.java b/src/main/java/org/springblade/modules/task/service/impl/TaskReportForRepairsServiceImpl.java
index ad46aff..80e0a26 100644
--- a/src/main/java/org/springblade/modules/task/service/impl/TaskReportForRepairsServiceImpl.java
+++ b/src/main/java/org/springblade/modules/task/service/impl/TaskReportForRepairsServiceImpl.java
@@ -20,6 +20,7 @@
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import org.springblade.common.cache.SysCache;
 import org.springblade.common.param.CommonParamSet;
+import org.springblade.common.utils.RoleUtil;
 import org.springblade.common.utils.SpringUtils;
 import org.springblade.core.mp.base.BaseServiceImpl;
 import org.springblade.core.secure.utils.AuthUtil;
@@ -92,7 +93,7 @@
 		}
 		String userRole = AuthUtil.getUserRole();
 		List<String> aoiCodeList = new ArrayList<>();
-		if (userRole.contains("wygly") || userRole.contains("wyxmjl")|| userRole.contains("wyfwry")) {
+		if (RoleUtil.isProperty(userRole)) {
 			taskReportForRepairs.setRoleType("wy");
 			// 查询小区id
 			IPropertyDistrictUserService propertyDistrictUserService = SpringUtils.getBean(IPropertyDistrictUserService.class);

--
Gitblit v1.9.3