Skip to content

fix(bigquery-jdbc): pass connection proxy settings to OpenTelemetry exporters #14011

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
keshavdandeva merged 11 commits into from
Aug 12, 2026
Merged

fix(bigquery-jdbc): pass connection proxy settings to OpenTelemetry exporters #14011

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
41f8daa
fix(bigquery-jdbc): pass connection proxy settings to OpenTelemetry t…
keshavdandeva Aug 7, 2026
5dbb7e9
add debug statemetns
keshavdandeva Aug 7, 2026
9712fa9
revert debug statements
keshavdandeva Aug 7, 2026
ece6c0b
add proxy test for trace exporter
keshavdandeva Aug 8, 2026
dacab8c
test(bigquery-jdbc): clean up integration tests and finalize trace pr…
keshavdandeva Aug 8, 2026
5fc36cf
Merge branch 'main' into jdbc/otel-proxy-fix
keshavdandeva Aug 8, 2026
e9e6b2b
fix(bigquery-jdbc): address PR review comments for createProxyOptions
keshavdandeva Aug 8, 2026
013bed9
use ProxyOptions.create(InetSocketAddress) directly
keshavdandeva Aug 12, 2026
560f11c
add proxy env flag to otel it
keshavdandeva Aug 12, 2026
2a7210a
pr feedback
keshavdandeva Aug 12, 2026
df0f46d
Merge branch 'main' into jdbc/otel-proxy-fix
keshavdandeva Aug 12, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 11 additions & 3 deletions java-bigquery-jdbc/src/main/java/com/google/cloud/bigquery/jdbc/BigQueryConnection.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -200,6 +200,7 @@ public class BigQueryConnection extends BigQueryNoOpsConnection {
List<ConnectionProperty> queryProperties;
Map<String, String> authProperties;
Map<String, String> overrideProperties;
Map<String, String> proxyProperties;
Credentials credentials;
boolean useStatelessQueryMode;
int numBufferedRows;
Expand Down Expand Up @@ -299,7 +300,7 @@ public class BigQueryConnection extends BigQueryNoOpsConnection {
String.valueOf(ds.getRequestGoogleDriveScope()),
BigQueryJdbcUrlUtility.REQUEST_GOOGLE_DRIVE_SCOPE_PROPERTY_NAME);

Map<String, String> proxyProperties =
this.proxyProperties =
BigQueryJdbcProxyUtility.parseProxyProperties(ds, this.connectionClassName);

this.sslTrustStorePath = ds.getSSLTrustStorePath();
Expand Down Expand Up @@ -1204,14 +1205,21 @@ private OpenTelemetry getOpenTelemetryInstance() {
this.customOpenTelemetry,
this.gcpTelemetryCredentials,
effectiveProjectId,
this.credentials);
this.credentials,
this.proxyProperties);

boolean hasExternalOtel = this.customOpenTelemetry != null || this.useGlobalOpenTelemetry;
Logging localLoggingClient = null;
if (this.enableGcpLogExporter && !hasExternalOtel) {
localLoggingClient =
BigQueryJdbcOpenTelemetry.createLoggingClient(
true, null, this.gcpTelemetryCredentials, effectiveProjectId, this.credentials);
true,
null,
this.gcpTelemetryCredentials,
effectiveProjectId,
this.credentials,
this.httpTransportOptions,
this.headerProvider);
}

