Write Mode Overwrite Check (#489)

Fixes the check for write mode overwrite
PeerDB-io · Oct 5, 2023 · dd178cc · dd178cc
1 parent 5b9f98d
commit dd178cc
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/nexus/flow-rs/src/grpc.rs b/nexus/flow-rs/src/grpc.rs
@@ -291,9 +291,17 @@ impl FlowGrpcClient {
             }
         }
         if !cfg.initial_copy_only {
-            return anyhow::Result::Err(anyhow::anyhow!(
-                "write mode overwrite can only be set with initial_copy_only = true"
-            ));
+            if let Some(QRepWriteMode {
+                write_type: wt,
+                upsert_key_columns: _,
+            }) = cfg.write_mode
+            {
+                if wt == QRepWriteType::QrepWriteModeOverwrite as i32 {
+                    return anyhow::Result::Err(anyhow::anyhow!(
+                        "write mode overwrite can only be set with initial_copy_only = true"
+                    ));
+                }
+            }
         }
         self.start_query_replication_flow(&cfg).await
     }