From dcd97d683792b5d47cc13966b24e3bbb92ee9854 Mon Sep 17 00:00:00 2001
From: Pu Zhibing <393733352@qq.com>
Date: 星期三, 11 十二月 2024 19:11:36 +0800
Subject: [PATCH] 修改安全相关功能和合作商权限功能

---
 ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/controller/SiteController.java |   81 ++++++++++++++++++++++++++--------------
 1 files changed, 53 insertions(+), 28 deletions(-)

diff --git a/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/controller/SiteController.java b/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/controller/SiteController.java
index 8b3b6ac..66b6498 100644
--- a/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/controller/SiteController.java
+++ b/ruoyi-service/ruoyi-chargingPile/src/main/java/com/ruoyi/chargingPile/controller/SiteController.java
@@ -18,7 +18,7 @@
 import com.ruoyi.chargingPile.api.query.SiteDetailQuery;
 import com.ruoyi.chargingPile.api.query.SiteQuery;
 import com.ruoyi.chargingPile.api.vo.*;
-import com.ruoyi.chargingPile.domain.SiteMenu;
+import com.ruoyi.chargingPile.api.domain.SiteMenu;
 import com.ruoyi.chargingPile.dto.ChargingPercentBack;
 import com.ruoyi.common.core.dto.ChargingOrderGroup;
 import com.ruoyi.common.core.dto.ChargingPercentProvinceDto;
@@ -30,6 +30,8 @@
 import com.ruoyi.common.core.web.page.PageInfo;
 import com.ruoyi.common.log.annotation.Log;
 import com.ruoyi.common.log.enums.BusinessType;
+import com.ruoyi.common.security.annotation.Logical;
+import com.ruoyi.common.security.annotation.RequiresPermissions;
 import com.ruoyi.common.security.service.TokenService;
 import com.ruoyi.common.security.utils.SecurityUtils;
 import com.ruoyi.order.api.feignClient.ChargingOrderClient;
@@ -54,7 +56,6 @@
 import javax.annotation.Resource;
 import javax.servlet.http.HttpServletRequest;
 import java.math.BigDecimal;
-import java.time.LocalDate;
 import java.time.LocalTime;
 import java.time.temporal.ChronoUnit;
 import java.util.*;
@@ -144,6 +145,11 @@
 				Long userId = tokenService.getLoginUserApplet().getUserId();
 				// 根据id查询用户信息
 				TAppUser appUser = appUserClient.getUserById(userId).getData();
+				if(Objects.nonNull(byId1.getDiscount())){
+					siteInfoVO.setVipElectrovalence((tAccountingStrategyDetail.getServiceCharge().multiply(byId1.getDiscount()).add(tAccountingStrategyDetail.getElectrovalence())).setScale(4, BigDecimal.ROUND_HALF_UP));
+				}else {
+					siteInfoVO.setVipElectrovalence((tAccountingStrategyDetail.getServiceCharge().add(tAccountingStrategyDetail.getElectrovalence())).setScale(4, BigDecimal.ROUND_HALF_UP));
+				}
 				if(Objects.nonNull(appUser)){
 //					TVip vip = vipClient.getInfo1(appUser.getVipId()).getData();
 					GetAppUserVipDetail getAppUserVipDetail = new GetAppUserVipDetail();
@@ -156,14 +162,8 @@
 						TVip vip = JSON.parseObject(vipJson, TVip.class);
 						if(Objects.nonNull(vip) && vip.getType() == 2){
 							siteInfoVO.setVipElectrovalence(vip.getDiscount().divide(new BigDecimal(10)).multiply(tAccountingStrategyDetail.getServiceCharge()).add(tAccountingStrategyDetail.getElectrovalence()));
-						}else {
-							siteInfoVO.setVipElectrovalence((tAccountingStrategyDetail.getServiceCharge().multiply(byId1.getDiscount()).add(tAccountingStrategyDetail.getElectrovalence())).setScale(4, BigDecimal.ROUND_HALF_UP));
 						}
-					}else {
-						siteInfoVO.setVipElectrovalence(tAccountingStrategyDetail.getElectrovalence().add(tAccountingStrategyDetail.getServiceCharge()).setScale(4, BigDecimal.ROUND_HALF_UP));
 					}
-				}else {
-					siteInfoVO.setVipElectrovalence(tAccountingStrategyDetail.getElectrovalence().add(tAccountingStrategyDetail.getServiceCharge()).setScale(4, BigDecimal.ROUND_HALF_UP));
 				}
 			}
 		}
@@ -196,6 +196,11 @@
 				Long userId = tokenService.getLoginUserApplet().getUserId();
 				// 根据id查询用户信息
 				TAppUser appUser = appUserClient.getUserById(userId).getData();
