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 | 50 ++++++++++++++++++++++++++++++++++++++++++++++----
1 files changed, 46 insertions(+), 4 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 b2885c6..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,12 +30,12 @@
import javax.servlet.http.HttpServletResponse;
import java.text.SimpleDateFormat;
import java.util.*;
+import java.util.stream.Collectors;
/**
* 司机控制器
*/
@Api
-@CrossOrigin
@RestController
@RequestMapping("")
public class DriverController {
@@ -73,6 +72,12 @@
@Autowired
private IOrderLogisticsService orderLogisticsService;
+
+ @Autowired
+ private ICompanyService companyService;
+
+ @Autowired
+ private ICompanyCityService companyCityService;
@@ -1197,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