4040import com .google .cloud .trace .v1 .TraceServiceClient ;
4141import com .google .cloud .trace .v1 .TraceServiceSettings ;
4242import com .google .devtools .cloudtrace .v1 .Trace ;
43- import io .opentelemetry .api .GlobalOpenTelemetry ;
4443import io .opentelemetry .api .trace .Span ;
4544import io .opentelemetry .api .trace .Tracer ;
4645import io .opentelemetry .context .Scope ;
4746import java .io .IOException ;
4847import java .util .Arrays ;
4948import java .util .Collection ;
5049import org .junit .AfterClass ;
51- import org .junit .Before ;
5250import org .junit .BeforeClass ;
5351import org .junit .ClassRule ;
5452import 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