Merge branch 'implement-buildScan' of https://github.com/GoogleCloudD…

…ataproc/spark-spanner-connector into implement-buildScan
GoogleCloudDataproc · Sep 7, 2023 · 451d35e · 451d35e
2 parents ba42f4f + 3804b28
commit 451d35e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/spark-3.1-spanner-lib/src/main/java/com/google/cloud/spark/spanner/SpannerScanBuilder.java b/spark-3.1-spanner-lib/src/main/java/com/google/cloud/spark/spanner/SpannerScanBuilder.java
@@ -83,6 +83,10 @@ public InputPartition[] planInputPartitions() {
     // TODO: Receive the columns and filters that were pushed down.
     BatchClientWithCloser batchClient = SpannerUtils.batchClientFromProperties(this.opts);
     String sqlStmt = "SELECT * FROM " + this.opts.get("table");
+    Filter[] filters = this.pushedFilters();
+    if (filters.length > 0) {
+      sqlStmt += " WHERE " + SparkFilterUtils.getCompiledFilter(true, filters);
+    }
     try (BatchReadOnlyTransaction txn =
         batchClient.batchClient.batchReadOnlyTransaction(TimestampBound.strong())) {
       String mapAsJSON = SpannerUtils.serializeMap(this.opts.asCaseSensitiveMap());