Add skip_create_new_actors field to RescheduleOptions

risingwavelabs · Mar 6, 2024 · 753b9c2 · 753b9c2
1 parent 87320b5
commit 753b9c2
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 0 deletions.
diff --git a/src/meta/service/src/scale_service.rs b/src/meta/service/src/scale_service.rs
@@ -210,6 +210,7 @@ impl ScaleService for ScaleServiceImpl {
                     .collect(),
                 RescheduleOptions {
                     resolve_no_shuffle_upstream,
+                    skip_create_new_actors: false,
                 },
                 Some(table_parallelisms),
             )

diff --git a/src/meta/src/barrier/recovery.rs b/src/meta/src/barrier/recovery.rs
@@ -697,6 +697,7 @@ impl GlobalBarrierManagerContext {
                     plan,
                     RescheduleOptions {
                         resolve_no_shuffle_upstream: true,
+                        skip_create_new_actors: true,
                     },
                     Some(&mut compared_table_parallelisms),
                 )
@@ -828,6 +829,7 @@ impl GlobalBarrierManagerContext {
                     plan,
                     RescheduleOptions {
                         resolve_no_shuffle_upstream: true,
+                        skip_create_new_actors: true,
                     },
                     Some(&mut compared_table_parallelisms),
                 )

diff --git a/src/meta/src/stream/scale.rs b/src/meta/src/stream/scale.rs
@@ -2818,6 +2818,7 @@ impl GlobalStreamManager {
             reschedules,
             RescheduleOptions {
                 resolve_no_shuffle_upstream: false,
+                skip_create_new_actors: false,
             },
             None,
         )

diff --git a/src/meta/src/stream/stream_manager.rs b/src/meta/src/stream/stream_manager.rs
@@ -727,6 +727,7 @@ impl GlobalStreamManager {
                     reschedules,
                     RescheduleOptions {
                         resolve_no_shuffle_upstream: false,
+                        skip_create_new_actors: false,
                     },
                     Some(table_parallelism_assignment),
                 )