+				if(Objects.nonNull(byId1.getDiscount())){
+					siteInfoVO.setVipElectrovalence((tAccountingStrategyDetail.getServiceCharge().multiply(byId1.getDiscount()).add(tAccountingStrategyDetail.getElectrovalence())).setScale(4, BigDecimal.ROUND_HALF_UP));
+				}else {
+					siteInfoVO.setVipElectrovalence((tAccountingStrategyDetail.getServiceCharge().add(tAccountingStrategyDetail.getElectrovalence())).setScale(4, BigDecimal.ROUND_HALF_UP));
+				}
 				if(Objects.nonNull(appUser)){
 //					TVip vip = vipClient.getInfo1(appUser.getVipId()).getData();
 					GetAppUserVipDetail getAppUserVipDetail = new GetAppUserVipDetail();
@@ -208,14 +213,8 @@
 						TVip vip = JSON.parseObject(vipJson, TVip.class);
 						if(Objects.nonNull(vip) && vip.getType() == 2){
 							siteInfoVO.setVipElectrovalence(vip.getDiscount().divide(new BigDecimal(10)).multiply(tAccountingStrategyDetail.getServiceCharge()).add(tAccountingStrategyDetail.getElectrovalence()));
-						}else{
-							siteInfoVO.setVipElectrovalence((tAccountingStrategyDetail.getServiceCharge().multiply(byId1.getDiscount()).add(tAccountingStrategyDetail.getElectrovalence())).setScale(4, BigDecimal.ROUND_HALF_UP));
 						}
-					}else {
-						siteInfoVO.setVipElectrovalence(tAccountingStrategyDetail.getElectrovalence().add(tAccountingStrategyDetail.getServiceCharge()).setScale(4, BigDecimal.ROUND_HALF_UP));
 					}
-				}else {
-					siteInfoVO.setVipElectrovalence(tAccountingStrategyDetail.getElectrovalence().add(tAccountingStrategyDetail.getServiceCharge()).setScale(4, BigDecimal.ROUND_HALF_UP));
 				}
 			}
 		}
@@ -238,6 +237,9 @@
 		return R.ok(siteInfoVO);
 	}
 	
+	@RequiresPermissions(value = {"/systemUserManagement/update", "/site", "/platformRevenueAnalysis", "/chargingOperationAnalysis",
+			"/chargingUserAnalysis", "/operationMaintenanceAnalysis", "/chargingEquipmentMonitoring", "/activityExpense", "/chargeEvaluation/list",
+			"/parkingLot", "/workbench"}, logical = Logical.OR)
 	@GetMapping("/getSiteList")
 	@ApiOperation(value = "获取站点列表", tags = {"管理后台-站点管理"})
 	public AjaxResult<PageInfo<GetSiteListDTO>> getSiteList(GetSiteList siteList){
@@ -252,9 +254,10 @@
 		PageInfo<GetSiteListDTO> list = siteService.getSiteList1(userId);
 		return R.ok(list.getRecords());
 	}
-
-
-
+	
+	
+	
+	@RequiresPermissions(value = {"/chargeDistribution"}, logical = Logical.OR)
 	@GetMapping("/map/getSiteList")
 	@ApiOperation(value = "获取站点", tags = {"管理后台-数据分析-电站分析"})
 	public R<List<GetSiteListDTO>> getMapSiteList(GetSiteList siteList){
@@ -302,8 +305,9 @@
 		}
 		return R.ok(list.getRecords());
 	}
-
-
+	
+	
+	@RequiresPermissions(value = {"/chargeDistribution"}, logical = Logical.OR)
 	@GetMapping("/map/getPercent")
 	@ApiOperation(value = "获取该充电站的利用率", tags = {"管理后台-数据分析-电站分析"})
 	public R getPercent(Integer siteId){
@@ -326,10 +330,10 @@
 
 		return R.ok(divide);
 	}
