Fixed incompatible install_simple with Tokio runtimes exporters (#21)

abdolence · Sep 25, 2023 · 42e8601 · 42e8601
1 parent 88b8635
commit 42e8601
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -22,7 +22,7 @@ path = "src/lib.rs"
 
 [dependencies]
 tracing = "0.1"
-opentelemetry = { version = "0.20" }
+opentelemetry = { version = "0.20", features = ["rt-tokio"] }
 opentelemetry-semantic-conventions = { version = "0.12" }
 gcloud-sdk = { version = "0.20", features = ["google-devtools-cloudtrace-v2"] }
 tonic = { version = "0.9", features = ["tls"] }
@@ -32,7 +32,6 @@ rsb_derive = "0.5"
 tokio = { version = "1" }
 tokio-stream = "0.1"
 futures = "0.3"
-futures-util = "0.3"
 async-trait = "0.1"
 
 

diff --git a/src/lib.rs b/src/lib.rs
@@ -69,8 +69,8 @@ impl GcpCloudTraceExporterBuilder {
     ) -> Result<opentelemetry::sdk::trace::Tracer, opentelemetry::trace::TraceError> {
         let exporter = GcpCloudTraceExporter::new(&self.google_project_id).await?;
 
-        let mut provider_builder =
-            opentelemetry::sdk::trace::TracerProvider::builder().with_simple_exporter(exporter);
+        let mut provider_builder = opentelemetry::sdk::trace::TracerProvider::builder()
+            .with_batch_exporter(exporter, opentelemetry::runtime::Tokio);
         provider_builder = if let Some(config) = self.trace_config {
             provider_builder.with_config(config)
         } else {

diff --git a/src/span_exporter.rs b/src/span_exporter.rs
@@ -1,7 +1,7 @@
 use crate::errors::*;
 use crate::google_trace_exporter_client::GcpCloudTraceExporterClient;
 use crate::TraceExportResult;
-use futures_util::future::{BoxFuture, TryFutureExt};
+use futures::future::{BoxFuture, TryFutureExt};
 use gcloud_sdk::GoogleEnvironment;
 use opentelemetry::sdk::export::trace::{ExportResult, SpanData, SpanExporter};
 use std::fmt::Formatter;