From 993a85ca2a2f4bb33a0322326f7589c5d61a8c39 Mon Sep 17 00:00:00 2001 From: Pu Zhibing <393733352@qq.com> Date: 星期二, 16 九月 2025 09:43:13 +0800 Subject: [PATCH] 新增加司机选择所属企业接口 --- DriverOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/DriverController.java | 49 ++++++++++++++++++++++++++++++++++++++++++++++--- 1 files changed, 46 insertions(+), 3 deletions(-) diff --git a/DriverOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/DriverController.java b/DriverOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/DriverController.java index 1169c8f..dacaf57 100644 --- a/DriverOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/DriverController.java +++ b/DriverOKTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/api/DriverController.java @@ -1,6 +1,7 @@ package com.stylefeng.guns.modular.api; import com.alibaba.fastjson.JSON; +import com.baomidou.mybatisplus.mapper.EntityWrapper; import com.stylefeng.guns.core.util.ToolUtil; import com.stylefeng.guns.modular.crossCity.model.OrderCrossCity; import com.stylefeng.guns.modular.crossCity.server.IOrderCrossCityService; @@ -8,13 +9,11 @@ import com.stylefeng.guns.modular.smallLogistics.server.IOrderLogisticsService; import com.stylefeng.guns.modular.specialTrain.model.OrderPrivateCar; import com.stylefeng.guns.modular.specialTrain.server.IOrderPrivateCarService; -import com.stylefeng.guns.modular.system.model.Income; +import com.stylefeng.guns.modular.system.model.*; import com.stylefeng.guns.modular.system.util.ALiSendSms; //import com.stylefeng.guns.modular.system.util.ICBCPayUtil; import com.stylefeng.guns.modular.system.util.HuaWeiSMSUtil; import com.stylefeng.guns.modular.system.warpper.ActivityWarpper; -import com.stylefeng.guns.modular.system.model.Driver; -import com.stylefeng.guns.modular.system.model.OrderPosition; import com.stylefeng.guns.modular.system.service.*; import com.stylefeng.guns.modular.system.util.ResultUtil; import com.stylefeng.guns.modular.system.warpper.*; @@ -31,6 +30,7 @@ import javax.servlet.http.HttpServletResponse; import java.text.SimpleDateFormat; import java.util.*; +import java.util.stream.Collectors; /** * 司机控制器 @@ -72,6 +72,12 @@ @Autowired private IOrderLogisticsService orderLogisticsService; + + @Autowired + private ICompanyService companyService; + + @Autowired + private ICompanyCityService companyCityService; @@ -1196,4 +1202,41 @@ return ResultUtil.runErr(); } } + + + + @ResponseBody + @PostMapping("/base/driver/getCompanyByCityCode") + @ApiOperation(value = "根据省市区编号获取可选择的企业", tags = {"司机端-注册"}, notes = "") + @ApiImplicitParams({ + @ApiImplicitParam(value = "省编号", name = "provinceCode", required = true, dataType = "String"), + @ApiImplicitParam(value = "市编号", name = "cityCode", required = false, dataType = "String"), + @ApiImplicitParam(value = "区编号", name = "districtCode", required = false, dataType = "String"), + }) + public ResultUtil<List<Map<String, Object>>> getCompanyByCityCode(String provinceCode, String cityCode, String districtCode){ + List<Integer> companyId = null; + if(ToolUtil.isNotEmpty(districtCode)){ + List<CompanyCity> areaCode = companyCityService.selectList(new EntityWrapper<CompanyCity>().eq("areaCode", districtCode)); + companyId = areaCode.stream().map(CompanyCity::getCompanyId).collect(Collectors.toList()); + } + if(ToolUtil.isNotEmpty(cityCode) && (null == companyId ||companyId.isEmpty())){ + List<CompanyCity> areaCode = companyCityService.selectList(new EntityWrapper<CompanyCity>().eq("cityCode", cityCode).isNull("areaCode")); + companyId = areaCode.stream().map(CompanyCity::getCompanyId).collect(Collectors.toList()); + } + if(ToolUtil.isNotEmpty(provinceCode) && (null == companyId ||companyId.isEmpty())){ + List<CompanyCity> areaCode = companyCityService.selectList(new EntityWrapper<CompanyCity>().eq("provinceCode", provinceCode).isNull("cityCode").isNull("areaCode")); + companyId = areaCode.stream().map(CompanyCity::getCompanyId).collect(Collectors.toList()); + } + if(null != companyId && !companyId.isEmpty()){ + List<Company> companies = companyService.selectBatchIds(companyId); + List<Map<String, Object>> collect = companies.stream().map(s -> { + Map<String, Object> map = new HashMap<>(); + map.put("id", s.getId()); + map.put("name", s.getName()); + return map; + }).collect(Collectors.toList()); + return ResultUtil.success(collect); + } + return ResultUtil.success(new ArrayList<>()); + } } -- Gitblit v1.7.1