[beam] branch master updated: [BEAM-11271] WgetDataset instead of getTable for query location inference

[pabloem]

This is an automated email from the ASF dual-hosted git repository.

pabloem pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/beam.git


The following commit(s) were added to refs/heads/master by this push:
     new b379644  [BEAM-11271] WgetDataset instead of getTable for query 
location inference
     new 8a62265  Merge pull request #14395 from [BEAM-11271] getDataset 
instead of getTable for query location infer…
b379644 is described below

commit b37964426a5a2fb133a2df7f383b84c24e1b23ed
Author: Pablo Estrada <pabl...@apache.org>
AuthorDate: Wed Mar 31 14:30:26 2021 -0700

    [BEAM-11271] WgetDataset instead of getTable for query location inference
---
 .../org/apache/beam/sdk/io/gcp/bigquery/BigQueryQueryHelper.java     | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git 
a/sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigquery/BigQueryQueryHelper.java
 
b/sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigquery/BigQueryQueryHelper.java
index 4193ba6..fc1fe94 100644
--- 
a/sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigquery/BigQueryQueryHelper.java
+++ 
b/sdks/java/io/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigquery/BigQueryQueryHelper.java
@@ -109,7 +109,10 @@ class BigQueryQueryHelper {
               .getReferencedTables();
       if (referencedTables != null && !referencedTables.isEmpty()) {
         TableReference referencedTable = referencedTables.get(0);
-        effectiveLocation = 
tableService.getTable(referencedTable).getLocation();
+        effectiveLocation =
+            tableService
+                .getDataset(referencedTable.getProjectId(), 
referencedTable.getDatasetId())
+                .getLocation();
       }
     }