From a1d8ea1a72cddf7651742b87f042602ab006c879 Mon Sep 17 00:00:00 2001
From: linwe <872216996@qq.com>
Date: Sat, 11 Nov 2023 17:17:54 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'

---
 src/main/java/org/springblade/modules/house/service/impl/HouseRentalServiceImpl.java |   22 ++++++++++++++++++++--
 1 files changed, 20 insertions(+), 2 deletions(-)

diff --git a/src/main/java/org/springblade/modules/house/service/impl/HouseRentalServiceImpl.java b/src/main/java/org/springblade/modules/house/service/impl/HouseRentalServiceImpl.java
index ec3fd73..292a0f8 100644
--- a/src/main/java/org/springblade/modules/house/service/impl/HouseRentalServiceImpl.java
+++ b/src/main/java/org/springblade/modules/house/service/impl/HouseRentalServiceImpl.java
@@ -19,6 +19,7 @@
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import org.springblade.core.secure.utils.AuthUtil;
+import org.springblade.modules.grid.service.IGridService;
 import org.springblade.modules.house.entity.HouseRentalEntity;
 import org.springblade.modules.house.entity.HouseTenantEntity;
 import org.springblade.modules.house.service.IHouseTenantService;
@@ -51,6 +52,9 @@
 	@Autowired
 	private IHouseTenantService houseTenantService;
 
+	@Autowired
+	private IGridService gridService;
+
 	/**
 	 * 自定义分页查询
 	 * @param page
@@ -59,7 +63,14 @@
 	 */
 	@Override
 	public IPage<HouseRentalTenantVO> selectHouseRentalPage(IPage<HouseRentalTenantVO> page, HouseRentalTenantVO houseRental) {
-		return page.setRecords(baseMapper.selectHouseRentalPage(page, houseRental));
+		List<String> list = new ArrayList<>();
+		if (null!=houseRental.getRoleName() && !houseRental.getRoleName().equals("")){
+			if (houseRental.getRoleName().equals("网格员")){
+				// 查询对应的房屋地址code
+				list = gridService.getAddressCodeListByUserId(AuthUtil.getUserId());
+			}
+		}
+		return page.setRecords(baseMapper.selectHouseRentalPage(page, houseRental,list));
 	}
 
 	/**
@@ -196,8 +207,15 @@
 	 */
 	@Override
 	public Object getStatistics(HouseRentalTenantVO houseRental) {
+		List<String> list = new ArrayList<>();
+		if (null!=houseRental.getRoleName() && !houseRental.getRoleName().equals("")){
+			if (houseRental.getRoleName().equals("网格员")){
+				// 查询对应的房屋地址code
+				list = gridService.getAddressCodeListByUserId(AuthUtil.getUserId());
+			}
+		}
 		// 查询
-		List<HouseRentalStatistics> statistics = baseMapper.getStatistics(houseRental);
+		List<HouseRentalStatistics> statistics = baseMapper.getStatistics(houseRental,list);
 		// 返回
 		return statistics;
 	}

--
Gitblit v1.9.3