Skip to content

Commit 9c1cd08

Browse files
committed
Remove update to GlobalTelemetry. Use opentelemetrysdk created in test.
1 parent 74b901d commit 9c1cd08

File tree

2 files changed

+12
-21
lines changed

2 files changed

+12
-21
lines changed

google-cloud-spanner/src/test/java/com/google/cloud/spanner/IntegrationTestEnv.java

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,6 @@
2929
import com.google.cloud.spanner.testing.RemoteSpannerHelper;
3030
import com.google.common.collect.Iterators;
3131
import com.google.spanner.admin.instance.v1.CreateInstanceMetadata;
32-
import io.opentelemetry.api.GlobalOpenTelemetry;
3332
import io.opentelemetry.api.trace.propagation.W3CTraceContextPropagator;
3433
import io.opentelemetry.context.propagation.ContextPropagators;
3534
import io.opentelemetry.sdk.OpenTelemetrySdk;
@@ -167,7 +166,6 @@ protected void before() throws Throwable {
167166
}
168167

169168
public SpannerOptions spannerOptionsWithEndToEndTracing(SpannerOptions options) {
170-
GlobalOpenTelemetry.resetForTest(); // reset global context for test
171169
assumeFalse("This test requires credentials", EmulatorSpannerHelper.isUsingEmulator());
172170

173171
TraceConfiguration.Builder traceConfigurationBuilder = TraceConfiguration.builder();
@@ -178,8 +176,7 @@ public SpannerOptions spannerOptionsWithEndToEndTracing(SpannerOptions options)
178176
TraceExporter.createWithConfiguration(
179177
traceConfigurationBuilder.setProjectId(options.getProjectId()).build());
180178

181-
String serviceName =
182-
"java-spanner-jdbc-integration-tests-" + ThreadLocalRandom.current().nextInt();
179+
String serviceName = "java-spanner-integration-tests-" + ThreadLocalRandom.current().nextInt();
183180
SdkTracerProvider sdkTracerProvider =
184181
SdkTracerProvider.builder()
185182
// Always sample in this test to ensure we know what we get.
@@ -191,7 +188,7 @@ public SpannerOptions spannerOptionsWithEndToEndTracing(SpannerOptions options)
191188
OpenTelemetrySdk.builder()
192189
.setTracerProvider(sdkTracerProvider)
193190
.setPropagators(ContextPropagators.create(W3CTraceContextPropagator.getInstance()))
194-
.buildAndRegisterGlobal();
191+
.build();
195192
SpannerOptions.enableOpenTelemetryTraces();
196193
return options
197194
.toBuilder()

google-cloud-spanner/src/test/java/com/google/cloud/spanner/it/ITEndToEndTracingTest.java

Lines changed: 10 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -40,15 +40,13 @@
4040
import com.google.cloud.trace.v1.TraceServiceClient;
4141
import com.google.cloud.trace.v1.TraceServiceSettings;
4242
import com.google.devtools.cloudtrace.v1.Trace;
43-
import io.opentelemetry.api.GlobalOpenTelemetry;
4443
import io.opentelemetry.api.trace.Span;
4544
import io.opentelemetry.api.trace.Tracer;
4645
import io.opentelemetry.context.Scope;
4746
import java.io.IOException;
4847
import java.util.Arrays;
4948
import java.util.Collection;
5049
import org.junit.AfterClass;
51-
import org.junit.Before;
5250
import org.junit.BeforeClass;
5351
import org.junit.ClassRule;
5452
import org.junit.Test;
@@ -70,7 +68,7 @@ public class ITEndToEndTracingTest {
7068
SpannerOptions.enableOpenTelemetryTraces();
7169
}
7270

73-
private String selectValueQuery;
71+
private static String selectValueQuery = "SELECT @p1 + @p1";
7472

7573
@BeforeClass
7674
public static void setUp() {
@@ -88,11 +86,6 @@ public static void teardown() {
8886
ConnectionOptions.closeSpanner();
8987
}
9088

91-
@Before
92-
public void initSelectValueQuery() {
93-
selectValueQuery = "SELECT @p1 + @p1 ";
94-
}
95-
9689
private void assertTrace(String traceId) throws IOException, InterruptedException {
9790
TraceServiceSettings settings =
9891
env.getTestHelper().getOptions().getCredentials() == null
@@ -104,18 +97,15 @@ private void assertTrace(String traceId) throws IOException, InterruptedExceptio
10497
.build();
10598
try (TraceServiceClient client = TraceServiceClient.create(settings)) {
10699
// It can take a few seconds before the trace is visible.
107-
Thread.sleep(15000);
100+
Thread.sleep(10000);
108101
boolean foundTrace = false;
109102
for (int attempts = 0; attempts < 2; attempts++) {
110103
try {
111-
Trace clientTrace =
112-
client.getTrace(env.getTestHelper().getInstanceId().getProject(), traceId);
104+
Trace trace = client.getTrace(env.getTestHelper().getInstanceId().getProject(), traceId);
113105
// Assert Spanner Frontend Trace is present
114106
assertTrue(
115-
clientTrace.getSpansList().stream()
116-
.anyMatch(
117-
span ->
118-
"Spanner.ExecuteStreamingSql".equals(span.getName())));
107+
trace.getSpansList().stream()
108+
.anyMatch(span -> "Spanner.ExecuteStreamingSql".equals(span.getName())));
119109
foundTrace = true;
120110
break;
121111
} catch (ApiException apiException) {
@@ -149,7 +139,11 @@ private Struct executeWithRowResultType(Statement statement, Type expectedRowTyp
149139

150140
@Test
151141
public void simpleSelect() throws IOException, InterruptedException {
152-
Tracer tracer = GlobalOpenTelemetry.getTracer(ITEndToEndTracingTest.class.getName());
142+
Tracer tracer =
143+
env.getTestHelper()
144+
.getOptions()
145+
.getOpenTelemetry()
146+
.getTracer(ITEndToEndTracingTest.class.getName());
153147
Span span = tracer.spanBuilder("simpleSelect").startSpan();
154148
Scope scope = span.makeCurrent();
155149
Type rowType = Type.struct(StructField.of("", Type.int64()));

0 commit comments

Comments
 (0)