From b4a1866be5a5dbdd84e8332a0a233c8f1b49ada2 Mon Sep 17 00:00:00 2001
From: Yiru Tang <yiru@google.com>
Date: Mon, 26 Sep 2022 19:36:28 +0000
Subject: [PATCH] feat: add get location support

---
 .../com/google/cloud/bigquery/storage/v1/JsonStreamWriter.java   | 1 +
 1 file changed, 1 insertion(+)

diff --git a/google-cloud-bigquerystorage/src/main/java/com/google/cloud/bigquery/storage/v1/JsonStreamWriter.java b/google-cloud-bigquerystorage/src/main/java/com/google/cloud/bigquery/storage/v1/JsonStreamWriter.java
index 7078fd6da1..35cfc21707 100644
--- a/google-cloud-bigquerystorage/src/main/java/com/google/cloud/bigquery/storage/v1/JsonStreamWriter.java
+++ b/google-cloud-bigquerystorage/src/main/java/com/google/cloud/bigquery/storage/v1/JsonStreamWriter.java
@@ -407,6 +407,7 @@ private Builder(String streamOrTableName, TableSchema tableSchema, BigQueryWrite
         TableSchema writeStreamTableSchema = writeStream.getTableSchema();
 
         this.tableSchema = writeStreamTableSchema;
+        this.location = writeStream.getLocation();
       } else {
         this.tableSchema = tableSchema;
       }