query layer support

PeerDB-io · Nov 29, 2024 · e9ea720 · e9ea720
1 parent b336ac5
commit e9ea720
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/nexus/analyzer/src/qrep.rs b/nexus/analyzer/src/qrep.rs
@@ -93,6 +93,12 @@ const QREP_OPTIONS: &[QRepOptionType] = &[
         default_value: false,
         required: false,
     },
+    QRepOptionType::String {
+        name: "watermark_query",
+        default_val: Some(""),
+        required: true,
+        accepted_values: None,
+    },
 ];
 
 pub fn process_options(

diff --git a/nexus/flow-rs/src/grpc.rs b/nexus/flow-rs/src/grpc.rs
@@ -173,6 +173,7 @@ impl FlowGrpcClient {
                 Value::String(s) => match key.as_str() {
                     "destination_table_name" => cfg.destination_table_identifier.clone_from(s),
                     "watermark_column" => cfg.watermark_column.clone_from(s),
+                    "watermark_query" => cfg.watermark_query.clone_from(s),
                     "watermark_table_name" => cfg.watermark_table.clone_from(s),
                     "mode" => {
                         let mut wm = QRepWriteMode {