From b8c20abe5c8a1ebd5227d5948ba310606fde5dc7 Mon Sep 17 00:00:00 2001
From: 无关风月 <443237572@qq.com>
Date: 星期一, 16 十二月 2024 09:01:09 +0800
Subject: [PATCH] 合作商权限

---
 ruoyi-service/ruoyi-order/src/main/java/com/ruoyi/order/service/impl/TChargingOrderServiceImpl.java |   85 ++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 81 insertions(+), 4 deletions(-)

diff --git a/ruoyi-service/ruoyi-order/src/main/java/com/ruoyi/order/service/impl/TChargingOrderServiceImpl.java b/ruoyi-service/ruoyi-order/src/main/java/com/ruoyi/order/service/impl/TChargingOrderServiceImpl.java
index c1898d6..e21b823 100644
--- a/ruoyi-service/ruoyi-order/src/main/java/com/ruoyi/order/service/impl/TChargingOrderServiceImpl.java
+++ b/ruoyi-service/ruoyi-order/src/main/java/com/ruoyi/order/service/impl/TChargingOrderServiceImpl.java
@@ -60,6 +60,8 @@
 import com.ruoyi.payment.api.model.RefundResp;
 import com.ruoyi.payment.api.model.WxPaymentRefundModel;
 import com.ruoyi.payment.api.vo.*;
+import com.ruoyi.system.api.domain.SysUser;
+import com.ruoyi.system.api.feignClient.SysUserClient;
 import io.seata.spring.annotation.GlobalTransactional;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -100,7 +102,8 @@
 	private ChargingGunClient chargingGunClient;
 	@Resource
 	private TSettlementConfirmMapper tSettlementConfirmMapper;
-
+	@Resource
+	private PartnerClient partnerClient;
 	@Resource
 	private SiteClient siteClient;
 	
@@ -1212,6 +1215,9 @@
 			dto.setUserIds(data);
 		}
 		Long userId = tokenService.getLoginUser().getUserid();
+		// 查询当前登陆人按钮权限
+		SysUser sysUser = sysUserClient.getSysUser(tokenService.getLoginUser().getUserid()).getData();
+		Integer roleType = sysUser.getRoleType();
 		//如果没传siteId,获取当前登陆人所有的siteIds
 		List<Integer> siteIds = new ArrayList<>();
 		if (dto.getSiteId()==null){
@@ -1226,7 +1232,22 @@
 		}
 		if (siteIds.isEmpty()){
 			siteIds.add(-1);
+		}else{
+			if (roleType == 2){
+				List<Integer> integers = new ArrayList<>();
+				for (Integer siteId : siteIds) {
+					// 校验有没有这个站点的权限
+					List<Boolean> t1= partnerClient.getChargingOrderMenu(sysUser.getObjectId(),siteId).getData();
+
+					Boolean b = t1.get(1);
+					if (b){
+						integers.add(siteId);
+					}
+				}
+				siteIds = integers;
+			}
 		}
+		if (siteIds.isEmpty())siteIds.add(-1);
 		dto.setSiteIds(siteIds);
 		List<ChargingOrderVO> list = this.baseMapper.chargingOrder(pageInfo,dto,startTime1,startTime2,endTime1,endTime2);
 		List<ChargingOrderVO> list1 = this.baseMapper.chargingOrder(pageInfo1,dto,startTime1,startTime2,endTime1,endTime2);
@@ -1238,7 +1259,16 @@
 		 BigDecimal commissionMoney = new BigDecimal("0");
 		 BigDecimal refundMoney = new BigDecimal("0");
 		 BigDecimal paymentMoney = new BigDecimal("0");
+
 		for (ChargingOrderVO chargingOrderVO : list) {
+			if (roleType == 2){
+				for (Integer siteId : siteIds) {
+					// 校验有没有这个站点的权限
+					List<Boolean> t1= partnerClient.getChargingOrderMenu(sysUser.getObjectId(),siteId).getData();
+					chargingOrderVO.setAuthRecord(t1.get(0));
+				}
+
+			}
 			chargingOrderVO.setCommissionAmount(chargingOrderVO.getOrderAmount()!=null?chargingOrderVO.getOrderAmount().multiply(new BigDecimal("0.006")):new BigDecimal("0"));
 			chargingOrderVO.setPlatFormMoney(chargingOrderVO.getOrderAmount()!=null?chargingOrderVO.getOrderAmount().multiply(new BigDecimal("0.006")).setScale(2,BigDecimal.ROUND_DOWN):new BigDecimal("0"));
 			chargingOrderVO.setUid(chargingOrderVO.getId()+"");
@@ -1359,8 +1389,8 @@
 	}
 
 	@Override
