Add --cutover flag to schema-sync (#608)

Author: levkk

integration/schema_sync/dev.sh

@@ -24,6 +24,13 @@ ${PGDOG_BIN_PATH} \
     --publication pgdog \
     --data-sync-complete
 
+${PGDOG_BIN_PATH} \
+    schema-sync \
+    --from-database source \
+    --to-database destination \
+    --publication pgdog \
+    --cutover
+
 pg_dump \
     --schema-only \
     --exclude-schema pgdog \

pgdog/src/backend/schema/sync/pg_dump.rs

@@ -367,7 +367,7 @@ impl PgDumpOutput {
                                         sequence,
                                         sql: original,
                                     });
-                                } else {
+                                } else if state == SyncState::Cutover {
                                     let sql = sequence
                                         .setval_from_column(&column)
                                         .map_err(|_| Error::MissingEntity)?;
@@ -438,10 +438,12 @@ impl PgDumpOutput {
 
                         NodeEnum::AlterOwnerStmt(stmt) => {
                             if stmt.object_type() != ObjectType::ObjectPublication {
-                                result.push(Statement::Other {
-                                    sql: original.to_string(),
-                                    idempotent: true,
-                                });
+                                if state == SyncState::PreData {
+                                    result.push(Statement::Other {
+                                        sql: original.to_string(),
+                                        idempotent: true,
+                                    });
+                                }
                             }
                         }
 

pgdog/src/cli.rs

@@ -126,6 +126,10 @@ pub enum Commands {
         /// Data sync has been complete.
         #[arg(long)]
         data_sync_complete: bool,
+
+        /// Execute cutover statements.
+        #[arg(long)]
+        cutover: bool,
     },
 
     /// Perform cluster configuration steps
@@ -271,14 +275,15 @@ pub async fn data_sync(commands: Commands) -> Result<(), Box<dyn std::error::Err
 
 #[allow(clippy::print_stdout)]
 pub async fn schema_sync(commands: Commands) -> Result<(), Box<dyn std::error::Error>> {
-    let (source, destination, publication, dry_run, ignore_errors, data_sync_complete) =
+    let (source, destination, publication, dry_run, ignore_errors, data_sync_complete, cutover) =
         if let Commands::SchemaSync {
             from_database,
             to_database,
             publication,
             dry_run,
             ignore_errors,
             data_sync_complete,
+            cutover,
         } = commands
         {
             let source = databases().schema_owner(&from_database)?;
@@ -291,6 +296,7 @@ pub async fn schema_sync(commands: Commands) -> Result<(), Box<dyn std::error::E
                 dry_run,
                 ignore_errors,
                 data_sync_complete,
+                cutover,
             )
         } else {
             return Ok(());
@@ -300,6 +306,8 @@ pub async fn schema_sync(commands: Commands) -> Result<(), Box<dyn std::error::E
     let output = dump.dump().await?;
     let state = if data_sync_complete {
         SyncState::PostData
+    } else if cutover {
+        SyncState::Cutover
     } else {
         SyncState::PreData
     };