From 7a76e0ea1a094d64ae32f1abf0b3e7d0c5e8c72e Mon Sep 17 00:00:00 2001
From: Phong Chuong <147636638+PhongChuong@users.noreply.github.com>
Date: Tue, 23 Jan 2024 14:31:56 -0500
Subject: [PATCH 1/4] Universe domain feat
---
.../bigquery/spi/v2/HttpBigQueryRpc.java | 45 ++++++++++-
.../cloud/bigquery/it/ITBigQueryTest.java | 81 +++++++++++++++++++
2 files changed, 124 insertions(+), 2 deletions(-)
diff --git a/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/spi/v2/HttpBigQueryRpc.java b/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/spi/v2/HttpBigQueryRpc.java
index 96a10deb3..841a790ee 100644
--- a/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/spi/v2/HttpBigQueryRpc.java
+++ b/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/spi/v2/HttpBigQueryRpc.java
@@ -105,7 +105,7 @@ public HttpBigQueryRpc(BigQueryOptions options) {
this.options = options;
bigquery =
new Bigquery.Builder(transport, new GsonFactory(), initializer)
- .setRootUrl(options.getHost())
+ .setRootUrl(options.getResolvedApiaryHost("bigquery"))
.setApplicationName(options.getApplicationName())
.build();
}
@@ -114,9 +114,16 @@ private static BigQueryException translate(IOException exception) {
return new BigQueryException(exception);
}
+ private void validateRPC() throws BigQueryException, IOException {
+ if (!this.options.hasValidUniverseDomain()) {
+ throw new BigQueryException(BigQueryException.UNKNOWN_CODE, "Invalid universe domain");
+ }
+ }
+
@Override
public Dataset getDataset(String projectId, String datasetId, Map