package com.stylefeng.guns.modular.system.controller.general; import com.alibaba.fastjson.JSONObject; import com.baomidou.mybatisplus.mapper.EntityWrapper; import com.baomidou.mybatisplus.mapper.Wrapper; import com.stylefeng.guns.core.base.controller.BaseController; import com.stylefeng.guns.core.base.tips.SuccessTip; import com.stylefeng.guns.core.log.LogObjectHolder; import com.stylefeng.guns.core.shiro.ShiroKit; import com.stylefeng.guns.core.shiro.ShiroUser; 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.ExcelUtil; import com.stylefeng.guns.modular.system.controller.util.HttpUtils; import com.stylefeng.guns.modular.system.controller.util.TokenUtils; import com.stylefeng.guns.modular.system.controller.util.UUIDUtil; import com.stylefeng.guns.modular.system.enums.UserTypeEnum; import com.stylefeng.guns.modular.system.model.TDriver; import com.stylefeng.guns.modular.system.model.TDriverWork; import com.stylefeng.guns.modular.system.model.TRechargeRecord; import com.stylefeng.guns.modular.system.model.TRegion; import com.stylefeng.guns.modular.system.service.*; import com.stylefeng.guns.modular.system.util.DateUtil; import com.stylefeng.guns.modular.system.util.RedisUtil; import io.swagger.annotations.ApiImplicitParam; import io.swagger.annotations.ApiImplicitParams; import io.swagger.annotations.ApiOperation; import org.apache.poi.hssf.usermodel.HSSFWorkbook; import org.apache.shiro.subject.Subject; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.data.redis.core.RedisTemplate; import org.springframework.stereotype.Controller; import org.springframework.transaction.annotation.Transactional; import org.springframework.ui.Model; import org.springframework.util.StringUtils; import org.springframework.web.bind.annotation.PathVariable; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RequestParam; import org.springframework.web.bind.annotation.ResponseBody; import org.springframework.web.multipart.MultipartFile; import javax.servlet.http.HttpServletResponse; import java.io.File; import java.io.IOException; import java.io.OutputStream; import java.math.BigDecimal; import java.text.DateFormat; import java.text.SimpleDateFormat; import java.time.LocalDate; import java.time.Period; import java.util.ArrayList; import java.util.Date; import java.util.List; import java.util.Objects; import java.util.stream.Collectors; /** * 控制器 * * @author fengshuonan * @Date 2023-02-20 09:07:06 */ @Controller @RequestMapping("/tDriver") public class TDriverController extends BaseController { private String PREFIX = "/system/tDriver/"; @Autowired private ITDriverService tDriverService; @Autowired private ITRegionService tRegionService; @Autowired private ITAgentService tAgentService; @Autowired private ITBranchOfficeService tBranchOfficeService; @Autowired private RedisTemplate redisTemplate; @Autowired private HttpUtils httpUtils; @Autowired private TokenUtils tokenUtils; @Autowired private ITDriverWorkService tDriverWorkService; @Autowired private RedisUtil redisUtil; @Autowired private ITRechargeRecordService tRechargeRecordService; private Logger log = LoggerFactory.getLogger(this.getClass()); /** * 跳转到首页 */ @RequestMapping("") public String index() { return PREFIX + "tDriver.html"; } /** * 跳转到佣金首页 */ @RequestMapping("/commission") public String commissionIndex() { return PREFIX + "tDriverCommission.html"; } /** * 跳转到添加 */ @RequestMapping("/tDriver_add") public String tDriverAdd() { return PREFIX + "tDriver_add.html"; } /** * 跳转到修改 */ @RequestMapping("/tDriver_update") public String tDriverUpdate( Integer tDriverId, Model model) { TDriver tDriver = tDriverService.selectById(tDriverId); 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().eq("code", tDriver.getAreaCode()) .last("LIMIT 1")); TRegion city = tRegionService.selectOne(new EntityWrapper().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()); // } if(StringUtils.hasLength(tDriver.getAreaName()) && !tDriver.getAreaName().equals(tDriver.getCityName())){ tDriverResp.setArea(tDriver.getProvinceName()+"/"+tDriver.getCityName()+"/"+tDriver.getAreaName()); }else { tDriverResp.setArea(tDriver.getProvinceName()+"/"+tDriver.getCityName()); } if(Objects.nonNull(city)){ if(Objects.nonNull(district) && !district.getCode().equals(city.getCode())){ tDriverResp.setAreaId(city.getParentId()+"/"+city.getId()+"/"+district.getId()); }else { tDriverResp.setAreaId(city.getParentId()+"/"+city.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.auditPage(tDriverId,model); return PREFIX + "tDriverDetail.html"; } /** * 佣金跳转详情页面 */ @RequestMapping("/commission/driverCommissionDetail") public String driverCommissionDetail(Integer tDriverId,Integer levelFlag, Model model) { tDriverService.driverCommissionDetail(tDriverId,levelFlag,model); return PREFIX + "tDriverCommissionDetail.html"; } /** * 跳转区域页面编辑 */ @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]); 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]); if(split1.length>2) { model.addAttribute("districtId", split1[2]); }else { model.addAttribute("districtId", ""); } List tRegions = tRegionService.selectList(new EntityWrapper().eq("parent_id", 0)); model.addAttribute("provinceList",tRegions); List provinceIds = tRegions.stream().map(TRegion::getId).collect(Collectors.toList()); // 查询市 List tRegions1 = tRegionService.selectList(new EntityWrapper().eq("parent_id", split1[0])); model.addAttribute("cityList",tRegions1); // 查询区 List cityIds = tRegions1.stream().map(TRegion::getId).collect(Collectors.toList()); List tRegions2 = tRegionService.selectList(new EntityWrapper().in("parent_id", split1[1])); model.addAttribute("districtList",tRegions2); return PREFIX + "tDriverAreaUpdate.html"; } /** * 跳转区域页面新增 */ @RequestMapping("/areaPageAdd") public String areaPageAdd(String area,String areaId,Model model) { String[] split1 = areaId.split("/"); List tRegions = tRegionService.selectList(new EntityWrapper().eq("parent_id", 0)); model.addAttribute("provinceList",tRegions); // List provinceIds = tRegions.stream().map(TRegion::getId).collect(Collectors.toList()); // // 查询市 // List tRegions1 = tRegionService.selectList(new EntityWrapper().eq("parent_id", split1[0])); // model.addAttribute("cityList",tRegions1); // // // 查询区 // List cityIds = tRegions1.stream().map(TRegion::getId).collect(Collectors.toList()); // List tRegions2 = tRegionService.selectList(new EntityWrapper().in("parent_id", cityIds)); // model.addAttribute("districtList",tRegions2); if(StringUtils.hasLength(area) && StringUtils.hasLength(areaId)){ String[] split = area.split("/"); model.addAttribute("provinceName",split[0]); model.addAttribute("cityName",split[1]); if(split.length>2){ model.addAttribute("districtName",split[2]); }else { model.addAttribute("districtName",""); } model.addAttribute("provinceId",split1[0]); model.addAttribute("cityId",split1[1]); if(split1.length>2) { model.addAttribute("districtId", split1[2]); }else { model.addAttribute("districtId", ""); } List tRegions1 = tRegionService.selectList(new EntityWrapper().eq("parent_id", split1[0])); model.addAttribute("cityList",tRegions1); // 查询区 List tRegions2 = tRegionService.selectList(new EntityWrapper().in("parent_id", split1[1])); model.addAttribute("districtList",tRegions2); }else { model.addAttribute("cityList",new ArrayList<>()); model.addAttribute("districtList",new ArrayList<>()); model.addAttribute("provinceName",""); model.addAttribute("cityName","split[1]"); model.addAttribute("districtName",""); model.addAttribute("provinceId",""); model.addAttribute("cityId","split1[1]"); model.addAttribute("districtId", ""); } return PREFIX + "tDriverAreaAdd.html"; } /** * 获取列表 */ @RequestMapping(value = "/list") @ResponseBody public Object list(String createTime,String phone,Integer status) { EntityWrapper wrapper = tDriverService.getPageList(createTime,phone,status); wrapper.orderBy("createTime",false); List tDrivers = tDriverService.selectList(wrapper); List tDriverResp = tDriverService.getTDriverResp(tDrivers); for (TDriverResp driverResp : tDriverResp) { TDriverWork tDriverWork = tDriverWorkService.selectOne(new EntityWrapper() .eq("driverId", driverResp.getId()) .orderBy("workTime", false) .last("LIMIT 1")); if(Objects.nonNull(tDriverWork)){ // 如果是下班状态,计算未上线天数,,如果为上班状态,则设置为0 if(tDriverWork.getStatus() == 1){ driverResp.setNotOnlineCount(0); }else { Period period = Period.between(DateUtil.dateToLocalDate(tDriverWork.getOffWorkTime()), LocalDate.now()); driverResp.setNotOnlineCount(period.getYears()*365 + period.getMonths()*30 + Math.abs(period.getDays())); } }else { if(Objects.nonNull(driverResp.getApprovalTime())){ // 没有上班记录,计算审核时间 Period period = Period.between(DateUtil.dateToLocalDate(driverResp.getApprovalTime()), LocalDate.now()); driverResp.setNotOnlineCount(period.getYears()*365 + period.getMonths()*30 + Math.abs(period.getDays())); }else { driverResp.setNotOnlineCount(0); } } } return tDriverResp; } /** * 获取佣金列表 */ @RequestMapping(value = "/commission/list") @ResponseBody public Object commissionList(String name,String phone,Integer status) { EntityWrapper wrapper = tDriverService.getCommissionPageList(name,phone,status); List tDrivers = tDriverService.selectList(wrapper); List commissionResp = tDriverService.getTDriverCommissionResp(tDrivers); for (TDriverCommissionResp tDriverCommissionResp : commissionResp) { tDriverCommissionResp.setLevelFlag(0); } return commissionResp; } /** * 获取异常列表 */ @RequestMapping(value = "/exceptionList") @ResponseBody public Object exceptionList(String createTime,String phone,Integer status) { EntityWrapper wrapper = tDriverService.getPageList(createTime,phone,status); wrapper.eq("isException",2); wrapper.orderBy("createTime",false); List tDrivers = tDriverService.selectList(wrapper); List tDriverResp = tDriverService.getTDriverResp(tDrivers); for (TDriverResp driverResp : tDriverResp) { TDriverWork tDriverWork = tDriverWorkService.selectOne(new EntityWrapper() .eq("driverId", driverResp.getId()) .orderBy("workTime", false) .last("LIMIT 1")); if(Objects.nonNull(tDriverWork)){ // 如果是下班状态,计算未上线天数,,如果为上班状态,则设置为0 if(tDriverWork.getStatus() == 1){ driverResp.setNotOnlineCount(0); }else { Period period = Period.between(DateUtil.dateToLocalDate(tDriverWork.getOffWorkTime()), LocalDate.now()); driverResp.setNotOnlineCount(period.getYears()*365 + period.getMonths()*30 + Math.abs(period.getDays())); } }else { if(Objects.nonNull(driverResp.getApprovalTime())){ // 没有上班记录,计算审核时间 Period period = Period.between(DateUtil.dateToLocalDate(driverResp.getApprovalTime()), LocalDate.now()); driverResp.setNotOnlineCount(period.getYears()*365 + period.getMonths()*30 + Math.abs(period.getDays())); }else { driverResp.setNotOnlineCount(0); } } } return tDriverResp; } /** * 佣金关联列表 */ @RequestMapping(value = "/commission/connect/list") @ResponseBody public Object commissionConnectList(String driverId,Integer levelFlag) { // 查询当前用户的用户信息 TDriver tDriver = tDriverService.selectById(driverId); EntityWrapper wrapper = new EntityWrapper<>(); if(StringUtils.hasLength(driverId)){ wrapper.eq("inviterId",driverId); } wrapper.orderBy("createTime",false); List list = tDriverService.selectList(wrapper); List commissionResp = new ArrayList<>(); try{ commissionResp = tDriverService.getTDriverCommissionResp(list); for (TDriverCommissionResp tDriverCommissionResp : commissionResp) { if(Objects.nonNull(tDriver)){ tDriverCommissionResp.setInviterName(tDriver.getName()); tDriverCommissionResp.setInviterPhone(tDriver.getPhone()); if(levelFlag == 0){ tDriverCommissionResp.setLevelFlag(1); tDriverCommissionResp.setLevel("一级"); } if(levelFlag == 1){ tDriverCommissionResp.setLevelFlag(2); tDriverCommissionResp.setLevel("二级"); } if(levelFlag == 2){ tDriverCommissionResp.setLevelFlag(3); tDriverCommissionResp.setLevel("三级"); } } } }catch (Exception e){ e.printStackTrace(); } return commissionResp; } /** * 获取列表 */ @RequestMapping(value = "/list-back") @ResponseBody public Object listBack(String condition) { return tDriverService.selectList(null); } /** * ocr */ @RequestMapping(value = "/ocr") @ResponseBody public JSONObject ocr(MultipartFile multipartFile) { /*String accessToken = redisTemplate.opsForValue().get(TokenUtils.ACCESS_TOKEN_CACHE_KEY); if (!StringUtils.hasLength(accessToken)) { accessToken = tokenUtils.getSimpleAccessToken(); } File file = new File(imgUrl); String url = Constant.OCR_URL.replace("MODE", "scan").replace("ACCESS_TOKEN", accessToken); String result = httpUtils.registRequest(url, file); JSONObject jsonObject = JSONObject.parseObject(result); System.err.println(jsonObject); return jsonObject;*/ String originalFilename = multipartFile.getOriginalFilename(); String[] filename = originalFilename.split("\\."); log.info("tupian::"+filename[0]+"//"+filename[1]); File file= null; try { file = File.createTempFile(filename[0], "."+filename[1]); multipartFile.transferTo(file); file.deleteOnExit(); log.info("file:"+file.getName()); } catch (IOException e) { log.error("ocr识别失败:{}",e.getMessage()); } return tDriverService.ocr(file); } /** * 审核 */ @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().eq("phone", tDriver.getPhone())); if(count>0){ return new SuccessTip(500,"该司机已存在!"); } Object o = tDriverService.addOrUpdate(tDriver); if(Objects.nonNull(o)){ return o; } // 默认值板块 tDriver.setName(tDriver.getName().replace(" ","")); tDriver.setEmergencyContact(tDriver.getEmergencyContact().replace(" ","")); tDriver.setCode(UUIDUtil.getNumberRandom(16)); tDriver.setBalance(BigDecimal.ZERO); tDriver.setBackgroundBalance(BigDecimal.ZERO); tDriver.setCommission(BigDecimal.ZERO); tDriverService.insert(tDriver); return SUCCESS_TIP; } /** * 删除 */ @RequestMapping(value = "/delete") @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); String value = redisUtil.getValue("DRIVER_" + tDriver.getPhone()); redisUtil.remove(value); redisUtil.remove("DRIVER_" + tDriver.getPhone()); TDriverWork tDriverWork = tDriverWorkService.selectOne(new EntityWrapper() .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); } tDriver.setRemark(remark); tDriverService.updateById(tDriver); return SUCCESS_TIP; } /** * 充值余额 */ @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()); Integer id1 = Objects.requireNonNull(ShiroKit.getUser()).getObjectId(); tRechargeRecord.setAgentId(id1); tRechargeRecordService.insert(tRechargeRecord); return SUCCESS_TIP; } /** * 修改 */ @RequestMapping(value = "/update") @ResponseBody public Object update(TDriver tDriver) { TDriver driver = tDriverService.selectOne(new EntityWrapper().eq("phone", tDriver.getPhone()) .last("LIMIT 1")); if(Objects.nonNull(driver) && !tDriver.getId().equals(driver.getId())){ return new SuccessTip(500,"该司机已存在!"); } // Object ocr = ocr("E:\\071bf986db0b00355c0ed190bbd3b16.png"); // System.err.println(ocr); Object o = tDriverService.addOrUpdate(tDriver); if(Objects.nonNull(o)){ return o; } tDriver.setName(tDriver.getName().replace(" ","")); tDriver.setEmergencyContact(tDriver.getEmergencyContact().replace(" ","")); tDriverService.updateById(tDriver); return SUCCESS_TIP; } /** * 详情 */ @RequestMapping(value = "/detail/{tDriverId}") @ResponseBody 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().eq("parent_id", 0)); } @ApiOperation(value = "市区查询",notes="市区查询") @RequestMapping(value = "/areaCity") @ResponseBody public Object areaCity(Integer parentId,Model model) { List tRegions = tRegionService.selectList(new EntityWrapper().eq("parent_id", parentId)); model.addAttribute("list",tRegions); return tRegions; } @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 wrapper = tDriverService.getPageList(createTime,phone,status); wrapper.orderBy("createTime",false); List tDrivers = tDriverService.selectList(wrapper); List tDriverResp = tDriverService.getTDriverResp(tDrivers); 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] = "删除"; } Integer approvalStatus = d.getApprovalStatus(); if(1 == approvalStatus){ values[i][15] = "待审核"; }else if(2 == approvalStatus){ values[i][15] = "已同意"; }else { values[i][15] = "已拒绝"; } values[i][16] = 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 wrapper = tDriverService.getPageList(createTime,phone,status); wrapper.eq("isException",2); wrapper.orderBy("createTime",false); List tDrivers = tDriverService.selectList(wrapper); List tDriverResp = tDriverService.getTDriverResp(tDrivers); for (TDriverResp driverResp : tDriverResp) { TDriverWork tDriverWork = tDriverWorkService.selectOne(new EntityWrapper() .eq("driverId", driverResp.getId()) .orderBy("workTime", false) .last("LIMIT 1")); if(Objects.nonNull(tDriverWork)){ // 如果是下班状态,计算未上线天数,,如果为上班状态,则设置为0 if(tDriverWork.getStatus() == 1){ driverResp.setNotOnlineCount(0); }else { Period period = Period.between(DateUtil.dateToLocalDate(tDriverWork.getOffWorkTime()), LocalDate.now()); driverResp.setNotOnlineCount(Math.abs(period.getDays())); } }else { // 没有上班记录,计算审核时间 Period period = Period.between(DateUtil.dateToLocalDate(driverResp.getApprovalTime()), LocalDate.now()); driverResp.setNotOnlineCount(Math.abs(period.getDays())); } } 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(); } } @ApiOperation(value = "导出司机佣金列表",notes="导出司机佣金列表") @ApiImplicitParams({ @ApiImplicitParam(name = "Authorization", value = "用户token(Bearer +token)", required = true, dataType = "String", paramType = "header", defaultValue = "Bearer eyJhbGciOiJIUzUxMiJ9....."), }) @RequestMapping(value = "/export-commission") @ResponseBody public void exportCommission(String name,String phone,Integer status,HttpServletResponse response) { try { Date date = new Date(); DateFormat format = new SimpleDateFormat("yyyyMMdd"); String time1 = format.format(date); String fileName = "CommissionInfo"+time1+".xls"; String[] title = new String[] {"姓名","手机号","所属代理商","推广人数","关联人数", "累计获得佣金","可提现佣金","已提现金额","状态"}; EntityWrapper wrapper = tDriverService.getCommissionPageList(name,phone,status); // wrapper.ne("isException",2); // 是否异常 List list = tDriverService.selectList(wrapper); List commissionResp = tDriverService.getTDriverCommissionResp(list); String[][] values = new String[commissionResp.size()][]; for (int i = 0; i < commissionResp.size(); i++) { TDriverCommissionResp d = commissionResp.get(i); values[i] = new String[title.length]; values[i][0] = d.getName(); values[i][1] = d.getPhone(); values[i][2] = d.getAgentName(); values[i][3] = String.valueOf(Objects.isNull(d.getNumberPromoters()) ? 0:d.getNumberPromoters()); values[i][4] = String.valueOf(d.getConnectedPersons()); values[i][5] = String.valueOf(d.getAccumulatedCommission()); values[i][6] = String.valueOf(Objects.isNull(d.getCommission())?0:d.getCommission()); values[i][7] = String.valueOf(d.getWithdrawnAmount()); Integer status1 = d.getStatus(); if(1 == status1){ values[i][8] = "正常"; }else if(2 == status1){ values[i][8] = "冻结"; }else if(3 == status1){ values[i][8] = "已删除"; } } 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(); } } }