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

Signed-off-by: Jiao Mingye <mxdzs0612@gmail.com>
StarRocks · Jan 21, 2025 · 9ae9ada · 9ae9ada
1 parent d650aa8
commit 9ae9ada
Showing 1 changed file with 9 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;
@@ -984,6 +988,11 @@ private Map<Pair<Long, Long>, Set<Long>> getPartitionTablets(long beId, TStorage
                 }
             }
 
+            OlapTable olapTable = getOlapTableById(tabletMeta.getDbId(), tabletMeta.getTableId());
+            if (olapTable != null && !olapTable.needSchedule(true)) {
+                continue;
+            }
+
             Pair<Long, Long> key = new Pair<>(tabletMeta.getPhysicalPartitionId(), tabletMeta.getIndexId());
             partitionTablets.computeIfAbsent(key, k -> Sets.newHashSet()).add(tabletId);
         }