From 0af5efbcac903cdc6ba1e7978cb9ac32220e8cde Mon Sep 17 00:00:00 2001
From: xuhy <3313886187@qq.com>
Date: 星期四, 31 十月 2024 16:05:36 +0800
Subject: [PATCH] 修改

---
 ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/TChargingPileNotificationServiceImpl.java |   20 ++++++++++++++------
 1 files changed, 14 insertions(+), 6 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..0c55471 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);
@@ -91,17 +98,18 @@
 		switch (type){
 			case 1:
 				// 验证码
-				notification.setPhone(phone);
-				notification.setContent(data);
+				notification.setSiteId(null);
+				notification.setChargingPileId(null);
 				break;
 			default:
 				// 停车占位
 				notification.setSiteId(siteId);
 				notification.setChargingPileId(chargingPileId);
-				notification.setPhone(phone);
-				notification.setContent(data);
 				break;
 		}
+		notification.setPhone(phone);
+		notification.setContent(data);
+		this.save(notification);
 	}
 
 }

--
Gitblit v1.7.1