@@ -283,7 +283,7 @@ impl DatadogPipelineBuilder {
283
283
}
284
284
285
285
/// Install the Datadog trace exporter pipeline using a simple span processor.
286
- pub fn install_simple ( mut self ) -> Result < Tracer , TraceError > {
286
+ pub fn install_simple ( mut self ) -> Result < ( Tracer , TracerProvider ) , TraceError > {
287
287
let ( config, service_name) = self . build_config_and_service_name ( ) ;
288
288
let exporter = self . build_exporter_with_service_name ( service_name) ?;
289
289
let mut provider_builder = TracerProvider :: builder ( ) . with_simple_exporter ( exporter) ;
@@ -295,13 +295,16 @@ impl DatadogPipelineBuilder {
295
295
. with_attributes ( None )
296
296
. build ( ) ;
297
297
let tracer = provider. tracer_with_scope ( scope) ;
298
- let _ = global:: set_tracer_provider ( provider) ;
299
- Ok ( tracer)
298
+ let _ = global:: set_tracer_provider ( provider. clone ( ) ) ;
299
+ Ok ( ( tracer, provider ) )
300
300
}
301
301
302
302
/// Install the Datadog trace exporter pipeline using a batch span processor with the specified
303
303
/// runtime.
304
- pub fn install_batch < R : RuntimeChannel > ( mut self , runtime : R ) -> Result < Tracer , TraceError > {
304
+ pub fn install_batch < R : RuntimeChannel > (
305
+ mut self ,
306
+ runtime : R ,
307
+ ) -> Result < ( Tracer , TracerProvider ) , TraceError > {
305
308
let ( config, service_name) = self . build_config_and_service_name ( ) ;
306
309
let exporter = self . build_exporter_with_service_name ( service_name) ?;
307
310
let mut provider_builder = TracerProvider :: builder ( ) . with_batch_exporter ( exporter, runtime) ;
@@ -313,8 +316,8 @@ impl DatadogPipelineBuilder {
313
316
. with_attributes ( None )
314
317
. build ( ) ;
315
318
let tracer = provider. tracer_with_scope ( scope) ;
316
- let _ = global:: set_tracer_provider ( provider) ;
317
- Ok ( tracer)
319
+ let _ = global:: set_tracer_provider ( provider. clone ( ) ) ;
320
+ Ok ( ( tracer, provider ) )
318
321
}
319
322
320
323
/// Assign the service name under which to group traces
0 commit comments