From dea785f9fab25a7c68ea890628a39c973c96a935 Mon Sep 17 00:00:00 2001
From: xuhy <3313886187@qq.com>
Date: 星期二, 28 三月 2023 18:23:16 +0800
Subject: [PATCH] 基础信息管理

---
 management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/service/impl/TDriverServiceImpl.java |   59 +++++++++++++++++++++++++++++++++++++++++++++--------------
 1 files changed, 45 insertions(+), 14 deletions(-)

diff --git a/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/service/impl/TDriverServiceImpl.java b/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/service/impl/TDriverServiceImpl.java
index 818891c..a6f5ed0 100644
--- a/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/service/impl/TDriverServiceImpl.java
+++ b/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/service/impl/TDriverServiceImpl.java
@@ -11,10 +11,14 @@
 import com.stylefeng.guns.modular.system.controller.util.UUIDUtil;
 import com.stylefeng.guns.modular.system.dao.*;
 import com.stylefeng.guns.modular.system.enums.OrderStateEnum;
+import com.stylefeng.guns.modular.system.enums.PayStatusEnum;
+import com.stylefeng.guns.modular.system.enums.StatusEnum;
+import com.stylefeng.guns.modular.system.enums.UserTypeEnum;
 import com.stylefeng.guns.modular.system.model.*;
 import com.stylefeng.guns.modular.system.service.ITDriverService;
 import com.baomidou.mybatisplus.service.impl.ServiceImpl;
 import com.stylefeng.guns.modular.system.service.ITRegionService;
+import org.apache.poi.hdf.extractor.TC;
 import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
@@ -28,10 +32,7 @@
 import java.time.LocalDate;
 import java.time.Period;
 import java.time.ZoneId;
-import java.util.ArrayList;
-import java.util.Date;
-import java.util.List;
-import java.util.Objects;
+import java.util.*;
 import java.util.concurrent.atomic.AtomicInteger;
 import java.util.stream.Collectors;
 
@@ -65,6 +66,11 @@
     private ITRegionService itRegionService;
     @Autowired
     private HttpUtils httpUtils;
+
+    @Autowired
+    private TRechargeRecordMapper tRechargeRecordMapper;
+    @Autowired
+    private TCashWithdrawalMapper tCashWithdrawalMapper;
 
     @Override
     public EntityWrapper<TDriver> getPageList(String createTime, String phone, Integer status) {
@@ -109,8 +115,10 @@
             TDriverResp tDriverResp = new TDriverResp();
             BeanUtils.copyProperties(tDriver,tDriverResp);
 
-            // 设置余额
-            tDriverResp.setBalance(tDriver.getBackgroundBalance().add(tDriver.getBalance()));
+            if(Objects.nonNull(tDriver.getBackgroundBalance()) && Objects.nonNull(tDriver.getBalance())){
+                // 设置余额
+                tDriverResp.setBalance(tDriver.getBackgroundBalance().add(tDriver.getBalance()));
+            }
 
             // 所属代理商
             TAgent tAgent = tAgentMapper.selectById(tDriver.getAgentId());
@@ -123,11 +131,11 @@
                 tDriverResp.setBranchName(tBranchOffice.getPrincipal());
             }
             //累计订单量
