File tree Expand file tree Collapse file tree 6 files changed +9
-27
lines changed
sdk-platform-java/gax-java/gax/src
main/java/com/google/api/gax/tracing
test/java/com/google/api/gax/tracing Expand file tree Collapse file tree 6 files changed +9
-27
lines changed Original file line number Diff line number Diff line change 3030
3131package com .google .api .gax .tracing ;
3232
33- import com .google .api .core .BetaApi ;
34- import com .google .api .core .InternalApi ;
3533import com .google .api .gax .logging .LoggerProvider ;
3634import com .google .api .gax .logging .LoggingUtils ;
3735import com .google .common .annotations .VisibleForTesting ;
4442 * An {@link ApiTracer} that logs actionable errors using {@link LoggingUtils} when an RPC attempt
4543 * fails.
4644 */
47- @ BetaApi
48- @ InternalApi
4945class LoggingTracer extends BaseApiTracer {
5046 private static final LoggerProvider LOGGER_PROVIDER =
5147 LoggerProvider .forClazz (LoggingTracer .class );
Original file line number Diff line number Diff line change 3030
3131package com .google .api .gax .tracing ;
3232
33- import com .google .api .core .BetaApi ;
3433import com .google .api .core .InternalApi ;
3534import com .google .api .gax .logging .LoggingUtils ;
3635import com .google .common .annotations .VisibleForTesting ;
4140 * managed automatically by the system, when {@link LoggingUtils#isLoggingEnabled()} returning
4241 * {@code true}.
4342 */
44- @ BetaApi
4543@ InternalApi
4644public class LoggingTracerFactory implements ApiTracerFactory {
4745 private final ApiTracerContext apiTracerContext ;
Original file line number Diff line number Diff line change 2929 */
3030package com .google .api .gax .tracing ;
3131
32- import com .google .api .core .BetaApi ;
33- import com .google .api .core .InternalApi ;
3432import io .opentelemetry .api .OpenTelemetry ;
3533
3634/**
3735 * A {@link ApiTracerFactory} to build instances of {@link OpenTelemetryMetricsTracer}.
3836 *
3937 * <p>This class is expected to be initialized once during client initialization.
4038 */
41- @ BetaApi
42- @ InternalApi
4339public class OpenTelemetryMetricsFactory implements ApiTracerFactory {
4440
4541 private ApiTracerContext clientLevelTracerContext ;
Original file line number Diff line number Diff line change 3131package com .google .api .gax .tracing ;
3232
3333import com .google .api .client .util .Strings ;
34- import com .google .api .core .BetaApi ;
35- import com .google .api .core .InternalApi ;
3634import com .google .api .gax .rpc .LibraryMetadata ;
3735import com .google .common .annotations .VisibleForTesting ;
3836import io .opentelemetry .api .OpenTelemetry ;
4644 *
4745 * <p>This class is expected to be initialized once during client initialization.
4846 */
49- @ BetaApi
50- @ InternalApi
5147public class OpenTelemetryTracingFactory implements ApiTracerFactory {
5248 private final Tracer tracer ;
5349 private final OpenTelemetry openTelemetry ;
Original file line number Diff line number Diff line change 3131package com .google .api .gax .tracing ;
3232
3333import com .google .api .client .util .Strings ;
34- import com .google .api .core .BetaApi ;
3534import com .google .api .core .InternalApi ;
3635import io .opentelemetry .api .trace .Span ;
3736import io .opentelemetry .api .trace .SpanBuilder ;
4241import java .util .concurrent .CancellationException ;
4342
4443/** An implementation of {@link ApiTracer} that uses OpenTelemetry to record traces. */
45- @ BetaApi
46- @ InternalApi
4744class OpenTelemetryTracingTracer implements ApiTracer {
4845
4946 static final String CONTENT_LENGTH_KEY = "Content-Length" ;
Original file line number Diff line number Diff line change @@ -69,17 +69,16 @@ void newTracerWithSpanName_shouldCreateBaseTracer_ifMetricsRecorderIsNull() {
6969
7070 @ Test
7171 void newTracerWithApiTracerContext_shouldMergeApiTracerContext () {
72- ApiTracerContext clientLevelTracerContext = mock ( ApiTracerContext . class , RETURNS_DEEP_STUBS );
73- ApiTracerContext methodLevelTracerContext = mock ( ApiTracerContext . class );
74- when ( clientLevelTracerContext . libraryMetadata (). artifactName ()). thenReturn ( "gax-java" );
75- when ( clientLevelTracerContext . libraryMetadata ().isEmpty ()). thenReturn ( false );
76- when ( clientLevelTracerContext . merge ( methodLevelTracerContext ))
77- . thenReturn ( clientLevelTracerContext );
78-
79- tracerFactory .withContext (clientLevelTracerContext );
80- ApiTracer actual = tracerFactory .newTracer (mock (ApiTracer .class ), methodLevelTracerContext );
72+ LibraryMetadata metadata =
73+ LibraryMetadata . newBuilder (). setArtifactName ( "gax-java" ). setVersion ( "1.0" ). build ( );
74+ ApiTracerContext clientLevelTracerContext =
75+ ApiTracerContext . newBuilder ().setLibraryMetadata ( metadata ). build ( );
76+ ApiTracerContext methodLevelTracerContext =
77+ ApiTracerContext . newBuilder (). setLibraryMetadata ( LibraryMetadata . empty ()). build ( );
78+
79+ ApiTracerFactory factoryWithContext = tracerFactory .withContext (clientLevelTracerContext );
80+ ApiTracer actual = factoryWithContext .newTracer (mock (ApiTracer .class ), methodLevelTracerContext );
8181
82- verify (clientLevelTracerContext ).merge (methodLevelTracerContext );
8382 assertThat (actual ).isInstanceOf (OpenTelemetryMetricsTracer .class );
8483 }
8584
You can’t perform that action at this time.
0 commit comments