feat: add get location support

googleapis · Sep 26, 2022 · b05f7b0 · b05f7b0
1 parent b3ffd77
commit b05f7b0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/...-bigquerystorage/src/main/java/com/google/cloud/bigquery/storage/v1/JsonStreamWriter.java b/...-bigquerystorage/src/main/java/com/google/cloud/bigquery/storage/v1/JsonStreamWriter.java
@@ -360,6 +360,7 @@ private Builder(String streamOrTableName, TableSchema tableSchema, BigQueryWrite
         WriteStream writeStream = this.client.getWriteStream(writeStreamRequest);
         TableSchema writeStreamTableSchema = writeStream.getTableSchema();
         this.tableSchema = writeStreamTableSchema;
+        this.location = writeStream.getLocation();
       } else {
         this.tableSchema = tableSchema;
       }