Skip to content
This repository was archived by the owner on Mar 23, 2026. It is now read-only.

Commit cb09a0a

Browse files
committed
chore: add missing default values
1 parent 736f9ed commit cb09a0a

4 files changed

Lines changed: 17 additions & 18 deletions

File tree

google-cloud-bigquery-jdbc/src/main/java/com/google/cloud/bigquery/jdbc/BigQueryConnection.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@
4040
import com.google.cloud.bigquery.storage.v1.BigQueryWriteClient;
4141
import com.google.cloud.bigquery.storage.v1.BigQueryWriteSettings;
4242
import com.google.cloud.http.HttpTransportOptions;
43+
import com.google.common.base.Splitter;
4344
import java.io.IOException;
4445
import java.io.InputStream;
4546
import java.sql.CallableStatement;
@@ -52,6 +53,7 @@
5253
import java.sql.Statement;
5354
import java.time.Duration;
5455
import java.util.ArrayList;
56+
import java.util.Collections;
5557
import java.util.ConcurrentModificationException;
5658
import java.util.List;
5759
import java.util.Map;
@@ -244,8 +246,8 @@ public class BigQueryConnection extends BigQueryNoOpsConnection {
244246
String additionalProjectsStr = ds.getAdditionalProjects();
245247
this.additionalProjects =
246248
(additionalProjectsStr == null || additionalProjectsStr.isEmpty())
247-
? java.util.Collections.emptyList()
248-
: java.util.Arrays.asList(additionalProjectsStr.split(","));
249+
? Collections.emptyList()
250+
: Splitter.on(',').trimResults().omitEmptyStrings().splitToList(additionalProjectsStr);
249251

250252
this.filterTablesOnDefaultDataset = ds.getFilterTablesOnDefaultDataset();
251253
this.requestGoogleDriveScope = ds.getRequestGoogleDriveScope();

google-cloud-bigquery-jdbc/src/main/java/com/google/cloud/bigquery/jdbc/BigQueryJdbcUrlUtility.java

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,10 @@ protected boolean removeEldestEntry(Map.Entry<String, Map<String, String>> eldes
134134
static final String BYOID_POOL_USER_PROJECT_PROPERTY_NAME = "BYOID_PoolUserProject";
135135
static final String BYOID_SA_IMPERSONATION_URI_PROPERTY_NAME = "BYOID_SA_Impersonation_Uri";
136136
static final String BYOID_SUBJECT_TOKEN_TYPE_PROPERTY_NAME = "BYOID_SubjectTokenType";
137+
static final String DEFAULT_BYOID_SUBJECT_TOKEN_TYPE_VALUE =
138+
"urn:ietf:params:oauth:tokentype:id_token";
137139
static final String BYOID_TOKEN_URI_PROPERTY_NAME = "BYOID_TokenUri";
140+
static final String DEFAULT_BYOID_TOKEN_URI_VALUE = "https://sts.googleapis.com/v1/token";
138141
static final String PARTNER_TOKEN_PROPERTY_NAME = "PartnerToken";
139142
private static final Pattern PARTNER_TOKEN_PATTERN =
140143
Pattern.compile(
@@ -513,7 +516,7 @@ protected boolean removeEldestEntry(Map.Entry<String, Map<String, String>> eldes
513516
+ " Corresponds\n"
514517
+ " to the subject_token_type property in the external account"
515518
+ " configuration file.")
516-
.setDefaultValue("urn:ietf:params:oauth:tokentype:id_token")
519+
.setDefaultValue(DEFAULT_BYOID_SUBJECT_TOKEN_TYPE_VALUE)
517520
.build(),
518521
BigQueryConnectionProperty.newBuilder()
519522
.setName(BYOID_TOKEN_URI_PROPERTY_NAME)
@@ -523,7 +526,7 @@ protected boolean removeEldestEntry(Map.Entry<String, Map<String, String>> eldes
523526
+ " tokens. Corresponds to the token_url property in the external"
524527
+ " account\n"
525528
+ " configuration file.")
526-
.setDefaultValue("https://sts.googleapis.com/v1/token")
529+
.setDefaultValue(DEFAULT_BYOID_TOKEN_URI_VALUE)
527530
.build(),
528531
BigQueryConnectionProperty.newBuilder()
529532
.setName(PARTNER_TOKEN_PROPERTY_NAME)
@@ -839,7 +842,7 @@ static Map<String, String> parsePropertiesMapFromValue(
839842
}
840843
Map<String, String> propertiesMap = new HashMap<>();
841844
for (String keyValuePair : Splitter.on(",").split(propertiesString)) {
842-
List<String> parts = Splitter.on("=").splitToList(keyValuePair);
845+
List<String> parts = Splitter.on("=").limit(2).splitToList(keyValuePair);
843846
if (parts.size() == 2) {
844847
propertiesMap.put(parts.get(0), parts.get(1));
845848
} else {

google-cloud-bigquery-jdbc/src/main/java/com/google/cloud/bigquery/jdbc/DataSource.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1182,15 +1182,19 @@ public void setByoidSAImpersonationUri(String byoidSAImpersonationUri) {
11821182
}
11831183

11841184
public String getByoidSubjectTokenType() {
1185-
return byoidSubjectTokenType;
1185+
return byoidSubjectTokenType != null
1186+
? byoidSubjectTokenType
1187+
: BigQueryJdbcUrlUtility.DEFAULT_BYOID_SUBJECT_TOKEN_TYPE_VALUE;
11861188
}
11871189

11881190
public void setByoidSubjectTokenType(String byoidSubjectTokenType) {
11891191
this.byoidSubjectTokenType = byoidSubjectTokenType;
11901192
}
11911193

11921194
public String getByoidTokenUri() {
1193-
return byoidTokenUri;
1195+
return byoidTokenUri != null
1196+
? byoidTokenUri
1197+
: BigQueryJdbcUrlUtility.DEFAULT_BYOID_TOKEN_URI_VALUE;
11941198
}
11951199

11961200
public void setByoidTokenUri(String byoidTokenUri) {

google-cloud-bigquery-jdbc/src/main/java/com/google/cloud/bigquery/jdbc/PooledConnectionDataSource.java

Lines changed: 1 addition & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -40,17 +40,7 @@ public PooledConnection getPooledConnection() throws SQLException {
4040
throw new BigQueryJdbcRuntimeException(
4141
"Cannot get pooled connection: unable to get underlying physical connection");
4242
}
43-
Long connectionPoolSize = this.getConnectionPoolSize();
44-
if (connectionPoolSize != null
45-
&& connectionPoolSize == BigQueryJdbcUrlUtility.DEFAULT_CONNECTION_POOL_SIZE_VALUE) {
46-
String connectionURl = ((BigQueryConnection) bqConnection).getConnectionUrl();
47-
String parsedValue =
48-
BigQueryJdbcUrlUtility.parseUriProperty(
49-
connectionURl, BigQueryJdbcUrlUtility.CONNECTION_POOL_SIZE_PROPERTY_NAME);
50-
if (parsedValue != null) {
51-
connectionPoolSize = Long.parseLong(parsedValue);
52-
}
53-
}
43+
Long connectionPoolSize = ((BigQueryConnection) bqConnection).getConnectionPoolSize();
5444
if (connectionPoolManager == null) {
5545
connectionPoolManager = new PooledConnectionListener(connectionPoolSize);
5646
}

0 commit comments

Comments
 (0)