feat: example of using the DefaultHttpClient in libdatadog

Author: Aaalibaba42

Cargo.lock

@@ -36,6 +36,7 @@ libdd-common = { version = "1.1.0", path = "../libdd-common", default-features =
 libdd-telemetry = { version = "2.0.0", path = "../libdd-telemetry", default-features = false }
 libdd-trace-protobuf = { version = "1.0.0", path = "../libdd-trace-protobuf" }
 libdd-trace-stats = { version = "1.0.0", path = "../libdd-trace-stats" }
+libdd-provider = { version = "0.1.0", path = "../libdd-provider" }
 libdd-trace-utils = { version = "1.0.0", path = "../libdd-trace-utils", default-features = false }
 libdd-ddsketch = { version = "1.0.0", path = "../libdd-ddsketch" }
 libdd-dogstatsd-client = { version = "1.0.0", path = "../libdd-dogstatsd-client", default-features = false }

libdd-data-pipeline/Cargo.toml

@@ -5,9 +5,9 @@
 
 use super::{schema::AgentInfo, AGENT_INFO_CACHE};
 use anyhow::{anyhow, Result};
-use http_body_util::BodyExt;
-use hyper::{self, header::HeaderName};
-use libdd_common::{hyper_migration, worker::Worker, Endpoint};
+use hyper::header::HeaderName;
+use libdd_common::{entity_id, worker::Worker, Endpoint};
+use libdd_provider::{DefaultHttpClient, HttpClientTrait, HttpRequest};
 use sha2::{Digest, Sha256};
 use std::sync::Arc;
 use std::time::Duration;
@@ -81,15 +81,25 @@ pub async fn fetch_info(info_endpoint: &Endpoint) -> Result<Box<AgentInfo>> {
 /// Returns a tuple of (state_hash, response_body_bytes).
 /// The hash is calculated using SHA256 to match the agent's calculation method.
 async fn fetch_and_hash_response(info_endpoint: &Endpoint) -> Result<(String, bytes::Bytes)> {
-    let req = info_endpoint
-        .to_request_builder(concat!("Libdatadog/", env!("CARGO_PKG_VERSION")))?
-        .method(hyper::Method::GET)
-        .body(hyper_migration::Body::empty());
-    let client = hyper_migration::new_default_client();
-    let res = client.request(req?).await?;
-
-    let body_bytes = res.into_body().collect().await?;
-    let body_data = body_bytes.to_bytes();
+    let user_agent = concat!("Libdatadog/", env!("CARGO_PKG_VERSION"));
+    let mut req =
+        HttpRequest::get(info_endpoint.url.to_string()).with_header("user-agent", user_agent);
+
+    // Add optional endpoint headers (api-key, test-token)
+    for (name, value) in info_endpoint.get_optional_headers() {
+        req = req.with_header(name, value);
+    }
+
+    // Add entity-related headers (container-id, entity-id, external-env)
+    for (name, value) in entity_id::get_entity_headers() {
+        req = req.with_header(name, value);
+    }
+
+    let res = DefaultHttpClient::request(req)
+        .await
+        .map_err(|e| anyhow!("{}", e))?;
+
+    let body_data = bytes::Bytes::from(res.body);
     let hash = format!("{:x}", Sha256::digest(&body_data));
 
     Ok((hash, body_data))

libdd-data-pipeline/src/agent_info/fetcher.rs

@@ -7,7 +7,7 @@
 //! compile target. Downstream code uses these types directly, avoiding
 //! generic contagion.
 
-pub use libdd_capabilities::HttpClientTrait;
+pub use libdd_capabilities::{HttpClientTrait, HttpError, HttpRequest, HttpResponse};
 
 #[cfg(not(target_arch = "wasm32"))]
 pub use libdd_common::capabilities::HyperHttpClient as DefaultHttpClient;