From 6f1a56eb02cb6eb7848e2df6773bc5abfee11efd Mon Sep 17 00:00:00 2001
From: puzhibing <393733352@qq.com>
Date: 星期二, 08 十月 2024 15:13:03 +0800
Subject: [PATCH] 合併代碼

---
 ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/service/impl/SiteServiceImpl.java |   32 ++++++++++++++++++++++----------
 1 files changed, 22 insertions(+), 10 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..e83c91e 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
@@ -8,6 +8,7 @@
 import com.ruoyi.chargingPile.api.model.Site;
 import com.ruoyi.chargingPile.api.model.TChargingGun;
 import com.ruoyi.chargingPile.api.model.TChargingPile;
+import com.ruoyi.chargingPile.api.model.TParkingLot;
 import com.ruoyi.chargingPile.api.query.GetSiteList;
 import com.ruoyi.chargingPile.api.query.SiteQuery;
 import com.ruoyi.chargingPile.api.vo.SiteDetailVO;
@@ -15,10 +16,7 @@
 import com.ruoyi.chargingPile.api.vo.TAccountingStrategyVO;
 import com.ruoyi.chargingPile.domain.SiteMenu;
 import com.ruoyi.chargingPile.mapper.SiteMapper;
-import com.ruoyi.chargingPile.service.IPartnerService;
-import com.ruoyi.chargingPile.service.ISiteService;
-import com.ruoyi.chargingPile.service.TChargingGunService;
-import com.ruoyi.chargingPile.service.TChargingPileService;
+import com.ruoyi.chargingPile.service.*;
 import com.ruoyi.common.core.utils.ServletUtils;
 import com.ruoyi.common.core.utils.StringUtils;
 import com.ruoyi.common.core.web.domain.AjaxResult;
@@ -36,9 +34,11 @@
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Propagation;
 import org.springframework.transaction.annotation.Transactional;
+import org.springframework.util.CollectionUtils;
 
 import javax.annotation.Resource;
 import java.util.*;
+import java.util.stream.Collectors;
 
 /**
  * @author zhibing.pu
@@ -72,6 +72,8 @@
 	private VipClient vipClient;
 	@Resource
 	private SysUserRoleClient sysUserRoleClient;
+	@Resource
+	private TParkingLotService parkingLotService;
 
 	
 	
@@ -133,7 +135,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 +259,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);
@@ -283,7 +287,7 @@
 			TVip vip = vipClient.getInfo1(appUser.getVipId()).getData();
 			if(Objects.nonNull(vip) && vip.getType() == 2){
 				list.forEach(item -> {
-					item.setVipElectrovalence(vip.getDiscount().multiply(item.getElectrovalence()));
+					item.setVipElectrovalence(vip.getDiscount().multiply(item.getServiceCharge()).add(item.getElectrovalenceOriginal()));
 				});
 			}
 		}
@@ -310,10 +314,17 @@
 			//非管理员需要根据角色和用户配置查询允许的站点数据
 			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);
 			}
+		}
+		// 查询所有的停车场的站点id
+		List<TParkingLot> list = parkingLotService.list();
+		List<Integer> siteIds = list.stream().map(TParkingLot::getSiteId).collect(Collectors.toList());
+		if(!CollectionUtils.isEmpty(ids) && !CollectionUtils.isEmpty(siteIds)){
+			ids.addAll(siteIds);
 		}
 		return this.baseMapper.getSiteListParkLot(ids);
 	}
@@ -332,7 +343,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