From 1b6c4cd60f3882c400d30af367bd4c5960fbd4a1 Mon Sep 17 00:00:00 2001
From: xuhy <3313886187@qq.com>
Date: 星期一, 10 四月 2023 15:02:24 +0800
Subject: [PATCH] 优化管理

---
 management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TDriverController.java |   95 ++++++++++++++++++++++++++++++++++++++---------
 1 files changed, 77 insertions(+), 18 deletions(-)

diff --git a/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TDriverController.java b/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TDriverController.java
index 54fc2f8..f139140 100644
--- a/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TDriverController.java
+++ b/management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TDriverController.java
@@ -1,5 +1,6 @@
 package com.stylefeng.guns.modular.system.controller.general;
 
+import cn.hutool.core.util.CreditCodeUtil;
 import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.mapper.EntityWrapper;
 import com.stylefeng.guns.core.base.controller.BaseController;
@@ -9,10 +10,9 @@
 import com.stylefeng.guns.modular.system.controller.resp.TDriverCommissionResp;
 import com.stylefeng.guns.modular.system.controller.resp.TDriverResp;
 import com.stylefeng.guns.modular.system.controller.util.*;
+import com.stylefeng.guns.modular.system.enums.UserTypeEnum;
 import com.stylefeng.guns.modular.system.model.*;
-import com.stylefeng.guns.modular.system.service.ITAgentService;
-import com.stylefeng.guns.modular.system.service.ITBranchOfficeService;
-import com.stylefeng.guns.modular.system.service.ITRegionService;
+import com.stylefeng.guns.modular.system.service.*;
 import io.swagger.annotations.ApiImplicitParam;
 import io.swagger.annotations.ApiImplicitParams;
 import io.swagger.annotations.ApiModelProperty;
@@ -23,6 +23,7 @@
 import org.springframework.beans.BeanUtils;
 import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.stereotype.Controller;
+import org.springframework.transaction.annotation.Transactional;
 import org.springframework.util.StringUtils;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.ResponseBody;
@@ -31,7 +32,6 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import com.stylefeng.guns.core.log.LogObjectHolder;
 import org.springframework.web.bind.annotation.RequestParam;
-import com.stylefeng.guns.modular.system.service.ITDriverService;
 
 import javax.servlet.http.HttpServletResponse;
 import java.io.File;
@@ -74,6 +74,8 @@
     private HttpUtils httpUtils;
     @Autowired
     private TokenUtils tokenUtils;
+    @Autowired
+    private ITRechargeRecordService tRechargeRecordService;
 
     /**
      * 跳转到首页
@@ -211,11 +213,46 @@
      */
     @RequestMapping("/areaPageUpdate")
     public String areaPageUpdate(String area,String areaId,Model model) {
+
+        String[] split = area.split("/");
+        model.addAttribute("provinceName",split[0]);
+        model.addAttribute("cityName",split[1]);
+        model.addAttribute("districtName",split[2]);
+
+        String[] split1 = areaId.split("/");
+        model.addAttribute("provinceId",split1[0]);
+        model.addAttribute("cityId",split1[1]);
+        model.addAttribute("districtId",split1[2]);
+
+
+        List<TRegion> tRegions = tRegionService.selectList(new EntityWrapper<TRegion>().eq("parent_id", 0));
+        model.addAttribute("provinceList",tRegions);
+        List<Integer> provinceIds = tRegions.stream().map(TRegion::getId).collect(Collectors.toList());
+
+
+        // 查询市
+        List<TRegion> tRegions1 = tRegionService.selectList(new EntityWrapper<TRegion>().eq("parent_id", split1[0]));
+        model.addAttribute("cityList",tRegions1);
+
+        // 查询区
+        List<Integer> cityIds = tRegions1.stream().map(TRegion::getId).collect(Collectors.toList());
+        List<TRegion> tRegions2 = tRegionService.selectList(new EntityWrapper<TRegion>().in("parent_id", cityIds));
+        model.addAttribute("districtList",tRegions2);
+
+        return PREFIX + "tDriverAreaUpdate.html";
+    }
+
+    /**
+     * 跳转区域页面新增
+     */
+    @RequestMapping("/areaPageAdd")
+    public String areaPageAdd(String area,String areaId,Model model) {
+        String[] split1 = areaId.split("/");
         List<TRegion> tRegions = tRegionService.selectList(new EntityWrapper<TRegion>().eq("parent_id", 0));
         model.addAttribute("provinceList",tRegions);
         List<Integer> provinceIds = tRegions.stream().map(TRegion::getId).collect(Collectors.toList());
         // 查询市
-        List<TRegion> tRegions1 = tRegionService.selectList(new EntityWrapper<TRegion>().in("parent_id", provinceIds));
+        List<TRegion> tRegions1 = tRegionService.selectList(new EntityWrapper<TRegion>().eq("parent_id", split1[0]));
         model.addAttribute("cityList",tRegions1);
 
         // 查询区
@@ -227,23 +264,28 @@
             String[] split = area.split("/");
             model.addAttribute("provinceName",split[0]);
             model.addAttribute("cityName",split[1]);
-            model.addAttribute("districtName",split[2]);
+            if(split.length>2){
+                model.addAttribute("districtName",split[2]);
+            }else {
+                model.addAttribute("districtName","");
+            }
 
-            String[] split1 = areaId.split("/");
             model.addAttribute("provinceId",split1[0]);
             model.addAttribute("cityId",split1[1]);
-            model.addAttribute("districtId",split1[2]);
-        }
-        return PREFIX + "tDriverAreaUpdate.html";
-    }
+            if(split1.length>2) {
+                model.addAttribute("districtId", split1[2]);
+            }else {
+                model.addAttribute("districtId", "");
+            }
+        }else {
+            model.addAttribute("provinceName","");
+            model.addAttribute("cityName","split[1]");
+            model.addAttribute("districtName","");
 
-    /**
-     * 跳转区域页面新增
-     */
-    @RequestMapping("/areaPageAdd")
-    public String areaPageAdd(Model model) {
-        List<TRegion> tRegions = tRegionService.selectList(new EntityWrapper<TRegion>().eq("parent_id", 0));
-        model.addAttribute("provinceList",tRegions);
+            model.addAttribute("provinceId","");
+            model.addAttribute("cityId","split1[1]");
+            model.addAttribute("districtId", "");
+        }
         return PREFIX + "tDriverAreaAdd.html";
     }
 
