also set singleton to 1 in migration script

Signed-off-by: Bugen Zhao <[email protected]>
risingwavelabs · Oct 12, 2024 · 94dadd9 · 94dadd9
1 parent d350f94
commit 94dadd9
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/src/meta/model_v2/migration/src/m20240911_083152_variable_vnode_count.rs b/src/meta/model_v2/migration/src/m20240911_083152_variable_vnode_count.rs
@@ -21,6 +21,18 @@ impl MigrationTrait for Migration {
             )
             .await?;
 
+        manager
+            .exec_stmt(
+                UpdateStatement::new()
+                    .table(Table::Table)
+                    .values([(Table::VnodeCount, Expr::value(1))])
+                    .and_where(Expr::col(Table::DistributionKey).eq(Expr::value("[]")))
+                    .and_where(Expr::col(Table::DistKeyInPk).eq(Expr::value("[]")))
+                    .and_where(Expr::col(Table::VnodeColIndex).is_null())
+                    .to_owned(),
+            )
+            .await?;
+
         manager
             .alter_table(
                 MigrationTable::alter()
@@ -30,6 +42,16 @@ impl MigrationTrait for Migration {
             )
             .await?;
 
+        manager
+            .exec_stmt(
+                UpdateStatement::new()
+                    .table(Fragment::Table)
+                    .values([(Fragment::VnodeCount, Expr::value(1))])
+                    .and_where(Expr::col(Fragment::DistributionType).eq(Expr::value("SINGLE")))
+                    .to_owned(),
+            )
+            .await?;
+
         manager
             .alter_table(
                 MigrationTable::alter()
@@ -74,12 +96,16 @@ impl MigrationTrait for Migration {
 enum Fragment {
     Table,
     VnodeCount,
+    DistributionType,
 }
 
 #[derive(DeriveIden)]
 enum Table {
     Table,
     VnodeCount,
+    DistributionKey,
+    DistKeyInPk,
+    VnodeColIndex,
 }
 
 #[derive(DeriveIden)]