-
-
-
-
+	
+	
+	
+	@RequiresPermissions(value = {"/chargeDistribution"}, logical = Logical.OR)
 	@PostMapping("/map/getPercentByprovinceCode")
 	@ApiOperation(value = "获取该充电站的利用率", tags = {"管理后台-数据分析-电站分析"})
 	public R getPercentByCityCode(@RequestBody ChargingPercentProvinceDto chargingPercentProvinceDto){
@@ -392,18 +396,25 @@
 	public AjaxResult<List<Site>> getSiteList1(){
 		return AjaxResult.success(siteService.list(new QueryWrapper<>()));
 	}
-
+	
+	@RequiresPermissions(value = {"/systemUserManagement", "/site", "/faultInformationList", "/repairRecord", "/balanceSheetRecord", "/chargingPileOrder",
+			"/chargeOrderMonitoring", "/chargingPile", "/chargingGun"}, logical = Logical.OR)
 	@GetMapping("/getSiteListGun")
 	@ApiOperation(value = "获取站点列表 不分页", tags = {"管理后台-接口信息使用"})
 	public AjaxResult<List<Site>> getSiteListGun(){
 		return AjaxResult.success(siteService.getSiteListGun());
 	}
-
+	
+	
+	@RequiresPermissions(value = {"/parkingLot/add", "/parkingLot/select", "/parkingLot/update"}, logical = Logical.OR)
 	@GetMapping("/getSiteListParkLot")
 	@ApiOperation(value = "获取站点列表不分页", tags = {"管理后台-停车场绑定"})
 	public AjaxResult<List<Site>> getSiteListParkLot(){
 		return AjaxResult.success(siteService.getSiteListParkLot());
 	}
+	
+	
+	
 	@ApiOperation(value = "获取站点分页列表", tags = {"小程序-站点管理-首页"})
 	@PostMapping("/pageList")
 	public AjaxResult<PageInfo<SiteVO>> pageList(@Validated @RequestBody SiteQuery query, HttpServletRequest request){
@@ -411,10 +422,13 @@
 		return AjaxResult.success(siteService.pageList(query,isLogin));
 	}
 
+	
+	
 	@ApiOperation(value = "获取站点列表", tags = {"小程序-站点管理-地图导航站点"})
 	@PostMapping("/getMapSiteList")
-	public AjaxResult<List<SiteVO>> getMapSiteList(@Validated @RequestBody SiteQuery query){
-		return AjaxResult.success(siteService.getMapSiteList(query));
+	public AjaxResult<List<SiteVO>> getMapSiteList(@Validated @RequestBody SiteQuery query, HttpServletRequest request){
+		int isLogin = StringUtils.isEmpty(SecurityUtils.getToken(request)) ? 0 : 1;
+		return AjaxResult.success(siteService.getMapSiteList(query,isLogin));
 	}
 	@ApiOperation(value = "获取站点详细信息", tags = {"小程序-站点管理-站点详情"})
 	@PostMapping("/getDetailById")
@@ -449,8 +463,10 @@
 	public AjaxResult<Map<String,ChargingGunCountVO>> getChargingGunCount(@RequestParam(value = "siteId")@ApiParam(value = "站点id", required = true)Integer siteId){
 		return AjaxResult.success(chargingGunService.getChargingGunCount(siteId));
 	}
-
 	
+	
+	
+	@RequiresPermissions(value = {"/site/add"}, logical = Logical.OR)
 	@ResponseBody
 	@PostMapping("/addSite")
 	@ApiOperation(value = "添加站点", tags = {"管理后台-站点管理"})
@@ -461,6 +477,7 @@
 	
 	
 	
+	@RequiresPermissions(value = {"/site/update"}, logical = Logical.OR)
 	@ResponseBody
 	@PostMapping("/getSiteInfo/{id}")
 	@ApiOperation(value = "获取站点详情", tags = {"管理后台-站点管理"})
@@ -471,6 +488,7 @@
 	
 	
 	
+	@RequiresPermissions(value = {"/site/update"}, logical = Logical.OR)
 	@ResponseBody
 	@PostMapping("/editSite")
 	@ApiOperation(value = "编辑站点", tags = {"管理后台-站点管理"})
@@ -479,6 +497,9 @@
 		return siteService.editSite(site);
 	}
 	
+	
+	
+	@RequiresPermissions(value = {"/site/del", "/site/batch_del"}, logical = Logical.OR)
 	@ResponseBody
 	@DeleteMapping("/delSite")
 	@ApiOperation(value = "删除站点", tags = {"管理后台-站点管理"})
@@ -491,6 +512,8 @@
 	}
 	
 	
+	
+	@RequiresPermissions(value = {"/site/accounting_strategy"}, logical = Logical.OR)
 	@ResponseBody
 	@PostMapping("/setAccountingStrategy/{id}")
 	@ApiOperation(value = "设置站点计费策略", tags = {"管理后台-站点管理"})
@@ -505,6 +528,8 @@
 	}
 	
 	
+	@RequiresPermissions(value = {"/roleManagement/add", "/roleManagement/info", "/roleManagement/update", "/systemNotification", "/chargingPeriodStatistics",
+			"/parkingPaymentOrder", "/generateStatement", "/chargingPile/add", "/chargingPile/select", "/chargingPile/update"}, logical = Logical.OR)
 	@ResponseBody
 	@GetMapping("/getAllSiteList")
 	@ApiOperation(value = "获取所有站点数据", tags = {"管理后台-充电桩信息", "管理后台-系统用户管理", "管理后台-角色管理", "管理后台-系统通知", "管理后台-充电评价"})

--
Gitblit v1.7.1