From 1ba9e7472788623746ace322754e588c593bfc89 Mon Sep 17 00:00:00 2001
From: Mike Eltsufin <meltsufin@google.com>
Date: Mon, 18 Mar 2024 21:49:20 -0400
Subject: [PATCH] fix: Use specified bigQueryWriteClient when creating
 JsonStreamWriter

Fixes: #2678.
---
 .../cloud/spring/bigquery/core/BigQueryJsonDataWriter.java      | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/spring-cloud-gcp-bigquery/src/main/java/com/google/cloud/spring/bigquery/core/BigQueryJsonDataWriter.java b/spring-cloud-gcp-bigquery/src/main/java/com/google/cloud/spring/bigquery/core/BigQueryJsonDataWriter.java
index 11bb29f898..28c9df93e0 100644
--- a/spring-cloud-gcp-bigquery/src/main/java/com/google/cloud/spring/bigquery/core/BigQueryJsonDataWriter.java
+++ b/spring-cloud-gcp-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;
   }