From d49df6e5b16eb16e47ea9a3bc0aa35f017835e02 Mon Sep 17 00:00:00 2001
From: xuhy <3313886187@qq.com>
Date: 星期一, 28 十月 2024 15:34:33 +0800
Subject: [PATCH] 修改

---
 ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/TChargingPileNotificationServiceImpl.java |   12 ++++++++++--
 1 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/TChargingPileNotificationServiceImpl.java b/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/TChargingPileNotificationServiceImpl.java
index 8297e9e..9bead20 100644
--- a/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/TChargingPileNotificationServiceImpl.java
+++ b/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/TChargingPileNotificationServiceImpl.java
@@ -15,6 +15,8 @@
 import com.ruoyi.other.api.feignClient.UserSiteClient;
 import com.ruoyi.system.api.domain.SysUser;
 import com.ruoyi.system.api.feignClient.SysUserClient;
+import com.ruoyi.system.api.feignClient.SysUserRoleClient;
+import com.ruoyi.system.api.model.SysUserRoleVo;
 import org.springframework.stereotype.Service;
 
 import javax.annotation.Resource;
@@ -44,6 +46,9 @@
 	
 	@Resource
 	private RoleSiteClient roleSiteClient;
+
+	@Resource
+	private SysUserRoleClient sysUserRoleClient;
 	
 	
 	
@@ -71,13 +76,15 @@
 				//非管理员需要根据角色和用户配置查询允许的站点数据
 				if(!SecurityUtils.isAdmin(SecurityUtils.getUserId())){
 					List<Integer> data = userSiteClient.getSiteIds(sysUser.getUserId()).getData();
-					List<Integer> data1 = roleSiteClient.getSiteIds(sysUser.getRoleId()).getData();
+					List<SysUserRoleVo> data2 = sysUserRoleClient.getRoleByUserId(sysUser.getUserId()).getData();
+					List<Integer> data1 = roleSiteClient.getSiteIds(data2.get(0).getRoleId()).getData();
 					data.addAll(data1);
 					siteIds = new HashSet<>(data);
 				}
 			}
 		}else{
-			siteIds = new HashSet<>(siteId);
+			siteIds = new HashSet<>();
+			siteIds.add(siteId);
 		}
 		PageInfo<TChargingPileNotification> pageInfo = new PageInfo<>(basePage.getPageCurr(), basePage.getPageSize());
 		List<TChargingPileNotification> list = this.baseMapper.chargingPileNotificationPageList(pageInfo, siteIds);
@@ -102,6 +109,7 @@
 				notification.setContent(data);
 				break;
 		}
+		this.save(notification);
 	}
 
 }

--
Gitblit v1.7.1