diff --git a/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/service/impl/AgencyServiceImpl.java b/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/service/impl/AgencyServiceImpl.java index efccd75ce2..1505018fa9 100644 --- a/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/service/impl/AgencyServiceImpl.java +++ b/epmet-module/gov-org/gov-org-server/src/main/java/com/epmet/service/impl/AgencyServiceImpl.java @@ -926,6 +926,7 @@ public class AgencyServiceImpl implements AgencyService { @Override public List getCustomerTree(TokenDto tokenDto) { + List resultList = new ArrayList<>(); List rootList = customerAgencyDao.selectRootCustomer(null, null); List treeList = new ArrayList<>(); Map treeMap = new HashMap<>(); @@ -936,24 +937,11 @@ public class AgencyServiceImpl implements AgencyService { }); // treeMap = treeList.stream().collect(Collectors.toMap(CustomerTreeDTO::getAreaCode, Function.identity(), (val1, val2) -> val2)); // 组合成组织树 - rootList = getCustomerTree(treeList); - // 回显客户名称 - rootList.forEach(item -> { - List customerNames = new ArrayList<>(); - List subList = customerAgencyDao.selectRootCustomer(item.getAreaCode(), item.getLevel()); - subList.forEach(customer -> { - CustomerDTO customerDTO = new CustomerDTO(); - customerDTO.setId(customer.getCustomerId()); - Result customerResult = operCrmFeignClient.getCustomerInfo(customerDTO); - if (customerResult.success() && null != customerResult.getData()) { - customerNames.add(customerResult.getData().getCustomerName()); - } - }); - item.setCustomerName(Joiner.on(",").join(customerNames)); - }); + resultList = getCustomerTree(treeList); + } - return rootList; + return resultList; } private CustomerAgencyEntity constructInsertEntity(AddAgencyV2FormDTO formDTO, CustomerAgencyDTO parent) { @@ -1033,6 +1021,21 @@ public class AgencyServiceImpl implements AgencyService { private List getCustomerTree(List treeList) { treeList = treeList.stream().distinct().collect(Collectors.toList()); + // 回显客户名称 + treeList.forEach(item -> { + List customerNames = new ArrayList<>(); + List subList = customerAgencyDao.selectRootCustomer(item.getAreaCode(), item.getLevel()); + subList.forEach(customer -> { + CustomerDTO customerDTO = new CustomerDTO(); + customerDTO.setId(customer.getCustomerId()); + Result customerResult = operCrmFeignClient.getCustomerInfo(customerDTO); + if (customerResult.success() && null != customerResult.getData()) { + customerNames.add(customerResult.getData().getCustomerName()); + } + }); + item.setCustomerName(Joiner.on(",").join(customerNames)); + }); + // 省 List province = new ArrayList(); // 市