From 572156dada6564a34d627c2c124feb5edbf106ee Mon Sep 17 00:00:00 2001
From: "chenboyang.922" <chenboyang.922@bytedance.com>
Date: Mon, 6 Nov 2023 17:40:03 +0800
Subject: [PATCH] fix: restore load job progress before retry load task

---
 .../src/main/java/org/apache/doris/load/loadv2/BulkLoadJob.java | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/fe/fe-core/src/main/java/org/apache/doris/load/loadv2/BulkLoadJob.java b/fe/fe-core/src/main/java/org/apache/doris/load/loadv2/BulkLoadJob.java
index 7b78efc7a50cd5..a9bda01c3e4183 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/load/loadv2/BulkLoadJob.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/load/loadv2/BulkLoadJob.java
@@ -231,6 +231,8 @@ public void onTaskFailed(long taskId, FailMsg failMsg) {
                 idToTasks.remove(loadTask.getSignature());
                 if (loadTask instanceof LoadLoadingTask) {
                     loadStatistic.removeLoad(((LoadLoadingTask) loadTask).getLoadId());
+                    // restore load progress
+                    Env.getCurrentProgressManager().registerProgressSimple(String.valueOf(id));
                 }
                 loadTask.updateRetryInfo();
                 idToTasks.put(loadTask.getSignature(), loadTask);