@@ -255,6 +297,7 @@
     public Object list(String createTime,String phone,Integer status) {
         EntityWrapper<TDriver> wrapper = tDriverService.getPageList(createTime,phone,status);
         wrapper.ne("isException",2);
+        wrapper.orderBy("createTime",false);
         List<TDriver> tDrivers = tDriverService.selectList(wrapper);
         return tDriverService.getTDriverResp(tDrivers);
     }
@@ -276,6 +319,7 @@
     public Object exceptionList(String createTime,String phone,Integer status) {
         EntityWrapper<TDriver> wrapper = tDriverService.getPageList(createTime,phone,status);
         wrapper.eq("isException",2);
+        wrapper.orderBy("createTime",false);
         List<TDriver> tDrivers = tDriverService.selectList(wrapper);
         return tDriverService.getTDriverResp(tDrivers);
     }
@@ -294,6 +338,7 @@
         if(StringUtils.hasLength(driverId)){
             wrapper.eq("inviterId",driverId);
         }
+        wrapper.orderBy("createTime",false);
         List<TDriver> list = tDriverService.selectList(wrapper);
         List<TDriverCommissionResp> commissionResp = tDriverService.getTDriverCommissionResp(list);
         for (TDriverCommissionResp tDriverCommissionResp : commissionResp) {
@@ -408,11 +453,25 @@
      */
     @ApiOperation(value = "充值余额")
     @RequestMapping(value = "/recharge-balance")
+    @Transactional(rollbackFor = Exception.class)
     @ResponseBody
     public Object rechargeBalance(Integer id, String money) {
         TDriver tDriver = tDriverService.selectById(id);
         tDriver.setBackgroundBalance(new BigDecimal(money).add(tDriver.getBackgroundBalance()));
         tDriverService.updateById(tDriver);
+
+        // 添加充值记录
+        TRechargeRecord tRechargeRecord = new TRechargeRecord();
+        tRechargeRecord.setType(UserTypeEnum.AGENT.getCode());
+        tRechargeRecord.setUserId(id);
+        tRechargeRecord.setCode(UUIDUtil.getNumberRandom(16));
+        tRechargeRecord.setAmount(new BigDecimal(money));
+        tRechargeRecord.setPayType(2);
+        tRechargeRecord.setPayTime(new Date());
+        tRechargeRecord.setPayStatus(2);
+        tRechargeRecord.setCreateTime(new Date());
+        tRechargeRecord.setAgentId(Objects.requireNonNull(ShiroKit.getUser()).getId());
+        tRechargeRecordService.insert(tRechargeRecord);
         return SUCCESS_TIP;
     }
 

--
Gitblit v1.7.1