[BugFix] Fix the issue of repeated migration of colocate tablets

Signed-off-by: Jiao Mingye <[email protected]>
StarRocks · Jan 15, 2025 · 41ab0d6 · 41ab0d6
1 parent d650aa8
commit 41ab0d6
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/fe/fe-core/src/main/java/com/starrocks/clone/DiskAndTabletLoadReBalancer.java b/fe/fe-core/src/main/java/com/starrocks/clone/DiskAndTabletLoadReBalancer.java
@@ -554,6 +554,10 @@ private List<TabletSchedCtx> balanceClusterDisk(ClusterLoadStatistic clusterStat
                             continue;
                         }
 
+                        if (!olapTable.needSchedule(false)) {
+                            continue;
+                        }
+
                         if (isDestBackendLocationMismatch(olapTable, hBackend.getId(), lBackend.getId(),
                                 physicalPartition.getParentId(), tabletId)) {
                             continue;
@@ -799,6 +803,11 @@ private void balanceBackendDisk(TStorageMedium medium, double avgUsedPercent,
                 if (olapTable == null) {
                     continue;
                 }
+
+                if (!olapTable.needSchedule(true)) {
+                    continue;
+                }
+
                 // check tablet healthy
                 if (isTabletUnhealthy(tabletMeta.getDbId(), olapTable, tabletId, tabletMeta, aliveBeIds)) {
                     continue;
@@ -1646,6 +1655,10 @@ private Map<Pair<Long, Long>, PartitionStat> getPartitionStats(TStorageMedium me
                         continue;
                     }
 
+                    if (!olapTbl.needSchedule(isLocalBalance)) {
+                        continue;
+                    }
+
                     for (Partition partition : globalStateMgr.getLocalMetastore().getAllPartitionsIncludeRecycleBin(olapTbl)) {
                         partitionChecked++;
                         if (partitionChecked % partitionBatchNum == 0) {