From 3eeb6f5962f114f9706d4150a121bd92f88d9126 Mon Sep 17 00:00:00 2001
From: 无关风月 <443237572@qq.com>
Date: 星期五, 13 十二月 2024 09:25:02 +0800
Subject: [PATCH] 合作商权限

---
 ruoyi-service/ruoyi-order/src/main/java/com/ruoyi/order/controller/TChargingOrderController.java |   84 ++++++++++++++++++++++++++++++++----------
 1 files changed, 64 insertions(+), 20 deletions(-)

diff --git a/ruoyi-service/ruoyi-order/src/main/java/com/ruoyi/order/controller/TChargingOrderController.java b/ruoyi-service/ruoyi-order/src/main/java/com/ruoyi/order/controller/TChargingOrderController.java
index b048a6b..f9031b0 100644
--- a/ruoyi-service/ruoyi-order/src/main/java/com/ruoyi/order/controller/TChargingOrderController.java
+++ b/ruoyi-service/ruoyi-order/src/main/java/com/ruoyi/order/controller/TChargingOrderController.java
@@ -12,10 +12,7 @@
 import com.ruoyi.account.api.feignClient.AppUserClient;
 import com.ruoyi.account.api.model.TAppUserCar;
 import com.ruoyi.chargingPile.api.dto.GetSiteListDTO;
-import com.ruoyi.chargingPile.api.feignClient.ChargingGunClient;
-import com.ruoyi.chargingPile.api.feignClient.ChargingPileClient;
-import com.ruoyi.chargingPile.api.feignClient.ParkingLotClient;
-import com.ruoyi.chargingPile.api.feignClient.SiteClient;
+import com.ruoyi.chargingPile.api.feignClient.*;
 import com.ruoyi.chargingPile.api.model.Site;
 import com.ruoyi.chargingPile.api.model.TChargingGun;
 import com.ruoyi.chargingPile.api.model.TChargingPile;
@@ -57,6 +54,8 @@
 import com.ruoyi.payment.api.feignClient.AliPaymentClient;
 import com.ruoyi.payment.api.feignClient.WxPaymentClient;
 import com.ruoyi.payment.api.vo.AliQueryOrder;
+import com.ruoyi.system.api.domain.SysUser;
+import com.ruoyi.system.api.feignClient.SysUserClient;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiImplicitParam;
 import io.swagger.annotations.ApiImplicitParams;