-	public Map<String, Object> countAll(LocalDate sixBefore) {
-		return this.baseMapper.countAll(sixBefore);
+	public Map<String, Object> countAll(LocalDate sixBefore,List<Integer> siteIds) {
+		return this.baseMapper.countAll(sixBefore,siteIds);
 	}
 
 	@Override
@@ -1388,7 +1418,8 @@
 		return this.baseMapper.getYearData(chargingOrderIds);
 	}
 
-	
+	@Resource
+	private SysUserClient sysUserClient;
 	@Override
 	public ChargingOrderTimeVO chargingList(ChargingListQuery dto) {
 		String startTime1 = null;
@@ -1433,6 +1464,9 @@
 			}
 		}
 		Long userId = tokenService.getLoginUser().getUserid();
+		// 查询当前登陆人按钮权限
+		SysUser sysUser = sysUserClient.getSysUser(tokenService.getLoginUser().getUserid()).getData();
+		Integer roleType = sysUser.getRoleType();
 		//如果没传siteId,获取当前登陆人所有的siteIds
 		List<Integer> siteIds = new ArrayList<>();
 		if (dto.getSiteId()==null){
@@ -1447,11 +1481,31 @@
 		}
 		if (siteIds.isEmpty()){
 			siteIds.add(-1);
+		}else{
+			if (roleType == 2){
+				List<Integer> integers = new ArrayList<>();
+				for (Integer siteId : siteIds) {
+					// 校验有没有这个站点的权限
+					List<Boolean> t1= partnerClient.getChargingOrderMenu(sysUser.getObjectId(),siteId).getData();
+
+					Boolean b = t1.get(1);
+					if (b){
+						integers.add(siteId);
+					}
+				}
+				siteIds = integers;
+			}
 		}
 		dto.setSiteIds(siteIds);
 		PageInfo<ChargingOrderListVO> pageInfo = new PageInfo<>(dto.getPageCurr(),dto.getPageSize());
 		List<ChargingOrderListVO> list = this.baseMapper.chargingList(pageInfo,dto,startTime1,startTime2,endTime1,endTime2);
+
 		for (ChargingOrderListVO chargingOrderListVO : list) {
+			if (roleType==2){
+				List<Boolean> data = partnerClient.getChargingListMenu(sysUser.getObjectId(), chargingOrderListVO.getSiteId()).getData();
+				chargingOrderListVO.setAuthInfo(data.get(0));
+				chargingOrderListVO.setAuthCurve(data.get(1));
+			}
 			chargingOrderListVO.setChargingCapacity(chargingOrderListVO.getElectricity());
 			BigDecimal bigDecimal = new BigDecimal("0.006");
 			if (chargingOrderListVO.getOrderAmount()!=null){
@@ -2718,6 +2772,9 @@
 		dto.setStartTime(startTime);
 		dto.setEndTime(endTime);
 		Long userId = tokenService.getLoginUser().getUserid();
+		// 查询当前登陆人按钮权限
+		SysUser sysUser = sysUserClient.getSysUser(tokenService.getLoginUser().getUserid()).getData();
+		Integer roleType = sysUser.getRoleType();
 		//如果没传siteId,获取当前登陆人所有的siteIds
 		List<Integer> siteIds = new ArrayList<>();
 		if (dto.getSiteIds()==null){
@@ -2732,11 +2789,31 @@
 		}
 		if (siteIds.isEmpty()){
 			siteIds.add(-1);
+		}else{
+			if (roleType == 2){
+				List<Integer> integers = new ArrayList<>();
+				for (Integer siteId : siteIds) {
+					// 校验有没有这个站点的权限
+					List<Boolean> t1= partnerClient.getSettlementListMenu(sysUser.getObjectId(),siteId).getData();
+
+					Boolean b = t1.get(2);
+					if (b){
+						integers.add(siteId);
+					}
+				}
+				siteIds = integers;
+			}
 		}
 		dto.setSiteIdss(siteIds);
 		PageInfo<TSettlementConfirm> pageInfo = new PageInfo<>(dto.getPageCurr(),dto.getPageSize());
 		List<TSettlementConfirm> list = tSettlementConfirmMapper.settlementList(pageInfo,dto);
+
 		for (TSettlementConfirm tSettlementConfirm : list) {
+			if (roleType==2){
+				List<Boolean> data = partnerClient.getSettlementListMenu(sysUser.getObjectId(), tSettlementConfirm.getSiteId()).getData();
+				tSettlementConfirm.setAuthDownLoad(data.get(0));
+				tSettlementConfirm.setAuthDelete(data.get(1));
+			}
 			tSettlementConfirm.setUid(tSettlementConfirm.getId().toString());
 			List<Site> data1 = siteClient.getSiteByIds(Arrays.asList(tSettlementConfirm.getSiteId())).getData();
 			if (!data1.isEmpty()){

--
Gitblit v1.7.1