Skip to content

Commit 2ad504d

Browse files
committed
update client usage for metrics flusher
1 parent 47aa08e commit 2ad504d

1 file changed

Lines changed: 9 additions & 9 deletions

File tree

bottlecap/src/bin/bottlecap/main.rs

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -80,8 +80,9 @@ use bottlecap::{
8080
use datadog_fips::reqwest_adapter::create_reqwest_client_builder;
8181
use decrypt::resolve_secrets;
8282
use dogstatsd::{
83-
aggregator_service::AggregatorHandle as MetricsAggregatorHandle,
84-
aggregator_service::AggregatorService as MetricsAggregatorService,
83+
aggregator::{
84+
AggregatorHandle as MetricsAggregatorHandle, AggregatorService as MetricsAggregatorService,
85+
},
8586
api_key::ApiKeyFactory,
8687
constants::CONTEXTS,
8788
datadog::{
@@ -95,7 +96,7 @@ use dogstatsd::{
9596
use libdd_trace_obfuscation::obfuscation_config;
9697
use reqwest::Client;
9798
use std::{collections::hash_map, env, path::Path, str::FromStr, sync::Arc};
98-
use tokio::time::{Duration, Instant};
99+
use tokio::time::Instant;
99100
use tokio::{sync::Mutex as TokioMutex, sync::mpsc::Sender};
100101
use tokio_util::sync::CancellationToken;
101102
use tracing::{debug, error, warn};
@@ -1179,10 +1180,12 @@ async fn start_dogstatsd(
11791180
});
11801181

11811182
// Get flushers with aggregator handle
1183+
let metrics_client = bottlecap::http::get_client(config);
11821184
let flushers = Arc::new(start_metrics_flushers(
11831185
Arc::clone(&api_key_factory),
11841186
&aggregator_handle,
11851187
config,
1188+
&metrics_client,
11861189
));
11871190

11881191
// Create Dogstatsd server
@@ -1214,6 +1217,7 @@ fn start_metrics_flushers(
12141217
api_key_factory: Arc<ApiKeyFactory>,
12151218
metrics_aggr_handle: &MetricsAggregatorHandle,
12161219
config: &Arc<Config>,
1220+
client: &Client,
12171221
) -> Vec<MetricsFlusher> {
12181222
let mut flushers = Vec::new();
12191223

@@ -1237,9 +1241,7 @@ fn start_metrics_flushers(
12371241
api_key_factory,
12381242
aggregator_handle: metrics_aggr_handle.clone(),
12391243
metrics_intake_url_prefix: metrics_intake_url.expect("can't parse site or override"),
1240-
https_proxy: config.proxy_https.clone(),
1241-
ca_cert_path: config.tls_cert_file.clone(),
1242-
timeout: Duration::from_secs(config.flush_timeout),
1244+
client: client.clone(),
12431245
retry_strategy: DsdRetryStrategy::Immediate(3),
12441246
compression_level: config.metrics_config_compression_level,
12451247
};
@@ -1267,9 +1269,7 @@ fn start_metrics_flushers(
12671269
api_key_factory: additional_api_key_factory,
12681270
aggregator_handle: metrics_aggr_handle.clone(),
12691271
metrics_intake_url_prefix: metrics_intake_url.clone(),
1270-
https_proxy: config.proxy_https.clone(),
1271-
ca_cert_path: config.tls_cert_file.clone(),
1272-
timeout: Duration::from_secs(config.flush_timeout),
1272+
client: client.clone(),
12731273
retry_strategy: DsdRetryStrategy::Immediate(3),
12741274
compression_level: config.metrics_config_compression_level,
12751275
};

0 commit comments

Comments
 (0)