Add spanner host param to Spanner CSV export template.

PiperOrigin-RevId: 350634680
AldaCL · Jan 7, 2021 · 2843755 · 2843755
1 parent ca0d47e
commit 2843755
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/src/main/java/com/google/cloud/teleport/templates/SpannerToText.java b/src/main/java/com/google/cloud/teleport/templates/SpannerToText.java
@@ -90,6 +90,7 @@ public static void main(String[] args) {
 
     SpannerConfig spannerConfig =
         SpannerConfig.create()
+            .withHost(options.getSpannerHost())
             .withProjectId(options.getSpannerProjectId())
             .withInstanceId(options.getSpannerInstanceId())
             .withDatabaseId(options.getSpannerDatabaseId());

diff --git a/src/main/java/com/google/cloud/teleport/templates/common/SpannerConverters.java b/src/main/java/com/google/cloud/teleport/templates/common/SpannerConverters.java
@@ -48,6 +48,7 @@
 import org.apache.beam.sdk.io.gcp.spanner.ExposedSpannerAccessor;
 import org.apache.beam.sdk.io.gcp.spanner.ReadOperation;
 import org.apache.beam.sdk.io.gcp.spanner.SpannerConfig;
+import org.apache.beam.sdk.options.Default;
 import org.apache.beam.sdk.options.Description;
 import org.apache.beam.sdk.options.PipelineOptions;
 import org.apache.beam.sdk.options.ValueProvider;
@@ -93,6 +94,13 @@ public interface SpannerReadOptions extends PipelineOptions {
 
     @SuppressWarnings("unused")
     void setSpannerDatabaseId(ValueProvider<String> spannerDatabaseId);
+
+    @Description("Spanner host")
+    @Default.String("https://batch-spanner.googleapis.com")
+    ValueProvider<String> getSpannerHost();
+
+    @SuppressWarnings("unused")
+    void setSpannerHost(ValueProvider<String> value);
   }
 
   /** Factory for Export transform class. */