From 802222275f344cc0b034a2c70f980e0ef4355d53 Mon Sep 17 00:00:00 2001
From: zhangyuan <zhangyuan@sarnasea.com>
Date: Tue, 19 Jul 2022 17:59:08 +0800
Subject: [PATCH] =?UTF-8?q?=E5=9B=9E=E6=98=BE=E5=AE=A2=E6=88=B7=E5=90=8D?=
 =?UTF-8?q?=E7=A7=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../epmet/service/impl/AgencyServiceImpl.java | 35 ++++++++++---------
 1 file changed, 19 insertions(+), 16 deletions(-)

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<CustomerTreeDTO> getCustomerTree(TokenDto tokenDto) {
+        List<CustomerTreeDTO> resultList = new ArrayList<>();
         List<CustomerTreeDTO> rootList = customerAgencyDao.selectRootCustomer(null, null);
         List<CustomerTreeDTO> treeList = new ArrayList<>();
         Map<String, CustomerTreeDTO> 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<String> customerNames = new ArrayList<>();
-                List<CustomerTreeDTO> subList = customerAgencyDao.selectRootCustomer(item.getAreaCode(), item.getLevel());
-                subList.forEach(customer -> {
-                    CustomerDTO customerDTO = new CustomerDTO();
-                    customerDTO.setId(customer.getCustomerId());
-                    Result<CustomerDTO> 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<CustomerTreeDTO> getCustomerTree(List<CustomerTreeDTO> treeList) {
         treeList = treeList.stream().distinct().collect(Collectors.toList());
 
+        // 回显客户名称
+        treeList.forEach(item -> {
+            List<String> customerNames = new ArrayList<>();
+            List<CustomerTreeDTO> subList = customerAgencyDao.selectRootCustomer(item.getAreaCode(), item.getLevel());
+            subList.forEach(customer -> {
+                CustomerDTO customerDTO = new CustomerDTO();
+                customerDTO.setId(customer.getCustomerId());
+                Result<CustomerDTO> customerResult = operCrmFeignClient.getCustomerInfo(customerDTO);
+                if (customerResult.success() && null != customerResult.getData()) {
+                    customerNames.add(customerResult.getData().getCustomerName());
+                }
+            });
+            item.setCustomerName(Joiner.on(",").join(customerNames));
+        });
+
         // 省
         List<CustomerTreeDTO> province = new ArrayList<CustomerTreeDTO>();
         // 市