-            List<TOrder> cumulativeOrderCount = tOrders.stream().filter(order -> order.getDriverId().equals(tDriver.getId())
+            List<TOrder> cumulativeOrderCount = tOrders.stream().filter(order -> tDriver.getId().equals(order.getDriverId())
                     && order.getState().equals(OrderStateEnum.WAIT_EVALUATED.getCode()) && order.getState().equals(OrderStateEnum.FINISH.getCode())).collect(Collectors.toList());
             tDriverResp.setCumulativeOrderCount(cumulativeOrderCount.size());
             //当月订单量
-            List<TOrder> monthOrderCount = tOrders.stream().filter(order -> order.getDriverId().equals(tDriver.getId())
+            List<TOrder> monthOrderCount = tOrders.stream().filter(order -> tDriver.getId().equals(order.getDriverId())
                     && order.getState().equals(OrderStateEnum.WAIT_EVALUATED.getCode()) && order.getState().equals(OrderStateEnum.FINISH.getCode())
             && new SimpleDateFormat("yyyyMM").format(order.getCreateTime()).equals(monthDate)).collect(Collectors.toList());
             tDriverResp.setMonthOrderCount(monthOrderCount.size());
@@ -231,19 +239,42 @@
 
         if(!CollectionUtils.isEmpty(tOrders)){
             //累计订单量
-            List<TOrder> cumulativeOrderCount = tOrders.stream().filter(order -> order.getDriverId().equals(tDriver.getId())
+            List<TOrder> cumulativeOrderCount = tOrders.stream().filter(order -> tDriver.getId().equals(order.getDriverId())
                     && order.getState().equals(OrderStateEnum.WAIT_EVALUATED.getCode()) && order.getState().equals(OrderStateEnum.FINISH.getCode())).collect(Collectors.toList());
             model.addAttribute("cumulativeOrderCount",cumulativeOrderCount.size());
             //当月订单量
-            List<TOrder> monthOrderCount = tOrders.stream().filter(order -> order.getDriverId().equals(tDriver.getId())
+            List<TOrder> monthOrderCount = tOrders.stream().filter(order -> tDriver.getId().equals(order.getDriverId())
                     && order.getState().equals(OrderStateEnum.WAIT_EVALUATED.getCode()) && order.getState().equals(OrderStateEnum.FINISH.getCode())
                     && new SimpleDateFormat("yyyyMM").format(order.getCreateTime()).equals(monthDate)).collect(Collectors.toList());
             model.addAttribute("monthOrderCount",monthOrderCount.size());
             // 拒单次数
             List<TOrderRefusal> driverRefusalList = tOrderRefusalMapper.selectList(new EntityWrapper<TOrderRefusal>().eq("driver_id", tDriver.getId()));
             model.addAttribute("refusalCount",driverRefusalList.size());
-            // todo 累计收入
-            model.addAttribute("cumulativeIncome",0);
+            // 累计收入 司机提现记录加上余额减去充值金额,为累计收入
+            List<TCashWithdrawal> tCashWithdrawals = tCashWithdrawalMapper.selectList(new EntityWrapper<TCashWithdrawal>()
+                    .eq("type", UserTypeEnum.DRIVER.getCode())
+                    .eq("userDriverId", tDriver.getId())
+                    .eq("state", 2));
+            BigDecimal balance = tDriver.getBalance();
+            if(!CollectionUtils.isEmpty(tCashWithdrawals)){
+                Optional<BigDecimal> reduce = tCashWithdrawals.stream().map(TCashWithdrawal::getAmount).reduce(BigDecimal::add);
+                if(reduce.isPresent()){
+                    balance = balance.add(reduce.get());
+                }
+            }
+            List<TRechargeRecord> tRechargeRecords = tRechargeRecordMapper.selectList(new EntityWrapper<TRechargeRecord>()
+                    .eq("type", UserTypeEnum.DRIVER.getCode())
+                    .eq("userId", tDriver.getId())
+                    .eq("payType", 1)
+                    .eq("payStatus", PayStatusEnum.FINISH.getCode()));
+            if(!CollectionUtils.isEmpty(tRechargeRecords)){
+                Optional<BigDecimal> reduce = tRechargeRecords.stream().map(TRechargeRecord::getAmount).reduce(BigDecimal::add);
+                if(reduce.isPresent()){
+                    balance = balance.subtract(reduce.get());
+                }
+            }
+            // 减去充值金额
+            model.addAttribute("cumulativeIncome",balance);
         }else {
             model.addAttribute("cumulativeOrderCount",0);
             model.addAttribute("monthOrderCount",0);
@@ -289,7 +320,7 @@
         // 通过省市查询代理商
         List<TAgent> tAgent = tAgentMapper.selectList(new EntityWrapper<TAgent>().eq("provinceCode", province.getCode())
                 .eq("cityCode", city.getCode())
-                .eq("status", 1)
+                .eq("status", StatusEnum.NORMAL.getCode())
                 .last("LIMIT 1"));
         if(!CollectionUtils.isEmpty(tAgent)){
             tDriver.setAgentId(tAgent.get(0).getId());
@@ -310,7 +341,7 @@
         // 通过省市区查询分公司
         List<TBranchOffice> tBranchOffice = tBranchOfficeMapper.selectList(new EntityWrapper<TBranchOffice>().eq("provinceCode", province.getCode())
                 .eq("cityCode", city.getCode())
-                .eq("status", 1)
+                .eq("status", StatusEnum.NORMAL.getCode())
                 .eq("districtCode", area.getCode())
                 .last("LIMIT 1"));
         if(!CollectionUtils.isEmpty(tBranchOffice)){

--
Gitblit v1.7.1