fix: missing apply of source split discovery info after replacing tab…

…le (#14449)
risingwavelabs · Jan 10, 2024 · df3eb21 · df3eb21
1 parent e6a6558
commit df3eb21
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/meta/src/barrier/command.rs b/src/meta/src/barrier/command.rs
@@ -994,6 +994,21 @@ impl CommandContext {
                         init_split_assignment.clone(),
                     )
                     .await?;
+
+                // Apply the split changes in source manager.
+                self.barrier_manager_context
+                    .source_manager
+                    .drop_source_fragments(std::slice::from_ref(old_table_fragments))
+                    .await;
+                let source_fragments = new_table_fragments.stream_source_fragments();
+                self.barrier_manager_context
+                    .source_manager
+                    .apply_source_change(
+                        Some(source_fragments),
+                        Some(init_split_assignment.clone()),
+                        None,
+                    )
+                    .await;
             }
         }