package com.supersavedriving.driver.modular.system.service.impl;
|
|
import com.baomidou.mybatisplus.mapper.EntityWrapper;
|
import com.baomidou.mybatisplus.service.impl.ServiceImpl;
|
import com.supersavedriving.driver.modular.system.dao.AgentMapper;
|
import com.supersavedriving.driver.modular.system.dao.TBranchOfficeAreaMapper;
|
import com.supersavedriving.driver.modular.system.model.Agent;
|
import com.supersavedriving.driver.modular.system.model.BranchOffice;
|
import com.supersavedriving.driver.modular.system.model.TBranchOfficeArea;
|
import com.supersavedriving.driver.modular.system.service.IAgentService;
|
import com.supersavedriving.driver.modular.system.service.IBranchOfficeService;
|
import com.supersavedriving.driver.modular.system.service.ITBranchOfficeAreaService;
|
import com.supersavedriving.driver.modular.system.warpper.OpenCityWarpper;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
/**
|
* @author zhibing.pu
|
* @Date 2023/3/22 22:52
|
*/
|
@Service
|
public class AgentServiceImpl extends ServiceImpl<AgentMapper, Agent> implements IAgentService {
|
|
@Autowired
|
private TBranchOfficeAreaMapper branchOfficeAreaMapper;
|
|
@Autowired
|
private IBranchOfficeService branchOfficeService;
|
|
/**
|
* 获取开通省市数据
|
* @return
|
* @throws Exception
|
*/
|
@Override
|
public List<OpenCityWarpper> queryOpenCity() throws Exception {
|
List<BranchOffice> branchOffices = branchOfficeService.selectList(new EntityWrapper<BranchOffice>().eq("status", 1));
|
List<Integer> collect = new ArrayList<>();
|
if(branchOffices.size() == 0){
|
collect.add(-1);
|
}else{
|
collect = branchOffices.stream().map(BranchOffice::getId).collect(Collectors.toList());
|
}
|
List<TBranchOfficeArea> branchOfficeAreas = branchOfficeAreaMapper.selectList(new EntityWrapper<TBranchOfficeArea>().in("branchOfficeId", collect));
|
List<OpenCityWarpper> province = new ArrayList<>();
|
|
//遍历处理省级数据
|
List<String> provinceCodes = new ArrayList<>();
|
for (TBranchOfficeArea branchOfficeArea : branchOfficeAreas) {
|
String provinceCode = branchOfficeArea.getProvinceCode();
|
if(!provinceCodes.contains(provinceCode)){
|
provinceCodes.add(provinceCode);
|
|
OpenCityWarpper openCityWarpper = new OpenCityWarpper();
|
openCityWarpper.setCode(provinceCode);
|
openCityWarpper.setName(branchOfficeArea.getProvinceName());
|
province.add(openCityWarpper);
|
}
|
}
|
|
//遍历处理省级对应的市级数据
|
for (OpenCityWarpper openCityWarpper : province) {
|
String code = openCityWarpper.getCode();
|
|
List<OpenCityWarpper> city = new ArrayList<>();
|
List<String> cityCodes = new ArrayList<>();
|
for (TBranchOfficeArea branchOfficeArea : branchOfficeAreas) {
|
String provinceCode1 = branchOfficeArea.getProvinceCode();
|
String cityCode = branchOfficeArea.getCityCode();
|
|
if(code.equals(provinceCode1) && !cityCodes.contains(cityCode)){
|
cityCodes.add(cityCode);
|
|
OpenCityWarpper openCityWarpper1 = new OpenCityWarpper();
|
openCityWarpper1.setCode(cityCode);
|
openCityWarpper1.setName(branchOfficeArea.getCityName());
|
city.add(openCityWarpper1);
|
}
|
}
|
openCityWarpper.setSublevel(city);
|
}
|
return province;
|
}
|
// /**
|
// * 获取开通省市数据
|
// * @return
|
// * @throws Exception
|
// */
|
// @Override
|
// public List<OpenCityWarpper> queryOpenCity() throws Exception {
|
// List<Agent> branchOffices = this.selectList(new EntityWrapper<Agent>().eq("status", 1));
|
// List<OpenCityWarpper> province = new ArrayList<>();
|
//
|
// //遍历处理省级数据
|
// List<String> provinceCodes = new ArrayList<>();
|
// for (Agent branchOffice : branchOffices) {
|
// String provinceCode = branchOffice.getProvinceCode();
|
// if(!provinceCodes.contains(provinceCode)){
|
// provinceCodes.add(provinceCode);
|
//
|
// OpenCityWarpper openCityWarpper = new OpenCityWarpper();
|
// openCityWarpper.setCode(provinceCode);
|
// openCityWarpper.setName(branchOffice.getProvinceName());
|
// province.add(openCityWarpper);
|
// }
|
// }
|
//
|
// //遍历处理省级对应的市级数据
|
// for (OpenCityWarpper openCityWarpper : province) {
|
// String code = openCityWarpper.getCode();
|
//
|
// List<OpenCityWarpper> city = new ArrayList<>();
|
// List<String> cityCodes = new ArrayList<>();
|
// for (Agent branchOffice : branchOffices) {
|
// String provinceCode1 = branchOffice.getProvinceCode();
|
// String cityCode = branchOffice.getCityCode();
|
//
|
// if(code.equals(provinceCode1) && !cityCodes.contains(cityCode)){
|
// cityCodes.add(cityCode);
|
//
|
// OpenCityWarpper openCityWarpper1 = new OpenCityWarpper();
|
// openCityWarpper1.setCode(cityCode);
|
// openCityWarpper1.setName(branchOffice.getCityName());
|
// city.add(openCityWarpper1);
|
// }
|
// }
|
// openCityWarpper.setSublevel(city);
|
// }
|
// return province;
|
// }
|
|
}
|
