From a4c86b6b9d61bec99bff97415ac25e1fd36fe28a Mon Sep 17 00:00:00 2001
From: xuhy <3313886187@qq.com>
Date: 星期日, 09 四月 2023 19:17:56 +0800
Subject: [PATCH] 优化

---
 management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/service/impl/TDriverServiceImpl.java |   83 ++++++++++++++++++++++++++++++++++++-----
 1 files changed, 73 insertions(+), 10 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 ad9e895..66a752d 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) {
@@ -233,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);
@@ -291,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());
@@ -312,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)){
@@ -457,4 +486,38 @@
 //        commissionResp.setAccumulatedCommission(BigDecimal.ZERO);
 
     }
+
+    @Override
+    public void getDataStatistics(Integer agentId, Model model,Map<String, Object> map) {
+        // 司机总数
+        Integer driverTotal = tDriverMapper.selectCount(new EntityWrapper<TDriver>()
+                .eq("agentId", agentId));
+        // 司机最近一月数量
+        LocalDate now = LocalDate.now();
+        LocalDate localDate = now.plusMonths(1);
+        Integer driverMonthTotal = tDriverMapper.selectCount(new EntityWrapper<TDriver>()
+                .eq("agentId", agentId)
+                .between("createTime", localDate, now));
+        model.addAttribute("driverTotal",driverTotal);
+        model.addAttribute("driverMonthTotal",driverMonthTotal);
+        map.put("driverTotal",driverTotal);
+        map.put("driverMonthTotal",driverMonthTotal);
+    }
+
+    @Override
+    public void getDataStatisticsByIds(List<Integer> ids, Model model, Map<String, Object> map) {
+        // 司机总数
+        Integer driverTotal = tDriverMapper.selectCount(new EntityWrapper<TDriver>()
+                .in("agentId", ids));
+        // 司机最近一月数量
+        LocalDate now = LocalDate.now();
+        LocalDate localDate = now.plusMonths(1);
+        Integer driverMonthTotal = tDriverMapper.selectCount(new EntityWrapper<TDriver>()
+                .in("agentId", ids)
+                .between("createTime", localDate, now));
+        model.addAttribute("driverTotal",driverTotal);
+        model.addAttribute("driverMonthTotal",driverMonthTotal);
+        map.put("driverTotal",driverTotal);
+        map.put("driverMonthTotal",driverMonthTotal);
+    }
 }

--
Gitblit v1.7.1