From 39739e34af928aae204c55db7974213c9b938c5a Mon Sep 17 00:00:00 2001
From: jiangping <jp@doumee.com>
Date: 星期三, 21 二月 2024 18:33:14 +0800
Subject: [PATCH] 整理

---
 server/dmvisit_service/src/main/java/com/doumee/service/common/tree/CompanyTree.java |   25 ++++++++++++++++++-------
 1 files changed, 18 insertions(+), 7 deletions(-)

diff --git a/server/dmvisit_service/src/main/java/com/doumee/service/common/tree/CompanyTree.java b/server/dmvisit_service/src/main/java/com/doumee/service/common/tree/CompanyTree.java
index cd09625..365c4bc 100644
--- a/server/dmvisit_service/src/main/java/com/doumee/service/common/tree/CompanyTree.java
+++ b/server/dmvisit_service/src/main/java/com/doumee/service/common/tree/CompanyTree.java
@@ -1,6 +1,8 @@
 package com.doumee.service.common.tree;
 
+import com.doumee.config.DataSyncConfig;
 import com.doumee.dao.business.model.Company;
+import org.apache.commons.lang3.StringUtils;
 
 import java.util.ArrayList;
 import java.util.List;
@@ -40,13 +42,18 @@
         // treeNode锛氭煡璇㈠嚭鐨勬瘡涓�鏉℃暟鎹紙鑺傜偣锛�
         for (Company treeNode : nodeList){
             // 鍒ゆ柇褰撳墠鑺傜偣鏄惁涓烘牴鑺傜偣锛屾澶勬敞鎰忥細鑻arentId绫诲瀷鏄疭tring锛屽垯瑕侀噰鐢╡quals()鏂规硶鍒ゆ柇銆�
-            if(origin==0){
-                if (Objects.isNull(treeNode.getParentId())) {
+            if(origin== DataSyncConfig.origin.hk){
+                if (Objects.isNull(treeNode.getHkParentId())) {
                     // 鏄紝娣诲姞
                     rootNodeList.add(treeNode);
                 }
-            }else{
+            }else if(origin== DataSyncConfig.origin.erp){
                 if (Objects.isNull(treeNode.getErpParentId())) {
+                    // 鏄紝娣诲姞
+                    rootNodeList.add(treeNode);
+                }
+            }else {
+                if (Objects.isNull(treeNode.getParentId())) {
                     // 鏄紝娣诲姞
                     rootNodeList.add(treeNode);
                 }
@@ -84,17 +91,21 @@
         // nodeList锛氭墍鏈夎妭鐐归泦鍚堬紙鎵�鏈夋暟鎹級
         for (Company treeNode : nodeList) {
             // 鍒ゆ柇褰撳墠鑺傜偣鐨勭埗鑺傜偣ID鏄惁绛変簬鏍硅妭鐐圭殑ID锛屽嵆褰撳墠鑺傜偣涓哄叾涓嬬殑瀛愯妭鐐�
-
-            if(origin==0){
-                if (!Objects.isNull(treeNode.getParentId())&&treeNode.getParentId().equals(pNode.getId())) {
+            if(origin==DataSyncConfig.origin.hk){
+                if (!Objects.isNull(treeNode.getHkParentId())&&treeNode.getHkParentId().equals(pNode.getHkId())) {
                     // 鍐嶉�掑綊杩涜鍒ゆ柇褰撳墠鑺傜偣鐨勬儏鍐碉紝璋冪敤鑷韩鏂规硶
                     childTree.add(buildChildTree(treeNode));
                 }
-            }else{
+            }else  if(origin==DataSyncConfig.origin.erp){
                 if (!Objects.isNull(treeNode.getErpParentId())&&treeNode.getErpParentId().equals(pNode.getErpId())) {
                     // 鍐嶉�掑綊杩涜鍒ゆ柇褰撳墠鑺傜偣鐨勬儏鍐碉紝璋冪敤鑷韩鏂规硶
                     childTree.add(buildChildTree(treeNode));
                 }
+            }else  {
+                if (!Objects.isNull(treeNode.getParentId())&&treeNode.getParentId().equals(pNode.getId())) {
+                    // 鍐嶉�掑綊杩涜鍒ゆ柇褰撳墠鑺傜偣鐨勬儏鍐碉紝璋冪敤鑷韩鏂规硶
+                    childTree.add(buildChildTree(treeNode));
+                }
             }
         }
         // for寰幆缁撴潫锛屽嵆鑺傜偣涓嬫病鏈変换浣曡妭鐐癸紝鏍戝舰鏋勫缓缁撴潫锛岃缃爲缁撴灉

--
Gitblit v1.9.3