From 9367f44aa8d49a76f8bb9e7ce38900b0717013ff Mon Sep 17 00:00:00 2001
From: lin <yuanlinzhu1995@gmail.com>
Date: Fri, 9 Dec 2022 10:17:30 +0800
Subject: [PATCH] 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 0137336bf..2a78c6590 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
@@ -414,7 +414,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;
@@ -605,6 +604,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)) {