add proxy headers

syedriko · May 10, 2023 · 27459d0 · 27459d0
1 parent 5d3f619
commit 27459d0
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/http.rs b/src/http.rs
@@ -57,6 +57,7 @@ pub type HttpClientFuture = <HttpClient as Service<http::Request<Body>>>::Future
 pub struct HttpClient<B = Body> {
     client: Client<ProxyConnector<HttpsConnector<HttpConnector>>, B>,
     user_agent: HeaderValue,
+    proxy: ProxyConnector<HttpsConnector<HttpConnector>>,
 }
 
 impl<B> HttpClient<B>
@@ -78,13 +79,13 @@ where
         client_builder: &mut client::Builder,
     ) -> Result<HttpClient<B>, HttpError> {
         let proxy = build_proxy_connector(tls_settings.into(), proxy_config)?;
-        let client = client_builder.build(proxy);
+        let client = client_builder.build(proxy.clone());
 
         let version = crate::get_version();
         let user_agent = HeaderValue::from_str(&format!("Vector/{}", version))
             .expect("Invalid header value for version!");
 
-        Ok(HttpClient { client, user_agent })
+        Ok(HttpClient { client, user_agent, proxy })
     }
 
     pub fn send(
@@ -95,6 +96,11 @@ where
         let _enter = span.enter();
 
         default_request_headers(&mut request, &self.user_agent);
+        if let Some(proxy_headers) = self.proxy.http_headers(request.uri()) {
+            for (k, v) in proxy_headers {
+                request.headers_mut().insert(k, v.into());
+            }
+        }
 
         emit!(http_client::AboutToSendHttpRequest { request: &request });
 
@@ -216,6 +222,7 @@ impl<B> Clone for HttpClient<B> {
         Self {
             client: self.client.clone(),
             user_agent: self.user_agent.clone(),
+            proxy: self.proxy.clone(),
         }
     }
 }