if (this.enableGcpLogExporter || hasExternalOtel) {
Expand Down
64 changes: 57 additions & 7 deletions ...bigquery-jdbc/src/main/java/com/google/cloud/bigquery/jdbc/BigQueryJdbcOpenTelemetry.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,9 +16,11 @@

package com.google.cloud.bigquery.jdbc;

import com.google.api.gax.rpc.HeaderProvider;
import com.google.auth.Credentials;
import com.google.auth.oauth2.GoogleCredentials;
import com.google.cloud.bigquery.exception.BigQueryJdbcRuntimeException;
import com.google.cloud.http.HttpTransportOptions;
import com.google.cloud.logging.Logging;
import com.google.cloud.logging.LoggingOptions;
import com.google.common.hash.Hashing;
Expand All @@ -32,9 +34,16 @@
import io.opentelemetry.context.Context;
import io.opentelemetry.context.Scope;
import io.opentelemetry.exporter.otlp.http.trace.OtlpHttpSpanExporter;
import io.opentelemetry.exporter.otlp.http.trace.OtlpHttpSpanExporterBuilder;
import io.opentelemetry.exporter.otlp.trace.OtlpGrpcSpanExporter;
import io.opentelemetry.sdk.OpenTelemetrySdk;
import io.opentelemetry.sdk.autoconfigure.AutoConfiguredOpenTelemetrySdk;
import io.opentelemetry.sdk.common.export.ProxyOptions;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.Proxy;
import java.net.ProxySelector;
import java.net.SocketAddress;
import java.net.URI;
import java.nio.charset.StandardCharsets;
import java.sql.SQLException;
Expand Down Expand Up @@ -181,7 +190,9 @@ public static Logging createLoggingClient(
OpenTelemetry customOpenTelemetry,
String effectiveCredentials,
String effectiveProjectId,
Credentials fallbackCredentials) {
Credentials fallbackCredentials,
HttpTransportOptions httpTransportOptions,
HeaderProvider headerProvider) {

if (!enableGcpLogExporter || customOpenTelemetry != null) {
return null;
Expand All @@ -200,6 +211,12 @@ public static Logging createLoggingClient(
if (credentials != null) {
loggingOptionsBuilder.setCredentials(credentials);
}
if (httpTransportOptions != null) {
loggingOptionsBuilder.setTransportOptions(httpTransportOptions);
}
if (headerProvider != null) {
loggingOptionsBuilder.setHeaderProvider(headerProvider);
}
return loggingOptionsBuilder.build().getService();
} catch (Exception e) {
throw new BigQueryJdbcRuntimeException("Failed to initialize Logging client", e);
Expand Down Expand Up @@ -317,7 +334,8 @@ public static OpenTelemetry getOpenTelemetry(
OpenTelemetry customOpenTelemetry,
String gcpTelemetryCredentials,
String gcpTelemetryProjectId,
Credentials fallbackCredentials) {
Credentials fallbackCredentials,
Map<String, String> proxyProperties) {

if (customOpenTelemetry != null) {
return customOpenTelemetry;
Expand Down Expand Up @@ -415,11 +433,17 @@ public static OpenTelemetry getOpenTelemetry(
final Credentials finalCredentials = credentials;

if (spanExporter instanceof OtlpHttpSpanExporter) {
return ((OtlpHttpSpanExporter) spanExporter)
.toBuilder()
.setHeaders(
() -> getAuthHeaders(finalCredentials, gcpTelemetryProjectId))
.build();
OtlpHttpSpanExporterBuilder builder =
((OtlpHttpSpanExporter) spanExporter).toBuilder();
builder.setHeaders(
() -> getAuthHeaders(finalCredentials, gcpTelemetryProjectId));

ProxyOptions proxyOptions = createProxyOptions(proxyProperties);
if (proxyOptions != null) {
builder.setProxy(proxyOptions);
}

return builder.build();
}
if (spanExporter instanceof OtlpGrpcSpanExporter) {
return ((OtlpGrpcSpanExporter) spanExporter)
Expand Down Expand Up @@ -506,4 +530,30 @@ public static <T> T withTracing(
span.end();
}
}

private static ProxyOptions createProxyOptions(Map<String, String> proxyProperties) {
Comment thread
logachev marked this conversation as resolved.
if (proxyProperties == null
|| !proxyProperties.containsKey(BigQueryJdbcUrlUtility.PROXY_HOST_PROPERTY_NAME)
|| !proxyProperties.containsKey(BigQueryJdbcUrlUtility.PROXY_PORT_PROPERTY_NAME)) {
return null;
}

final String host = proxyProperties.get(BigQueryJdbcUrlUtility.PROXY_HOST_PROPERTY_NAME);
final int port =
Integer.parseInt(proxyProperties.get(BigQueryJdbcUrlUtility.PROXY_PORT_PROPERTY_NAME));

ProxySelector proxySelector =
Comment thread
logachev marked this conversation as resolved.
Outdated
new ProxySelector() {
@Override
public List<Proxy> select(URI uri) {
return Collections.singletonList(
new Proxy(Proxy.Type.HTTP, new InetSocketAddress(host, port)));
}

@Override
public void connectFailed(URI uri, SocketAddress sa, IOException ioe) {}
};
Comment thread
keshavdandeva marked this conversation as resolved.
Outdated

return ProxyOptions.create(proxySelector);
}
Comment thread
keshavdandeva marked this conversation as resolved.
}
3 changes: 2 additions & 1 deletion java-bigquery-jdbc/src/test/java/com/google/cloud/bigquery/jdbc/BigQueryConnectionTest.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -620,7 +620,7 @@ public void testOpenTelemetryPrecedenceHierarchy(
.when(
() ->
BigQueryJdbcOpenTelemetry.createLoggingClient(
anyBoolean(), any(), any(), any(), any()))
anyBoolean(), any(), any(), any(), any(), any(), any()))
.thenReturn(mockLogging);

// Stub getOpenTelemetry to return the expected mock based on inputs
Expand All @@ -634,6 +634,7 @@ public void testOpenTelemetryPrecedenceHierarchy(
hasCustom ? eq(mockCustomOtel) : isNull(),
any(),
any(),
any(),
any()))
.thenAnswer(
invocation -> {
Expand Down
28 changes: 16 additions & 12 deletions ...uery-jdbc/src/test/java/com/google/cloud/bigquery/jdbc/BigQueryJdbcOpenTelemetryTest.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ public void testGetOpenTelemetry_withCustomSdk_returnsCustom() {

OpenTelemetry result =
BigQueryJdbcOpenTelemetry.getOpenTelemetry(
false, false, false, mockCustomOtel, null, null, null);
false, false, false, mockCustomOtel, null, null, null, null);

assertThat(result).isSameInstanceAs(mockCustomOtel);
}
Expand All @@ -64,15 +64,16 @@ public void testGetOpenTelemetry_withCustomSdkAndFlags_returnsCustom() {
// Custom SDK always takes precedence over individual flags
OpenTelemetry result =
BigQueryJdbcOpenTelemetry.getOpenTelemetry(
false, true, true, mockCustomOtel, null, null, null);
false, true, true, mockCustomOtel, null, null, null, null);

assertThat(result).isSameInstanceAs(mockCustomOtel);
}

@Test
public void testGetOpenTelemetry_noFlags_returnsNoop() {
OpenTelemetry result =
BigQueryJdbcOpenTelemetry.getOpenTelemetry(false, false, false, null, null, null, null);
BigQueryJdbcOpenTelemetry.getOpenTelemetry(
false, false, false, null, null, null, null, null);

assertThat(result).isSameInstanceAs(OpenTelemetry.noop());
}
Expand All @@ -88,10 +89,10 @@ public void testGetTracer_respectsScopeName() {
public void testGetOpenTelemetry_cachesSdkInstances() {
OpenTelemetry result1 =
BigQueryJdbcOpenTelemetry.getOpenTelemetry(
false, true, false, null, null, "project1", null);
false, true, false, null, null, "project1", null, null);
OpenTelemetry result2 =
BigQueryJdbcOpenTelemetry.getOpenTelemetry(
false, true, false, null, null, "project1", null);
false, true, false, null, null, "project1", null, null);

assertThat(result1).isSameInstanceAs(result2);
}
Expand All @@ -100,40 +101,43 @@ public void testGetOpenTelemetry_cachesSdkInstances() {
public void testGetOpenTelemetry_createsNewInstanceForDifferentKey() {
OpenTelemetry result1 =
BigQueryJdbcOpenTelemetry.getOpenTelemetry(
false, true, false, null, null, "project1", null);
false, true, false, null, null, "project1", null, null);
OpenTelemetry result2 =
BigQueryJdbcOpenTelemetry.getOpenTelemetry(
false, true, false, null, null, "project2", null);
false, true, false, null, null, "project2", null, null);

assertThat(result1).isNotSameInstanceAs(result2);
}

@Test
public void testGetOpenTelemetry_createsNewInstanceForDifferentTraceFlag() {
OpenTelemetry result1 =
BigQueryJdbcOpenTelemetry.getOpenTelemetry(false, true, true, null, null, "project1", null);
BigQueryJdbcOpenTelemetry.getOpenTelemetry(
false, true, true, null, null, "project1", null, null);
OpenTelemetry result2 =
BigQueryJdbcOpenTelemetry.getOpenTelemetry(
false, false, true, null, null, "project1", null);
false, false, true, null, null, "project1", null, null);

assertThat(result1).isNotSameInstanceAs(result2);
}

@Test
public void testGetOpenTelemetry_ignoresEnableLogFlagInCacheKey() {
OpenTelemetry result1 =
BigQueryJdbcOpenTelemetry.getOpenTelemetry(false, true, true, null, null, "project1", null);
BigQueryJdbcOpenTelemetry.getOpenTelemetry(
false, true, true, null, null, "project1", null, null);
OpenTelemetry result2 =
BigQueryJdbcOpenTelemetry.getOpenTelemetry(
false, true, false, null, null, "project1", null);
false, true, false, null, null, "project1", null, null);

assertThat(result1).isSameInstanceAs(result2);
}

@Test
public void testGetOpenTelemetry_withUseGlobalOTel_returnsGlobal() {
OpenTelemetry result =
BigQueryJdbcOpenTelemetry.getOpenTelemetry(true, false, false, null, null, null, null);
BigQueryJdbcOpenTelemetry.getOpenTelemetry(
true, false, false, null, null, null, null, null);

assertThat(result).isSameInstanceAs(GlobalOpenTelemetry.get());
}
Expand Down
Loading