Skip to content

Commit 0b66dc8

Browse files
committed
Refactor trace_flusher so it stores an http client.
1 parent 130771f commit 0b66dc8

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

datadog-sidecar/src/service/tracing/trace_flusher.rs

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,8 @@ use datadog_ipc::platform::NamedShmHandle;
77
use datadog_trace_utils::trace_utils;
88
use datadog_trace_utils::trace_utils::SendData;
99
use datadog_trace_utils::trace_utils::SendDataResult;
10-
use ddcommon::{Endpoint, MutexExt};
10+
use ddcommon::hyper_migration::new_default_client;
11+
use ddcommon::{Endpoint, HttpClient, MutexExt};
1112
use futures::future::join_all;
1213
use http_body_util::BodyExt;
1314
use manual_future::{ManualFuture, ManualFutureCompleter};
@@ -95,6 +96,7 @@ pub(crate) struct TraceFlusher {
9596
pub(crate) min_force_drop_size_bytes: AtomicU32, // put a limit on memory usage
9697
remote_config: Mutex<AgentRemoteConfigs>,
9798
pub metrics: Mutex<TraceFlusherMetrics>,
99+
client: HttpClient,
98100
}
99101
impl Default for TraceFlusher {
100102
fn default() -> Self {
@@ -105,6 +107,7 @@ impl Default for TraceFlusher {
105107
min_force_drop_size_bytes: AtomicU32::new(trace_utils::MAX_PAYLOAD_SIZE as u32),
106108
remote_config: Mutex::new(Default::default()),
107109
metrics: Mutex::new(Default::default()),
110+
client: new_default_client(),
108111
}
109112
}
110113
}
@@ -246,7 +249,7 @@ impl TraceFlusher {
246249

247250
async fn send_and_handle_trace(&self, send_data: SendData) {
248251
let endpoint = send_data.get_target().clone();
249-
let response = send_data.send().await;
252+
let response = send_data.send(&self.client).await;
250253
self.metrics.lock_or_panic().update(&response);
251254
match response.last_result {
252255
Ok(response) => {

0 commit comments

Comments
 (0)