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

Commit b621d3b

Browse files
committed
chore: address gemini pr review comments
1 parent ce652cf commit b621d3b

3 files changed

Lines changed: 31 additions & 13 deletions

File tree

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -277,13 +277,13 @@ public class BigQueryConnection extends BigQueryNoOpsConnection {
277277
BigQueryJdbcUrlUtility.parseIntProperty(
278278
url,
279279
BigQueryJdbcUrlUtility.HTTP_CONNECT_TIMEOUT_PROPERTY_NAME,
280-
BigQueryJdbcUrlUtility.DEFAULT_HTTP_CONNECT_TIMEOUT_VALUE,
280+
null,
281281
this.connectionClassName);
282282
this.httpReadTimeout =
283283
BigQueryJdbcUrlUtility.parseIntProperty(
284284
url,
285285
BigQueryJdbcUrlUtility.HTTP_READ_TIMEOUT_PROPERTY_NAME,
286-
BigQueryJdbcUrlUtility.DEFAULT_HTTP_READ_TIMEOUT_VALUE,
286+
null,
287287
this.connectionClassName);
288288
this.httpTransportOptions =
289289
BigQueryJdbcProxyUtility.getHttpTransportOptions(

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

Lines changed: 16 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -123,26 +123,31 @@ static HttpTransportOptions getHttpTransportOptions(
123123
String callerClassName) {
124124
LOG.finest("++enter++\t" + callerClassName);
125125

126-
if (!proxyProperties.containsKey(BigQueryJdbcUrlUtility.PROXY_HOST_PROPERTY_NAME)
127-
&& sslTrustStorePath == null
128-
&& connectTimeout == null
129-
&& readTimeout == null) {
126+
boolean hasProxyOrSsl =
127+
proxyProperties.containsKey(BigQueryJdbcUrlUtility.PROXY_HOST_PROPERTY_NAME)
128+
|| sslTrustStorePath != null;
129+
boolean hasTimeoutConfig = connectTimeout != null || readTimeout != null;
130+
131+
if (!hasProxyOrSsl && !hasTimeoutConfig) {
130132
return null;
131133
}
132134

133135
HttpTransportOptions.Builder httpTransportOptionsBuilder = HttpTransportOptions.newBuilder();
134-
if (proxyProperties.containsKey(BigQueryJdbcUrlUtility.PROXY_HOST_PROPERTY_NAME)
135-
|| sslTrustStorePath != null) {
136+
if (hasProxyOrSsl) {
136137
httpTransportOptionsBuilder.setHttpTransportFactory(
137138
getHttpTransportFactory(
138139
proxyProperties, sslTrustStorePath, sslTrustStorePassword, callerClassName));
139140
}
140141

141-
if (connectTimeout != null) {
142-
httpTransportOptionsBuilder.setConnectTimeout(connectTimeout);
143-
}
144-
if (readTimeout != null) {
145-
httpTransportOptionsBuilder.setReadTimeout(readTimeout);
142+
if (hasTimeoutConfig) {
143+
httpTransportOptionsBuilder.setConnectTimeout(
144+
connectTimeout != null
145+
? connectTimeout
146+
: BigQueryJdbcUrlUtility.DEFAULT_HTTP_CONNECT_TIMEOUT_VALUE);
147+
httpTransportOptionsBuilder.setReadTimeout(
148+
readTimeout != null
149+
? readTimeout
150+
: BigQueryJdbcUrlUtility.DEFAULT_HTTP_READ_TIMEOUT_VALUE);
146151
}
147152

148153
return httpTransportOptionsBuilder.build();

google-cloud-bigquery-jdbc/src/test/java/com/google/cloud/bigquery/jdbc/BigQueryJdbcProxyUtilityTest.java

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -323,5 +323,18 @@ public void testGetHttpTransportOptions_withConnectTimeoutOnly_returnsOptions()
323323
Collections.<String, String>emptyMap(), null, null, 10000, null, "TestClass");
324324
assertNotNull(options);
325325
assertThat(options.getConnectTimeout()).isEqualTo(10000);
326+
assertThat(options.getReadTimeout())
327+
.isEqualTo(BigQueryJdbcUrlUtility.DEFAULT_HTTP_READ_TIMEOUT_VALUE);
328+
}
329+
330+
@Test
331+
public void testGetHttpTransportOptions_withReadTimeoutOnly_returnsOptions() {
332+
HttpTransportOptions options =
333+
BigQueryJdbcProxyUtility.getHttpTransportOptions(
334+
Collections.<String, String>emptyMap(), null, null, null, 20000, "TestClass");
335+
assertNotNull(options);
336+
assertThat(options.getReadTimeout()).isEqualTo(20000);
337+
assertThat(options.getConnectTimeout())
338+
.isEqualTo(BigQueryJdbcUrlUtility.DEFAULT_HTTP_CONNECT_TIMEOUT_VALUE);
326339
}
327340
}

0 commit comments

Comments
 (0)