Allow submitting Vec<Vec<Span>> asynchronously

Author: bengl

data-pipeline/src/trace_exporter/mod.rs

@@ -584,7 +584,16 @@ impl TraceExporter {
         trace_chunks: Vec<Vec<Span<T>>>,
     ) -> Result<AgentResponse, TraceExporterError> {
         self.check_agent_info();
-        self.send_trace_chunks_inner(trace_chunks)
+        self.runtime()?
+            .block_on(async { self.send_trace_chunks_inner(trace_chunks).await })
+    }
+
+    pub async fn send_trace_chunks_async<T: SpanText>(
+        &self,
+        trace_chunks: Vec<Vec<Span<T>>>,
+    ) -> Result<AgentResponse, TraceExporterError> {
+        self.check_agent_info();
+        self.send_trace_chunks_inner(trace_chunks).await
     }
 
     /// Deserializes, processes and sends trace chunks to the agent
@@ -614,7 +623,8 @@ impl TraceExporter {
             None,
         );
 
-        self.send_trace_chunks_inner(traces)
+        self.runtime()?
+            .block_on(async { self.send_trace_chunks_inner(traces).await })
     }
 
     /// Send traces payload to agent with retry and telemetry reporting
@@ -661,7 +671,7 @@ impl TraceExporter {
         self.handle_send_result(result, chunks, payload_len).await
     }
 
-    fn send_trace_chunks_inner<T: SpanText>(
+    async fn send_trace_chunks_inner<T: SpanText>(
         &self,
         mut traces: Vec<Vec<Span<T>>>,
     ) -> Result<AgentResponse, TraceExporterError> {
@@ -686,15 +696,13 @@ impl TraceExporter {
             ..self.endpoint.clone()
         };
 
-        self.runtime()?.block_on(async {
-            self.send_traces_with_telemetry(
-                &endpoint,
-                prepared.data,
-                prepared.headers,
-                prepared.chunk_count,
-            )
-            .await
-        })
+        self.send_traces_with_telemetry(
+            &endpoint,
+            prepared.data,
+            prepared.headers,
+            prepared.chunk_count,
+        )
+        .await
     }
 
     /// Handle the result of sending traces to the agent