@@ -332,9 +331,28 @@
     @ApiOperation(value = "列表", tags = {"管理后台-支付订单-退款订单"})
     public R<PageInfo<TChargingOrderRefund>> refundList(@RequestBody ChargingRefundDto chargingRefundDto) {
         R<PageInfo<TChargingOrderRefund>> refundList = chargingOrderService.getRefundList(chargingRefundDto);
-        for (TChargingOrderRefund record : refundList.getData().getRecords()) {
-            record.setUid(record.getId().toString());
+        Long userId = tokenService.getLoginUser().getUserid();
+        //如果没传siteId,获取当前登陆人所有的siteIds
+        List<Integer> siteIds = new ArrayList<>();
+            if (userId != null){
+                List<GetSiteListDTO> data = siteClient.getSiteListByUserId(userId).getData();
+                for (GetSiteListDTO datum : data) {
+                    siteIds.add(datum.getId());
+                }
+            }
+
+        if (siteIds.isEmpty()){
+            siteIds.add(-1);
         }
+        List<TChargingOrderRefund> tChargingOrderRefunds = new ArrayList<>();
+        for (TChargingOrderRefund record : refundList.getData().getRecords()) {
+            TChargingOrder byId = chargingOrderService.getById(record.getChargingOrderId());
+            record.setUid(record.getId().toString());
+            if (siteIds.contains(byId.getSiteId())){
+                tChargingOrderRefunds.add(record);
+            }
+        }
+        refundList.getData().setRecords(tChargingOrderRefunds);
         return refundList;
 
     }
@@ -659,7 +677,7 @@
     @GetMapping(value = "/six/charge")
     @ApiOperation(value = "电站收入分析", tags = {"后台-数据分析-平台收入分析"})
     public R<List<SixChargingDto>> charge(Integer siteId) {
-        Long userId = SecurityUtils.getUserId();
+        Long userId = tokenService.getLoginUser().getUserid();
         //如果没传siteId,获取当前登陆人所有的siteIds
         List<Integer> siteIds = new ArrayList<>();
         if (siteId==null){
@@ -669,6 +687,9 @@
             }
         }else {
             siteIds.add(siteId);
+        }
+        if (siteIds.isEmpty()){
+            siteIds.add(-1);
         }
         LocalDate sixBefore = PreviousSixMonths.get();
         //通过siteIds进行sql查询统计
@@ -709,7 +730,7 @@
     @GetMapping(value = "/six/circle")
     @ApiOperation(value = "电站收入占比", tags = {"后台-数据分析-平台收入分析"})
     public R<List<SixCircleDto>> circle() {
-        Long userId = SecurityUtils.getUserId();
+        Long userId = tokenService.getLoginUser().getUserid();
         //获取当前登录的siteIds
         List<Integer> siteIds = new ArrayList<>();
             List<GetSiteListDTO> data = siteClient.getSiteListByUserId(userId).getData();
@@ -718,7 +739,9 @@
             }
         //进行统计groupBySiteId
         LocalDate sixBefore = PreviousSixMonths.get();
-
+            if (siteIds.isEmpty()){
+                siteIds.add(-1);
+            }
         List<SixCircleDto> sixCircleDtos = chargingOrderService.circle(siteIds,sixBefore);
         for (SixCircleDto sixCircleDto : sixCircleDtos) {
             Site site = siteClient.getSiteByIds(Arrays.asList(sixCircleDto.getSiteId())).getData().get(0);
@@ -797,7 +820,18 @@
     public R<Map<String,Object>> total() {
         //count近6个月的数据
         LocalDate sixBefore = PreviousSixMonths.get();
-        Map<String,Object>  map = chargingOrderService.countAll(sixBefore);
+        Long userId = tokenService.getLoginUser().getUserid();
+        //获取当前登录的siteIds
+        List<Integer> siteIds = new ArrayList<>();
+        List<GetSiteListDTO> data9 = siteClient.getSiteListByUserId(userId).getData();
+        for (GetSiteListDTO datum : data9) {
+            siteIds.add(datum.getId());
+        }
+        //进行统计groupBySiteId
+        if (siteIds.isEmpty()){
+            siteIds.add(-1);
+        }
+        Map<String,Object>  map = chargingOrderService.countAll(sixBefore,siteIds);
         BigDecimal data = parkingLotClient.getRecordAmount(sixBefore).getData();
         if (map ==null){
             map = new HashMap<String,Object>();
@@ -812,8 +846,11 @@
 
         return R.ok(map);
     }
-    
-    
+
+    @Resource
+    private PartnerClient partnerClient;
+    @Resource
+    private SysUserClient sysUserClient;
     @RequiresPermissions(value = {"/chargeOrderMonitoring"}, logical = Logical.OR)
     @ResponseBody
     @PostMapping(value = "/watch/chargingOrder")
@@ -836,7 +873,14 @@
 
         List<ChargingOrderAndUploadRealTimeMonitoringDataDto> dtos = new ArrayList<>();
         Map<String,ChargingOrderVO> map  = new HashMap<>();
+        SysUser sysUser = sysUserClient.getSysUser(tokenService.getLoginUser().getUserid()).getData();
+        Integer roleType = sysUser.getRoleType();
         for (UploadRealTimeMonitoringData uploadRealTimeMonitoringData : data1.getRecords()) {
+            if (roleType==2){
+                List<Boolean> data = partnerClient.watchChargingOrder(sysUser.getObjectId(), uploadRealTimeMonitoringData.getSiteId()).getData();
+                uploadRealTimeMonitoringData.setAuthInfo(data.get(0));
+                uploadRealTimeMonitoringData.setAuthDelete(data.get(1));
+            }
             ChargingOrderAndUploadRealTimeMonitoringDataDto dataDto = new ChargingOrderAndUploadRealTimeMonitoringDataDto();
             ChargingOrderQuery dto = new ChargingOrderQuery();
             dto.setCode(uploadRealTimeMonitoringData.getTransaction_serial_number());
@@ -915,7 +959,7 @@
     public R<TCharingOrderMapVO> watchChargingOrder(@RequestBody ChargingStatisticsQueryDto statisticsQueryDto){
         List<Integer> siteIds =new ArrayList<>();
         if (statisticsQueryDto.getSiteId()==null) {
-            Long userId = SecurityUtils.getUserId();
+            Long userId = tokenService.getLoginUser().getUserid();
             //获取当前登录的siteIds
             List<GetSiteListDTO> data = siteClient.getSiteListByUserId(userId).getData();
             for (GetSiteListDTO datum : data) {
@@ -1147,7 +1191,7 @@
     public R<TCharingOrderPowerMapVO> power(@RequestBody ChargingStatisticsQueryDto statisticsQueryDto) {
         List<Integer> siteIds = new ArrayList<>();
         if (statisticsQueryDto.getSiteId() == null) {
-            Long userId = SecurityUtils.getUserId();
+            Long userId = tokenService.getLoginUser().getUserid();
             //获取当前登录的siteIds
             List<GetSiteListDTO> data = siteClient.getSiteListByUserId(userId).getData();
             for (GetSiteListDTO datum : data) {
@@ -1176,7 +1220,7 @@
     @PostMapping(value = "/charging/users")
     @ApiOperation(value = "除电站流量外", tags = {"管理后台-数据分析-充电用户分析"})
     public R<TCharingUserMapVO> users(@RequestBody ChargingStatisticsQueryDto statisticsQueryDto){
-        Long userId = tokenService.getLoginUserApplet().getUserId();
+        Long userId = tokenService.getLoginUser().getUserid();
 
         List<Integer> siteIds = new ArrayList<>();
         if (statisticsQueryDto.getSiteId()==null){
@@ -1352,7 +1396,7 @@
     public R<TCharingUserEvaluateVO> sites(@RequestBody ChargingStatisticsQueryDto statisticsQueryDto){
         List<Integer> siteIds =new ArrayList<>();
         if (statisticsQueryDto.getSiteId()==null) {
-            Long userId = SecurityUtils.getUserId();
+            Long userId = tokenService.getLoginUser().getUserid();
             //获取当前登录的siteIds
             List<GetSiteListDTO> data = siteClient.getSiteListByUserId(userId).getData();
             for (GetSiteListDTO datum : data) {
@@ -1398,7 +1442,7 @@
     public R<TCharingUserEquimentVO> equipment(@RequestBody ChargingStatisticsQueryDto statisticsQueryDto){
         List<Integer> siteIds =new ArrayList<>();
         if (statisticsQueryDto.getSiteId()==null||statisticsQueryDto.getSiteId()==0) {
-            Long userId = SecurityUtils.getUserId();
+            Long userId = tokenService.getLoginUser().getUserid();
             //获取当前登录的siteIds
             List<GetSiteListDTO> data = siteClient.getSiteListByUserId(userId).getData();
             for (GetSiteListDTO datum : data) {
@@ -1536,7 +1580,7 @@
     public R<TCharingWorkVO> workCharge(@RequestBody ChargingStatisticsQueryDto statisticsQueryDto) {
         List<Integer> siteIds = new ArrayList<>();
         if (statisticsQueryDto.getSiteId() == null) {
-            Long userId = SecurityUtils.getUserId();
+            Long userId = tokenService.getLoginUser().getUserid();
             //获取当前登录的siteIds
             List<GetSiteListDTO> data = siteClient.getSiteListByUserId(userId).getData();
             for (GetSiteListDTO datum : data) {
@@ -1582,7 +1626,7 @@
     public R workCharge(@RequestBody ChargingDetailQueryDto statisticsQueryDto) {
         List<Integer> siteIds = new ArrayList<>();
         if (statisticsQueryDto.getSiteId() == null) {
-            Long userId = SecurityUtils.getUserId();
+            Long userId = tokenService.getLoginUser().getUserid();
             //获取当前登录的siteIds
             List<GetSiteListDTO> data = siteClient.getSiteListByUserId(userId).getData();
             for (GetSiteListDTO datum : data) {
@@ -1708,7 +1752,7 @@
     public R workUse(@RequestBody ChargingDetailQueryDto statisticsQueryDto) {
         List<Integer> siteIds = new ArrayList<>();
         if (statisticsQueryDto.getSiteId() == null) {
-            Long userId = SecurityUtils.getUserId();
+            Long userId = tokenService.getLoginUser().getUserid();
             //获取当前登录的siteIds
             List<GetSiteListDTO> data = siteClient.getSiteListByUserId(userId).getData();
             for (GetSiteListDTO datum : data) {

--
Gitblit v1.7.1