|
28 | 28 | import com.google.api.gax.batching.BatcherImpl; |
29 | 29 | import com.google.api.gax.batching.FlowController; |
30 | 30 | import com.google.api.gax.core.BackgroundResource; |
31 | | -import com.google.api.gax.grpc.GaxGrpcProperties; |
32 | 31 | import com.google.api.gax.grpc.GrpcCallContext; |
33 | 32 | import com.google.api.gax.grpc.GrpcCallSettings; |
34 | 33 | import com.google.api.gax.grpc.GrpcRawCallableFactory; |
|
48 | 47 | import com.google.api.gax.rpc.UnaryCallSettings; |
49 | 48 | import com.google.api.gax.rpc.UnaryCallable; |
50 | 49 | import com.google.api.gax.tracing.ApiTracerFactory; |
51 | | -import com.google.api.gax.tracing.OpencensusTracerFactory; |
52 | 50 | import com.google.api.gax.tracing.SpanName; |
53 | 51 | import com.google.api.gax.tracing.TracedServerStreamingCallable; |
54 | 52 | import com.google.api.gax.tracing.TracedUnaryCallable; |
|
100 | 98 | import com.google.cloud.bigtable.data.v2.stub.metrics.BigtableTracerUnaryCallable; |
101 | 99 | import com.google.cloud.bigtable.data.v2.stub.metrics.BuiltinMetricsTracerFactory; |
102 | 100 | import com.google.cloud.bigtable.data.v2.stub.metrics.CompositeTracerFactory; |
103 | | -import com.google.cloud.bigtable.data.v2.stub.metrics.MetricsTracerFactory; |
104 | | -import com.google.cloud.bigtable.data.v2.stub.metrics.RpcMeasureConstants; |
105 | 101 | import com.google.cloud.bigtable.data.v2.stub.metrics.StatsHeadersServerStreamingCallable; |
106 | 102 | import com.google.cloud.bigtable.data.v2.stub.metrics.StatsHeadersUnaryCallable; |
107 | 103 | import com.google.cloud.bigtable.data.v2.stub.metrics.TracedBatcherUnaryCallable; |
|
126 | 122 | import com.google.cloud.bigtable.data.v2.stub.sql.SqlRowMergingCallable; |
127 | 123 | import com.google.cloud.bigtable.gaxx.retrying.ApiResultRetryAlgorithm; |
128 | 124 | import com.google.cloud.bigtable.gaxx.retrying.RetryInfoRetryAlgorithm; |
129 | | -import com.google.common.annotations.VisibleForTesting; |
130 | 125 | import com.google.common.base.Functions; |
131 | 126 | import com.google.common.base.MoreObjects; |
132 | 127 | import com.google.common.base.Preconditions; |
|
135 | 130 | import com.google.common.util.concurrent.MoreExecutors; |
136 | 131 | import com.google.protobuf.ByteString; |
137 | 132 | import io.grpc.MethodDescriptor; |
138 | | -import io.opencensus.stats.Stats; |
139 | | -import io.opencensus.stats.StatsRecorder; |
140 | | -import io.opencensus.tags.TagKey; |
141 | | -import io.opencensus.tags.TagValue; |
142 | | -import io.opencensus.tags.Tagger; |
143 | | -import io.opencensus.tags.Tags; |
144 | 133 | import io.opentelemetry.api.OpenTelemetry; |
145 | 134 | import io.opentelemetry.api.common.Attributes; |
146 | 135 | import java.io.IOException; |
@@ -226,47 +215,11 @@ public static BigtableClientContext createBigtableClientContext( |
226 | 215 | public static ApiTracerFactory createBigtableTracerFactory( |
227 | 216 | EnhancedBigtableStubSettings settings, @Nullable OpenTelemetry openTelemetry) |
228 | 217 | throws IOException { |
229 | | - return createBigtableTracerFactory( |
230 | | - settings, Tags.getTagger(), Stats.getStatsRecorder(), openTelemetry); |
231 | | - } |
232 | | - |
233 | | - @VisibleForTesting |
234 | | - public static ApiTracerFactory createBigtableTracerFactory( |
235 | | - EnhancedBigtableStubSettings settings, |
236 | | - Tagger tagger, |
237 | | - StatsRecorder stats, |
238 | | - @Nullable OpenTelemetry openTelemetry) |
239 | | - throws IOException { |
240 | | - String projectId = settings.getProjectId(); |
241 | | - String instanceId = settings.getInstanceId(); |
242 | | - String appProfileId = settings.getAppProfileId(); |
243 | | - |
244 | | - ImmutableMap<TagKey, TagValue> attributes = |
245 | | - ImmutableMap.<TagKey, TagValue>builder() |
246 | | - .put(RpcMeasureConstants.BIGTABLE_PROJECT_ID, TagValue.create(projectId)) |
247 | | - .put(RpcMeasureConstants.BIGTABLE_INSTANCE_ID, TagValue.create(instanceId)) |
248 | | - .put(RpcMeasureConstants.BIGTABLE_APP_PROFILE_ID, TagValue.create(appProfileId)) |
249 | | - .build(); |
250 | 218 |
|
251 | 219 | ImmutableList.Builder<ApiTracerFactory> tracerFactories = ImmutableList.builder(); |
252 | | - tracerFactories |
253 | | - .add( |
254 | | - // Add OpenCensus Tracing |
255 | | - new OpencensusTracerFactory( |
256 | | - ImmutableMap.<String, String>builder() |
257 | | - // Annotate traces with the same tags as metrics |
258 | | - .put(RpcMeasureConstants.BIGTABLE_PROJECT_ID.getName(), projectId) |
259 | | - .put(RpcMeasureConstants.BIGTABLE_INSTANCE_ID.getName(), instanceId) |
260 | | - .put(RpcMeasureConstants.BIGTABLE_APP_PROFILE_ID.getName(), appProfileId) |
261 | | - // Also annotate traces with library versions |
262 | | - .put("gax", GaxGrpcProperties.getGaxGrpcVersion()) |
263 | | - .put("grpc", GaxGrpcProperties.getGrpcVersion()) |
264 | | - .put("gapic", Version.VERSION) |
265 | | - .build())) |
266 | | - // Add OpenCensus Metrics |
267 | | - .add(MetricsTracerFactory.create(tagger, stats, attributes)) |
268 | | - // Add user configured tracer |
269 | | - .add(settings.getTracerFactory()); |
| 220 | + |
| 221 | + tracerFactories.add(settings.getTracerFactory()); |
| 222 | + |
270 | 223 | BuiltinMetricsTracerFactory builtinMetricsTracerFactory = |
271 | 224 | openTelemetry != null |
272 | 225 | ? BuiltinMetricsTracerFactory.create(openTelemetry, createBuiltinAttributes(settings)) |
|
0 commit comments