From a6354b57ab2614b6211fbb89e24c5c8bec76ce86 Mon Sep 17 00:00:00 2001
From: puzhibing <393733352@qq.com>
Date: 星期二, 16 五月 2023 10:23:44 +0800
Subject: [PATCH] 修改bug
---
driver/guns-admin/src/main/java/com/supersavedriving/driver/modular/system/service/impl/AgentServiceImpl.java | 55 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 55 insertions(+), 0 deletions(-)
diff --git a/driver/guns-admin/src/main/java/com/supersavedriving/driver/modular/system/service/impl/AgentServiceImpl.java b/driver/guns-admin/src/main/java/com/supersavedriving/driver/modular/system/service/impl/AgentServiceImpl.java
index ce2526b..2acf8e6 100644
--- a/driver/guns-admin/src/main/java/com/supersavedriving/driver/modular/system/service/impl/AgentServiceImpl.java
+++ b/driver/guns-admin/src/main/java/com/supersavedriving/driver/modular/system/service/impl/AgentServiceImpl.java
@@ -1,10 +1,15 @@
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.model.Agent;
import com.supersavedriving.driver.modular.system.service.IAgentService;
+import com.supersavedriving.driver.modular.system.warpper.OpenCityWarpper;
import org.springframework.stereotype.Service;
+
+import java.util.ArrayList;
+import java.util.List;
/**
* @author zhibing.pu
@@ -12,4 +17,54 @@
*/
@Service
public class AgentServiceImpl extends ServiceImpl<AgentMapper, Agent> implements IAgentService {
+
+
+ /**
+ * 获取开通省市数据
+ * @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;
+ }
+
}
--
Gitblit v1.7.1