From 8843b17a28de67fbe8d2562d224de3aceef2ffe0 Mon Sep 17 00:00:00 2001
From: "github-actions[bot]"
 <41898282+github-actions[bot]@users.noreply.github.com>
Date: Mon, 2 Dec 2024 15:04:11 +0800
Subject: [PATCH] branch-2.1: [improve](routine load) ensure load balance after
 scaling up BE nodes #44693 (#44799)

Cherry-picked from #44693

Co-authored-by: hui lai <laihui@selectdb.com>
---
 .../org/apache/doris/load/routineload/RoutineLoadManager.java    | 1 +
 1 file changed, 1 insertion(+)

diff --git a/fe/fe-core/src/main/java/org/apache/doris/load/routineload/RoutineLoadManager.java b/fe/fe-core/src/main/java/org/apache/doris/load/routineload/RoutineLoadManager.java
index 6121f34035ce06..c4cb6119e94525 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/load/routineload/RoutineLoadManager.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/load/routineload/RoutineLoadManager.java
@@ -486,6 +486,7 @@ public long getAvailableBeForTask(long jobId, long previousBeId) throws LoadExce
         // check if be has idle slot
         readLock();
         try {
+            updateBeIdToMaxConcurrentTasks();
             Map<Long, Integer> beIdToConcurrentTasks = getBeCurrentTasksNumMap();
             int previousBeIdleTaskNum = 0;