From f4d52e83eb99233c895a0fff0eef287792130752 Mon Sep 17 00:00:00 2001
From: xuhy <3313886187@qq.com>
Date: 星期二, 16 五月 2023 10:45:52 +0800
Subject: [PATCH] BUG修改

---
 management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TAgentController.java |   39 +++++++++++++++++++++++++++++++++------
 1 files changed, 33 insertions(+), 6 deletions(-)

diff --git a/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TAgentController.java b/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TAgentController.java
index d6a3819..6c988c0 100644
--- a/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TAgentController.java
+++ b/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TAgentController.java
@@ -5,6 +5,7 @@
 import com.stylefeng.guns.core.base.tips.SuccessTip;
 import com.stylefeng.guns.core.exception.GunsException;
 import com.stylefeng.guns.core.exception.ServiceExceptionEnum;
+import com.stylefeng.guns.core.shiro.ShiroKit;
 import com.stylefeng.guns.core.util.DateUtil;
 import com.stylefeng.guns.modular.system.controller.resp.TAgentResp;
 import com.stylefeng.guns.modular.system.controller.util.ExcelUtil;
@@ -13,6 +14,7 @@
 import com.stylefeng.guns.modular.system.enums.UserTypeEnum;
 import com.stylefeng.guns.modular.system.model.*;
 import com.stylefeng.guns.modular.system.service.*;
+import com.stylefeng.guns.modular.system.util.RedisUtil;
 import io.swagger.annotations.ApiImplicitParam;
 import io.swagger.annotations.ApiImplicitParams;
 import io.swagger.annotations.ApiOperation;
@@ -58,13 +60,18 @@
 
     @Autowired
     private ITDriverService tDriverService;
+    @Autowired
+    private ITDriverWorkService tDriverWorkService;
+    @Autowired
+    private RedisUtil redisUtil;
 
 
     /**
      * 跳转到首页
      */
     @RequestMapping("")
-    public String index() {
+    public String index(Model model) {
+        model.addAttribute("userType", Objects.requireNonNull(ShiroKit.getUser()).getRoleType());
         return PREFIX + "tAgent.html";
     }
 
@@ -130,8 +137,8 @@
      */
     @RequestMapping(value = "/list")
     @ResponseBody
-    public Object list(String principal,String principalPhone,String createTime) {
-        EntityWrapper<TAgent> wrapper = tAgentService.getAgentWrapper(principal,principalPhone,createTime);
+    public Object list(String principal,String principalPhone,String createTime,Integer status) {
+        EntityWrapper<TAgent> wrapper = tAgentService.getAgentWrapper(principal,principalPhone,createTime,status);
         List<TAgent> tAgents = tAgentService.selectList(wrapper);
         // 代理商列表数据封装(导出共用)
         return tAgentService.getAgentResp(tAgents);
@@ -158,7 +165,11 @@
         int count = tAgentService.selectCount(new EntityWrapper<TAgent>().eq("provinceName", split1[0]).eq("cityName",split1[1]));
         if(count>0){
             return new SuccessTip(500,"该代理商已存在!");
+
         }
+
+        tAgent.setPrincipal(tAgent.getPrincipal().replace(" ",""));
+
 
         String[] split = tAgent.getAreaId().split("/");
         // 查询省市
@@ -204,12 +215,27 @@
         for (TDriver tDriver : list) {
             if(1 == status){
                 tDriver.setStatus(2);
+                String value = redisUtil.getValue("DRIVER_" + tDriver.getPhone());
+                redisUtil.remove(value);
+                redisUtil.remove("DRIVER_" + tDriver.getPhone());
+                TDriverWork tDriverWork = tDriverWorkService.selectOne(new EntityWrapper<TDriverWork>()
+                        .eq("driverId", tDriver.getId())
+                        .eq("status", 1)
+                        .orderBy("workTime", false)
+                        .last("LIMIT 1"));
+                if(Objects.nonNull(tDriverWork)){
+                    tDriverWork.setStatus(2);
+                    tDriverWork.setOffWorkTime(new Date());
+                    tDriverWorkService.updateById(tDriverWork);
+                }
             }
             if(2 == status){
                 tDriver.setStatus(1);
             }
         }
-        tDriverService.updateBatchById(list);
+        if(!CollectionUtils.isEmpty(list)){
+            tDriverService.updateBatchById(list);
+        }
         return SUCCESS_TIP;
     }
 
@@ -219,6 +245,7 @@
     @RequestMapping(value = "/update")
     @ResponseBody
     public Object update(TAgent tAgent) {
+        tAgent.setPrincipal(tAgent.getPrincipal().replace(" ",""));
         tAgentService.updateById(tAgent);
         return SUCCESS_TIP;
     }
@@ -258,7 +285,7 @@
     @ApiOperation(value = "导出代理商列表",notes="导出代理商列表")
     @RequestMapping(value = "/export")
     @ResponseBody
-    public void export(String principal,String principalPhone,String createTime, HttpServletResponse response) {
+    public void export(String principal,String principalPhone,Integer status,String createTime, HttpServletResponse response) {
         try {
             Date date = new Date();
             DateFormat format = new SimpleDateFormat("yyyyMMdd");
@@ -266,7 +293,7 @@
             String fileName = "Agent"+time1+".xls";
             String[] title = new String[] {"时间","姓名","联系电话","代理区域","客服电话","订单数量",
                     "有效订单","已发放优惠券","已使用优惠券","累计优惠券金额","司机充值","司机数","状态"};
-            EntityWrapper<TAgent> wrapper = tAgentService.getAgentWrapper(principal,principalPhone,createTime);
+            EntityWrapper<TAgent> wrapper = tAgentService.getAgentWrapper(principal,principalPhone,createTime,status);
             // 是否异常
             List<TAgent> list = tAgentService.selectList(wrapper);
 

--
Gitblit v1.7.1