feat(aggregator-client): add ClientBuilder::with_relay_endpoint

Author: Alenar

internal/mithril-aggregator-client/src/builder.rs

@@ -1,5 +1,5 @@
 use anyhow::Context;
-use reqwest::{IntoUrl, Url};
+use reqwest::{Client, IntoUrl, Proxy, Url};
 use slog::{Logger, o};
 use std::collections::HashMap;
 use std::time::Duration;
@@ -15,6 +15,7 @@ pub struct AggregatorClientBuilder {
     api_version_provider: Option<APIVersionProvider>,
     additional_headers: Option<HashMap<String, String>>,
     timeout_duration: Option<Duration>,
+    relay_endpoint: Option<String>,
     logger: Option<Logger>,
 }
 
@@ -28,6 +29,7 @@ impl AggregatorClientBuilder {
             api_version_provider: None,
             additional_headers: None,
             timeout_duration: None,
+            relay_endpoint: None,
             logger: None,
         }
     }
@@ -56,6 +58,12 @@ impl AggregatorClientBuilder {
         self
     }
 
+    /// Set the address of the relay
+    pub fn with_relay_endpoint(mut self, relay_endpoint: String) -> Self {
+        self.relay_endpoint = Some(relay_endpoint);
+        self
+    }
+
     /// Returns an [AggregatorClient] based on the builder configuration
     pub fn build(self) -> StdResult<AggregatorClient> {
         let aggregator_endpoint =
@@ -65,6 +73,12 @@ impl AggregatorClientBuilder {
         let logger = self.logger.unwrap_or_else(|| Logger::root(slog::Discard, o!()));
         let api_version_provider = self.api_version_provider.unwrap_or_default();
         let additional_headers = self.additional_headers.unwrap_or_default();
+        let mut client_builder = Client::builder();
+
+        if let Some(relay_endpoint) = self.relay_endpoint {
+            client_builder = client_builder
+                .proxy(Proxy::all(relay_endpoint).with_context(|| "Relay proxy creation failed")?)
+        }
 
         Ok(AggregatorClient {
             aggregator_endpoint,
@@ -73,7 +87,9 @@ impl AggregatorClientBuilder {
                 .try_into()
                 .with_context(|| format!("Invalid headers: '{additional_headers:?}'"))?,
             timeout_duration: self.timeout_duration,
-            client: reqwest::Client::new(),
+            client: client_builder
+                .build()
+                .with_context(|| "HTTP client creation failed")?,
             logger,
         })
     }