Avoid spliting txn shards

w41ter · Nov 21, 2024 · 3060978 · 3060978
1 parent 0c8e35b
commit 3060978
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/server/src/root/schedule.rs b/src/server/src/root/schedule.rs
@@ -597,7 +597,7 @@ impl ScheduleContext {
         Ok(SchedResult::ack())
     }
 
-    /// Handle the spliting shard stask and update the sched stats.
+    /// Handle the spliting shard shards and update the sched stats.
     async fn handle_split_shard(&self, task: &mut SplitShardTask) -> Result<SchedResult> {
         let result = self.handle_split_shard_inner(task).await?;
         if result.ack {

diff --git a/src/server/src/root/stats.rs b/src/server/src/root/stats.rs
@@ -137,6 +137,10 @@ impl ClusterStats {
                     continue;
                 }
                 if let Some(&group_id) = table_stats.shard_indexes.get(&shard_stats.shard_id) {
+                    if group_id == sekas_schema::ROOT_GROUP_ID {
+                        // Don't split the root groups, such as txn shard.
+                        continue;
+                    }
                     target_shards.push((group_id, shard_stats.shard_id));
                     if target_shards.len() >= limit {
                         break;