fix: Use specified bigQueryWriteClient when creating JsonStreamWriter

Fixes: #2678.
GoogleCloudPlatform · Mar 19, 2024 · 1ba9e74 · 1ba9e74
1 parent adb57c6
commit 1ba9e74
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...-bigquery/src/main/java/com/google/cloud/spring/bigquery/core/BigQueryJsonDataWriter.java b/...-bigquery/src/main/java/com/google/cloud/spring/bigquery/core/BigQueryJsonDataWriter.java
@@ -78,7 +78,7 @@ public class BigQueryJsonDataWriter implements AutoCloseable {
     // For more information about JsonStreamWriter, see:
     // https://googleapis.dev/java/google-cloud-bigquerystorage/latest/com/google/cloud/bigquery/storage/v1beta2/JsonStreamWriter.html
     streamWriter =
-        JsonStreamWriter.newBuilder(writeStream.getName(), writeStream.getTableSchema()).build();
+        JsonStreamWriter.newBuilder(writeStream.getName(), writeStream.getTableSchema(), bigQueryWriteClient).build();
     this.bigQueryWriteClient = bigQueryWriteClient;
   }