Merge pull request #1479 from Polber:jkinard/fix-elasticsearch

PiperOrigin-RevId: 628428951
GoogleCloudPlatform · Apr 26, 2024 · 5c9625e · 5c9625e
2 parents 7e29512 + 52c58c1
commit 5c9625e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/...in/java/com/google/cloud/teleport/v2/elasticsearch/templates/BigQueryToElasticsearch.java b/...in/java/com/google/cloud/teleport/v2/elasticsearch/templates/BigQueryToElasticsearch.java
@@ -178,6 +178,7 @@ private static PipelineResult run(BigQueryToElasticsearchOptions options) {
     udfOut.apply(
         "WriteToElasticsearch",
         WriteToElasticsearch.newBuilder()
+            .setUserAgent("dataflow-bigquery-to-elasticsearch-template/v2")
             .setOptions(options.as(BigQueryToElasticsearchOptions.class))
             .build());
 

diff --git a/...rc/main/java/com/google/cloud/teleport/v2/elasticsearch/templates/GCSToElasticsearch.java b/...rc/main/java/com/google/cloud/teleport/v2/elasticsearch/templates/GCSToElasticsearch.java
@@ -227,8 +227,8 @@ private static PipelineResult run(GCSToElasticsearchOptions options) {
           .setPythonUdfFunctionName(options.getPythonExternalTextTransformFunctionName());
     } else {
       lineToFailsafeJsonBuilder
-          .setJavascriptUdfFileSystemPath(options.getPythonExternalTextTransformGcsPath())
-          .setJavascriptUdfFunctionName(options.getPythonExternalTextTransformFunctionName());
+          .setJavascriptUdfFileSystemPath(options.getJavascriptTextTransformGcsPath())
+          .setJavascriptUdfFunctionName(options.getJavascriptTextTransformFunctionName());
     }
     PCollectionTuple convertedCsvLines =
         readCsvLines.apply("ConvertLine", lineToFailsafeJsonBuilder.build());