司机管理新增，修改

xuhy

2023-02-22 c2a1f5f286b8a7372811b61acf2f31303b68db61

 management/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TDriverController.java

@@ -2,7 +2,25 @@

import com.baomidou.mybatisplus.mapper.EntityWrapper;
import com.stylefeng.guns.core.base.controller.BaseController;
import com.stylefeng.guns.core.shiro.ShiroKit;
import com.stylefeng.guns.core.shiro.ShiroUser;
import com.stylefeng.guns.modular.system.controller.resp.TDriverResp;
import com.stylefeng.guns.modular.system.controller.util.ExcelUtil;
import com.stylefeng.guns.modular.system.controller.util.UUIDUtil;
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 io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiModelProperty;
import io.swagger.annotations.ApiOperation;
import org.apache.logging.log4j.core.util.UuidUtil;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.shiro.subject.Subject;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Controller;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.ui.Model;
@@ -10,8 +28,16 @@
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.model.TDriver;
import com.stylefeng.guns.modular.system.service.ITDriverService;

import javax.servlet.http.HttpServletResponse;
import java.io.OutputStream;
import java.math.BigDecimal;
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;
import java.util.Objects;

/**
 * 控制器
@@ -27,6 +53,15 @@

    @Autowired
    private ITDriverService tDriverService;

    @Autowired
    private ITRegionService tRegionService;

    @Autowired
    private ITAgentService tAgentService;

    @Autowired
    private ITBranchOfficeService tBranchOfficeService;

    /**
     * 跳转到首页
@@ -47,12 +82,109 @@
    /**
     * 跳转到修改
     */
    @RequestMapping("/tDriver_update/{tDriverId}")
    public String tDriverUpdate(@PathVariable Integer tDriverId, Model model) {
    @RequestMapping("/tDriver_update")
    public String tDriverUpdate( Integer tDriverId, Model model) {
        TDriver tDriver = tDriverService.selectById(tDriverId);
        model.addAttribute("item",tDriver);
        TDriverResp tDriverResp = new TDriverResp();
        BeanUtils.copyProperties(tDriver,tDriverResp);

        // 查询邀请人
        TDriver tDriver1 = tDriverService.selectById(tDriver.getInviterId());
        if(Objects.nonNull(tDriver1)){
            tDriverResp.setInviterName(tDriver1.getName());
            tDriverResp.setInviterPhone(tDriver1.getPhone());
        }

        // 查询区域
        TRegion district = tRegionService.selectOne(new EntityWrapper<TRegion>().eq("code", tDriver.getAreaCode())
                .last("LIMIT 1"));
        TRegion city = tRegionService.selectOne(new EntityWrapper<TRegion>().eq("code", tDriver.getCityCode())
                .last("LIMIT 1"));

        tDriverResp.setArea(tDriver.getProvinceName()+"/"+tDriver.getCityName()+"/"+tDriver.getAreaName());
        if(Objects.nonNull(district) && Objects.nonNull(city)){
            tDriverResp.setAreaId(city.getParentId()+"/"+city.getId()+"/"+district.getId());
        }

        model.addAttribute("item",tDriverResp);
        LogObjectHolder.me().set(tDriver);
        return PREFIX + "tDriver_edit.html";
    }

    /**
     * 跳转冻结页面
     * @return
     */
    @RequestMapping("/tDriver_start_and_stop")
    public String tAppUserStartAndStop( Integer id,
                                        Model model) {
        TDriver tDriver = tDriverService.selectById(id);

        model.addAttribute("id",id);
        model.addAttribute("status",tDriver.getStatus());
        return PREFIX + "tDriverStartAndStop.html";
    }

    /**
     * 跳转充值页面
     * @return
     */
    @RequestMapping("/rechargeBalancePage")
    public String rechargeBalancePage( Integer id,
                                        Model model) {
        model.addAttribute("id",id);
        return PREFIX + "tDriverRechargeBalancePage.html";
    }

    /**
     * 跳转异常页面
     * @return
     */
    @RequestMapping("/tDriver_exception")
    public String tDriverException(Model model) {
        model.addAttribute("isException",2);
        return PREFIX + "tDriverException.html";
    }
    /**
     * 跳转异常启用冻结页面
     * @return
     */
    @RequestMapping("/tDriverException_start_and_stop")
    public String tDriverExceptionStartAndStop(Integer id,Model model) {
        TDriver tDriver = tDriverService.selectById(id);
        model.addAttribute("id",id);
        model.addAttribute("status",tDriver.getStatus());
        return PREFIX + "tDriverExceptionStartAndStop.html";
    }

    /**
     * 跳转审核页面
     * @return
     */
    @RequestMapping("/auditPage")
    public String auditPage( Integer id,
                                       Model model) {
        tDriverService.auditPage(id,model);
        return PREFIX + "tDriverAudit.html";
    }

    /**
     * 跳转详情页面
     */
    @RequestMapping("/driverDetail")
    public String driverDetail(Integer tDriverId, Model model) {
        tDriverService.detail(tDriverId,model);
        return PREFIX + "tDriverDetail.html";
    }

    /**
     * 跳转区域页面
     */
    @RequestMapping("/areaPage")
    public String areaDetail(Model model) {
        List<TRegion> tRegions = tRegionService.selectList(new EntityWrapper<TRegion>().eq("parent_id", 0));
        model.addAttribute("province",tRegions);
        return PREFIX + "tDriverArea.html";
    }

    /**
@@ -62,7 +194,20 @@
    @ResponseBody
    public Object list(String createTime,String phone,Integer status) {
        EntityWrapper<TDriver> wrapper = tDriverService.getPageList(createTime,phone,status);
        return tDriverService.selectList(wrapper);
        wrapper.ne("isException",2);
        List<TDriver> tDrivers = tDriverService.selectList(wrapper);
        return tDriverService.getTDriverResp(tDrivers);
    }
    /**
     * 获取异常列表
     */
    @RequestMapping(value = "/exceptionList")
    @ResponseBody
    public Object exceptionList(String createTime,String phone,Integer status) {
        EntityWrapper<TDriver> wrapper = tDriverService.getPageList(createTime,phone,status);
        wrapper.eq("isException",2);
        List<TDriver> tDrivers = tDriverService.selectList(wrapper);
        return tDriverService.getTDriverResp(tDrivers);
    }

    /**
@@ -75,11 +220,79 @@
    }

    /**
     * 审核
     */
    @RequestMapping(value = "/auditSubmit")
    @ResponseBody
    public Object auditSubmit(Integer id,Integer approvalStatus,String approvalNotes) {
        TDriver tDriver = tDriverService.selectById(id);
        tDriver.setApprovalStatus(approvalStatus);
        tDriver.setApprovalNotes(approvalNotes);
        // 审核用户id
        Subject subject = ShiroKit.getSubject();
        ShiroUser shiroUser = (ShiroUser)subject.getPrincipal();
        tDriver.setApprovalUserId(shiroUser.getId());
        tDriver.setApprovalTime(new Date());
        tDriverService.updateById(tDriver);
        return SUCCESS_TIP;
    }

    /**
     * 新增
     */
    @RequestMapping(value = "/add")
    @ResponseBody
    public Object add(TDriver tDriver) {

        int count = tDriverService.selectCount(new EntityWrapper<TDriver>().eq("phone", tDriver.getPhone()));
        if(count>0){
            return "该司机已存在!";
        }

        // 对省市区做处理
        String[] split = tDriver.getAreaId().split("/");
        // 查询省市
        // 黑龙江省/大兴安岭地区
        // 702/852
        TRegion province = tRegionService.selectById(split[0]);
        tDriver.setProvinceName(province.getName());
        tDriver.setProvinceCode(province.getCode());

        TRegion city = tRegionService.selectById(split[1]);
        tDriver.setCityName(city.getName());
        tDriver.setCityCode(city.getCode());

        TRegion area = tRegionService.selectById(split[2]);
        tDriver.setAreaName(area.getName());
        tDriver.setAreaCode(area.getCode());

        tDriver.setCode(UUIDUtil.getNumberRandom(16));
        tDriver.setBalance(BigDecimal.ZERO);
        tDriver.setBackgroundBalance(BigDecimal.ZERO);

        // 通过省市查询代理商
        TAgent tAgent = tAgentService.selectOne(new EntityWrapper<TAgent>().eq("provinceCode", province.getCode())
                .eq("cityCode", city.getCode())
                .last("LIMIT 1"));
        if(Objects.nonNull(tAgent)){
            tDriver.setAgentId(tAgent.getId());
        }
        // 查询邀约人（司机端）
        TDriver emergencyDriver = tDriverService.selectOne(new EntityWrapper<TDriver>().eq("phone", tDriver.getEmergencyPhone())
                .last("LIMIT 1"));
        if(Objects.nonNull(emergencyDriver)){
            tDriver.setInviterId(emergencyDriver.getId());
            tDriver.setInviterType(2);
        }
        // 通过省市区查询分公司
        TBranchOffice tBranchOffice = tBranchOfficeService.selectOne(new EntityWrapper<TBranchOffice>().eq("provinceCode", province.getCode())
                .eq("cityCode", city.getCode())
                .eq("districtCode", area.getCode())
                .last("LIMIT 1"));
        if(Objects.nonNull(tBranchOffice)){
            tDriver.setBranchOfficeId(tBranchOffice.getId());
        }

        tDriverService.insert(tDriver);
        return SUCCESS_TIP;
    }
@@ -91,6 +304,38 @@
    @ResponseBody
    public Object delete(@RequestParam Integer tDriverId) {
        tDriverService.deleteById(tDriverId);
        return SUCCESS_TIP;
    }

    /**
     * 修改司机状态
     */
    @RequestMapping(value = "/update-status")
    @ResponseBody
    public Object updateStatus(Integer id,Integer status,
                               String remark) {
        TDriver tDriver = tDriverService.selectById(id);
        if(1 == status){
            tDriver.setStatus(2);
        }
        if(2 == status){
            tDriver.setStatus(1);
        }
        tDriver.setRemark(remark);
        tDriverService.updateById(tDriver);
        return SUCCESS_TIP;
    }

    /**
     * 充值余额
     */
    @ApiOperation(value = "充值余额")
    @RequestMapping(value = "/recharge-balance")
    @ResponseBody
    public Object rechargeBalance(Integer id, String money) {
        TDriver tDriver = tDriverService.selectById(id);
        tDriver.setBackgroundBalance(new BigDecimal(money).add(tDriver.getBackgroundBalance()));
        tDriverService.updateById(tDriver);
        return SUCCESS_TIP;
    }

@@ -112,4 +357,151 @@
    public Object detail(@PathVariable("tDriverId") Integer tDriverId) {
        return tDriverService.selectById(tDriverId);
    }

    @ApiOperation(value = "省查询",notes="省查询")
    @RequestMapping(value = "/areaProvince")
    @ResponseBody
    public Object areaProvince(Model model) {
        return tRegionService.selectList(new EntityWrapper<TRegion>().eq("parent_id", 0));
    }

    @ApiOperation(value = "市区查询",notes="市区查询")
    @RequestMapping(value = "/areaCity")
    @ResponseBody
    public Object areaCity(Integer parentId) {
        return tRegionService.selectList(new EntityWrapper<TRegion>().eq("parent_id",parentId));
    }



    @ApiOperation(value = "导出司机列表",notes="导出司机列表")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "Authorization", value = "用户token（Bearer +token）", required = true, dataType = "String", paramType = "header", defaultValue = "Bearer eyJhbGciOiJIUzUxMiJ9....."),
    })
    @RequestMapping(value = "/export")
    @ResponseBody
    public void export(String createTime,String phone,Integer status,HttpServletResponse response) {
        try {
            Date date = new Date();
            DateFormat format = new SimpleDateFormat("yyyyMMdd");
            String time1 = format.format(date);
            String fileName = "DriverInfo"+time1+".xls";
            String[] title = new String[] {"编号","姓名","手机号","性别","驾驶证号码","身份证号码",
                    "身份证","所属代理商","所属分公司","钱包余额","累计订单量","当月订单量","积分","拒单次数","状态","添加时间"};
            EntityWrapper<TDriver> wrapper = tDriverService.getPageList(createTime,phone,status);
            // 是否异常
            List<TDriver> list = tDriverService.selectList(wrapper);

            List<TDriverResp> tDriverResp = tDriverService.getTDriverResp(list);

            String[][] values = new String[tDriverResp.size()][];
            for (int i = 0; i < tDriverResp.size(); i++) {
                TDriverResp d = tDriverResp.get(i);
                values[i] = new String[title.length];
                values[i][0] = d.getCode();
                values[i][1] = d.getName();
                values[i][2] = d.getPhone();
                Integer sex = d.getSex();
                if(Objects.isNull(sex)){
                    values[i][3] = "未知";
                }else if(1 == sex){
                    values[i][3] = "男";
                }else if(2 == sex){
                    values[i][3] = "女";
                }
                values[i][4] = d.getDriverLicenseNumber();
                values[i][5] = d.getIdcard();
                if(StringUtils.hasLength(d.getIdcardFront())){
                    values[i][6] = "已上传";
                }else {
                    values[i][6] = "未上传";
                }
                values[i][7] = d.getAgentName();
                values[i][8] = d.getBranchName();
                values[i][9] = String.valueOf(d.getBalance());
                values[i][10] = String.valueOf(d.getCumulativeOrderCount());
                values[i][11] = String.valueOf(d.getMonthOrderCount());
                values[i][12] = String.valueOf(d.getIntegral());
                values[i][13] = String.valueOf(d.getRefusalCount());
                Integer status1 = d.getStatus();
                if(1 == status1){
                    values[i][14] = "正常";
                }else if(2 == status1){
                    values[i][14] = "冻结";
                }else {
                    values[i][14] = "删除";
                }
                values[i][15] = new SimpleDateFormat("yyyy/MM/dd HH:mm:ss").format(d.getCreateTime());
            }
            HSSFWorkbook wb = ExcelUtil.getHSSFWorkbook("Variance"+time1, title, values, null);
            ExcelUtil.setResponseHeader(response, fileName);
            OutputStream os = response.getOutputStream();
            wb.write(os);
            os.flush();
            os.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    @ApiOperation(value = "导出司机异常列表",notes="导出司机异常列表")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "Authorization", value = "用户token（Bearer +token）", required = true, dataType = "String", paramType = "header", defaultValue = "Bearer eyJhbGciOiJIUzUxMiJ9....."),
    })
    @RequestMapping(value = "/export-exception")
    @ResponseBody
    public void exportException(String createTime,String phone,Integer status,HttpServletResponse response) {
        try {
            Date date = new Date();
            DateFormat format = new SimpleDateFormat("yyyyMMdd");
            String time1 = format.format(date);
            String fileName = "DriverExceptionInfo"+time1+".xls";
            String[] title = new String[] {"编号","姓名","手机号","性别","驾驶证号码","身份证号码",
                    "身份证","累计订单量","当月订单量","积分","连续未上线（天）","状态","添加时间"};
            EntityWrapper<TDriver> wrapper = tDriverService.getPageList(createTime,phone,status);
            wrapper.eq("isException",2);
            // 是否异常
            List<TDriver> list = tDriverService.selectList(wrapper);

            List<TDriverResp> tDriverResp = tDriverService.getTDriverResp(list);

            String[][] values = new String[tDriverResp.size()][];
            for (int i = 0; i < tDriverResp.size(); i++) {
                TDriverResp d = tDriverResp.get(i);
                values[i] = new String[title.length];
                values[i][0] = d.getCode();
                values[i][1] = d.getName();
                values[i][2] = d.getPhone();
                Integer sex = d.getSex();
                if(Objects.isNull(sex)){
                    values[i][3] = "未知";
                }else if(1 == sex){
                    values[i][3] = "男";
                }else if(2 == sex){
                    values[i][3] = "女";
                }
                values[i][4] = d.getDriverLicenseNumber();
                values[i][5] = d.getIdcard();
                if(StringUtils.hasLength(d.getIdcardFront())){
                    values[i][6] = "已上传";
                }else {
                    values[i][6] = "未上传";
                }
                values[i][7] = String.valueOf(d.getCumulativeOrderCount());
                values[i][8] = String.valueOf(d.getMonthOrderCount());
                values[i][9] = String.valueOf(d.getIntegral());
                values[i][10] = String.valueOf(Objects.isNull(d.getNotOnlineCount())?0:d.getNotOnlineCount());
                values[i][11] = "异常";
                values[i][12] = new SimpleDateFormat("yyyy/MM/dd HH:mm:ss").format(d.getCreateTime());
            }
            HSSFWorkbook wb = ExcelUtil.getHSSFWorkbook("Variance"+time1, title, values, null);
            ExcelUtil.setResponseHeader(response, fileName);
            OutputStream os = response.getOutputStream();
            wb.write(os);
            os.flush();
            os.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}