From bcf10abe9960c7c7f150d6da9a501349ef1fb518 Mon Sep 17 00:00:00 2001
From: ylinzhu <46126022+ylinzhu@users.noreply.github.com>
Date: Fri, 9 Dec 2022 14:00:25 +0800
Subject: [PATCH] fix load data lost connection inner 2039

fix load data lost connection inner 2039
---
 .../backend/mysql/nio/handler/MultiNodeLoadDataHandler.java     | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main/java/com/actiontech/dble/backend/mysql/nio/handler/MultiNodeLoadDataHandler.java b/src/main/java/com/actiontech/dble/backend/mysql/nio/handler/MultiNodeLoadDataHandler.java
index 90f5d898b..886b4061b 100644
--- a/src/main/java/com/actiontech/dble/backend/mysql/nio/handler/MultiNodeLoadDataHandler.java
+++ b/src/main/java/com/actiontech/dble/backend/mysql/nio/handler/MultiNodeLoadDataHandler.java
@@ -413,7 +413,6 @@ public class MultiNodeLoadDataHandler extends MultiNodeHandler implements LoadDa
                             errorCount += ok.getWarningCount();
                         }
                         dnSet.add(rrn.getName());
-                        removeNode(rrn.getName());
                         transformOkPackage(ok, shardingService);
                         if (packet == null)
                             packet = ok;
@@ -604,6 +603,7 @@ public class MultiNodeLoadDataHandler extends MultiNodeHandler implements LoadDa
             RouteResultsetNode rrn = (RouteResultsetNode) ((MySQLResponseService) service).getAttachment();
             RowDataPacket rowDataPkg = new RowDataPacket(3);
             rowDataPkg.read(row);
+            removeNode(rrn.getName());
             Map<String, List<String>> warnings = LoadDataBatch.getInstance().getWarnings();
             String name = rrn.getName();
             if (!warnings.containsKey(name)) {