From fc938792642cb1b74ad10c25693a9756bf528adb Mon Sep 17 00:00:00 2001
From: ludc
Date: 星期三, 06 九月 2023 14:32:56 +0800
Subject: [PATCH] 代码提交

---
 Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/service/impl/StrategyServiceImpl.java |   35 ++++++++++++++++++++++++++++++-----
 1 files changed, 30 insertions(+), 5 deletions(-)

diff --git a/Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/service/impl/StrategyServiceImpl.java b/Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/service/impl/StrategyServiceImpl.java
index 45fd332..906f470 100644
--- a/Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/service/impl/StrategyServiceImpl.java
+++ b/Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/service/impl/StrategyServiceImpl.java
@@ -1,9 +1,11 @@
 package com.vci.ubcs.system.service.impl;
 
+import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.vci.ubcs.common.constant.CommonConstant;
+import com.vci.ubcs.common.cache.NacosConfigCache;
 import com.vci.ubcs.system.entity.Strategy;
 import com.vci.ubcs.system.mapper.StrategyMapper;
 import com.vci.ubcs.system.service.IStrategyService;
@@ -11,8 +13,8 @@
 import org.springblade.core.cache.utils.CacheUtil;
 import org.springblade.core.log.exception.ServiceException;
 import org.springblade.core.mp.support.Query;
+import org.springblade.core.secure.utils.AuthUtil;
 import org.springblade.core.tool.utils.Func;
-import org.springframework.data.domain.PageImpl;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
@@ -41,6 +43,12 @@
 	private IUserClient userClient;
 
 	/**
+	 * nacos涓婃媺鍙栫殑閰嶇疆
+	 */
+	@Resource
+	private NacosConfigCache nacosConfigCache;
+
+	/**
 	 * 閫氳繃ID鏌ヨ鍗曟潯鏁版嵁
 	 *
 	 * @param id 涓婚敭
@@ -57,7 +65,13 @@
 	 */
 	@Override
 	public Strategy queryByIsDefault() {
-		return this.strategyMapper.queryByIsDefault();
+		return this.strategyMapper.queryByIsDefault(AuthUtil.getTenantId()
+			.equals(
+				nacosConfigCache
+					.getAdminUserInfo()
+					.getTenantId()
+			) ? null:AuthUtil.getTenantId()
+		);
 	}
 
 	/**
@@ -67,9 +81,20 @@
 	 * @return 鏌ヨ缁撴灉
 	 */
 	@Override
-	public PageImpl<Strategy> queryAllByPage(Query query) {
+	public IPage<Strategy> queryAllByPage(Query query) {
 		Page<Strategy> strategyPage = new Page<>(query.getCurrent(), query.getSize());
-		return new PageImpl<>(this.strategyMapper.queryAllByPage(strategyPage));
+		// 娣诲姞绉熸埛鏌ヨ鏉′欢
+		IPage<Strategy> strategyIPage = this.strategyMapper.queryAllByPage(
+			strategyPage,
+			AuthUtil.getTenantId()
+				.equals(
+					nacosConfigCache
+					.getAdminUserInfo()
+					.getTenantId()
+				)
+			? null:AuthUtil.getTenantId()
+		);
+		return strategyIPage;
 	}
 
 	/**
@@ -131,7 +156,7 @@
 			if(Func.isNotEmpty(oldIsDefaultStrategy)){
 				temp2 = updateUserStrategyDefault(temp1, oldIsDefaultStrategy);
 			}
-			return temp1 && temp2;
+			return temp1 || temp2;
 		}
 	}
 

--
Gitblit v1.9.3