From b627e9b177de4d16f38dcb19941722c8ca7a8edc Mon Sep 17 00:00:00 2001
From: puzhibing <393733352@qq.com>
Date: 星期一, 30 九月 2024 11:55:32 +0800
Subject: [PATCH] 合併代碼

---
 ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/SiteServiceImpl.java |   14 +++++++++-----
 1 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/SiteServiceImpl.java b/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/SiteServiceImpl.java
index 97458c6..80b0d10 100644
--- a/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/SiteServiceImpl.java
+++ b/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/SiteServiceImpl.java
@@ -133,7 +133,8 @@
 			//非管理员需要根据角色和用户配置查询允许的站点数据
 			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);
 				ids = new HashSet<>(data);
 			}
@@ -256,9 +257,10 @@
 	public AjaxResult delSite(Integer[] ids) {
 		//查询是否有关联数据
 		//充电桩
-		long count1 = chargingPileService.count(new LambdaQueryWrapper<TChargingPile>().eq(TChargingPile::getSiteId, Arrays.asList(ids)).eq(TChargingPile::getDelFlag, 0));
+		long count1 = chargingPileService.count(new LambdaQueryWrapper<TChargingPile>().in(TChargingPile::getSiteId, Arrays.asList(ids))
+				.eq(TChargingPile::getDelFlag, 0));
 		if(count1 > 0){
-			return AjaxResult.error("该站点有关联充电桩,删除失败!");
+			return AjaxResult.error("该站点已添加充电桩,不可删除。");
 		}
 		for (Integer id : ids) {
 			Site site = this.getById(id);
@@ -310,7 +312,8 @@
 			//非管理员需要根据角色和用户配置查询允许的站点数据
 			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);
 				ids = new HashSet<>(data);
 			}
@@ -332,7 +335,8 @@
 			//非管理员需要根据角色和用户配置查询允许的站点数据
 			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);
 				ids = new HashSet<>(data);
 			}

--
Gitblit v1.7.1