@@ -1030,9 +1030,9 @@ public class ITBigQueryTest {
10301030
10311031 private static BigQuery bigquery ;
10321032 private static Storage storage ;
1033+ private static OpenTelemetry otel ;
10331034
1034- private static class CustomSpanExporter
1035- implements io .opentelemetry .sdk .trace .export .SpanExporter {
1035+ private static class TestSpanExporter implements io .opentelemetry .sdk .trace .export .SpanExporter {
10361036 @ Override
10371037 public CompletableResultCode export (Collection <SpanData > collection ) {
10381038 if (collection .isEmpty ()) {
@@ -1064,6 +1064,13 @@ public static void beforeClass() throws InterruptedException, IOException {
10641064 RemoteBigQueryHelper bigqueryHelper = RemoteBigQueryHelper .create ();
10651065 RemoteStorageHelper storageHelper = RemoteStorageHelper .create ();
10661066 Map <String , String > labels = ImmutableMap .of ("test-job-name" , "test-load-job" );
1067+ SdkTracerProvider tracerProvider =
1068+ SdkTracerProvider .builder ()
1069+ .addSpanProcessor (SimpleSpanProcessor .create (new TestSpanExporter ()))
1070+ .setSampler (Sampler .alwaysOn ())
1071+ .build ();
1072+ otel = OpenTelemetrySdk .builder ().setTracerProvider (tracerProvider ).buildAndRegisterGlobal ();
1073+
10671074 bigquery = bigqueryHelper .getOptions ().getService ();
10681075 storage = storageHelper .getOptions ().getService ();
10691076 storage .create (BucketInfo .of (BUCKET ));
@@ -7537,14 +7544,6 @@ public void testStatementType() throws InterruptedException {
75377544
75387545 @ Test
75397546 public void testOpenTelemetryTracingDatasets () {
7540- SdkTracerProvider tracerProvider =
7541- SdkTracerProvider .builder ()
7542- .addSpanProcessor (SimpleSpanProcessor .create (new CustomSpanExporter ()))
7543- .setSampler (Sampler .alwaysOn ())
7544- .build ();
7545-
7546- OpenTelemetry otel =
7547- OpenTelemetrySdk .builder ().setTracerProvider (tracerProvider ).buildAndRegisterGlobal ();
75487547 Tracer tracer = otel .getTracer ("Test Tracer" );
75497548 BigQueryOptions otelOptions =
75507549 BigQueryOptions .newBuilder ()
@@ -7618,14 +7617,6 @@ public void testOpenTelemetryTracingDatasets() {
76187617
76197618 @ Test
76207619 public void TestOpenTelemetryTracingTables () {
7621- SdkTracerProvider tracerProvider =
7622- SdkTracerProvider .builder ()
7623- .addSpanProcessor (SimpleSpanProcessor .create (new CustomSpanExporter ()))
7624- .setSampler (Sampler .alwaysOn ())
7625- .build ();
7626-
7627- OpenTelemetry otel =
7628- OpenTelemetrySdk .builder ().setTracerProvider (tracerProvider ).buildAndRegisterGlobal ();
76297620 Tracer tracer = otel .getTracer ("Test Tracer" );
76307621 BigQueryOptions otelOptions =
76317622 BigQueryOptions .newBuilder ()
@@ -7670,14 +7661,6 @@ public void TestOpenTelemetryTracingTables() {
76707661
76717662 @ Test
76727663 public void testOpenTelemetryTracingQuery () throws InterruptedException {
7673- SdkTracerProvider tracerProvider =
7674- SdkTracerProvider .builder ()
7675- .addSpanProcessor (SimpleSpanProcessor .create (new CustomSpanExporter ()))
7676- .setSampler (Sampler .alwaysOn ())
7677- .build ();
7678-
7679- OpenTelemetry otel =
7680- OpenTelemetrySdk .builder ().setTracerProvider (tracerProvider ).buildAndRegisterGlobal ();
76817664 Tracer tracer = otel .getTracer ("Test Tracer" );
76827665 BigQueryOptions otelOptions =
76837666 BigQueryOptions .newBuilder ()
0 commit comments