From 2550a534393ffbed3744a2e9877ad53f5357b1dd Mon Sep 17 00:00:00 2001
From: linwei <872216696@qq.com>
Date: Fri, 19 Jan 2024 15:21:43 +0800
Subject: [PATCH] 物业公司只查询自己公司发布的公告

---
 src/main/java/org/springblade/modules/article/service/impl/ArticleServiceImpl.java |   14 ++++++++++++++
 1 files changed, 14 insertions(+), 0 deletions(-)

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 149c022..73b9de1 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
@@ -1,11 +1,16 @@
 package org.springblade.modules.article.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.utils.SpringUtils;
+import org.springblade.core.secure.utils.AuthUtil;
 import org.springblade.modules.article.entity.Article;
 import org.springblade.modules.article.mapper.ArticleMapper;
 import org.springblade.modules.article.service.ArticleService;
 import org.springblade.modules.article.vo.ArticleVO;
+import org.springblade.modules.property.entity.PropertyCompanyDistrictEntity;
+import org.springblade.modules.property.service.IPropertyCompanyDistrictService;
 import org.springframework.stereotype.Service;
 
 import java.util.List;
@@ -28,6 +33,15 @@
 	 */
 	@Override
 	public IPage<Article> selectArticlePage(IPage<Article> page, ArticleVO article) {
+		String userRole = AuthUtil.getUserRole();
+		// 物业身份,只查询该物业公司下的公告
+		if (userRole.contains("wygly")) {
+			IPropertyCompanyDistrictService bean = SpringUtils.getBean(IPropertyCompanyDistrictService.class);
+			PropertyCompanyDistrictEntity one = bean.getOne(Wrappers.<PropertyCompanyDistrictEntity>lambdaQuery()
+				.eq(PropertyCompanyDistrictEntity::getUserId, AuthUtil.getUserId()).last("limit 1"));
+			article.setDistrictId(one.getDistrictId());
+		}
+
 		return page.setRecords(baseMapper.selectArticlePage(page, article));
 	}
 

--
Gitblit v1.9.3