Let user configure the http pipeline (#421)

graph-http/src/client.rs

@@ -1,8 +1,8 @@
 use crate::blocking::BlockingClient;
 use crate::traits::ODataQuery;
 
-use crate::internal::{ExponentialBackoffRetryPolicy, ThrottleRetryPolicy, TransportPolicy};
 use crate::http_pipeline::HttpPipelinePolicy;
+use crate::internal::{ExponentialBackoffRetryPolicy, ThrottleRetryPolicy, TransportPolicy};
 use graph_error::GraphResult;
 use reqwest::header::{HeaderMap, HeaderValue, ACCEPT, USER_AGENT};
 use reqwest::redirect::Policy;
@@ -24,6 +24,7 @@ struct ClientConfiguration {
     connection_verbose: bool,
     https_only: bool,
     min_tls_version: Version,
+    pipeline: Vec<Arc<dyn HttpPipelinePolicy + Send + Sync>>,
 }
 
 impl ClientConfiguration {
@@ -42,6 +43,7 @@ impl ClientConfiguration {
             /// TLS 1.2 required to support all features in Microsoft Graph
             /// See [Reliability and Support](https://learn.microsoft.com/en-us/graph/best-practices-concept#reliability-and-support)
             min_tls_version: Version::TLS_1_2,
+            pipeline: vec![],
         }
     }
 }
@@ -136,6 +138,30 @@ impl GraphClientConfiguration {
         self
     }
 
+    pub fn pipeline(
+        mut self,
+        pipeline: Vec<Arc<dyn HttpPipelinePolicy + Send + Sync>>,
+    ) -> GraphClientConfiguration {
+        self.config.pipeline = pipeline;
+        self
+    }
+
+    pub fn add_pipeline_policy(
+        mut self,
+        policy: Arc<dyn HttpPipelinePolicy + Send + Sync>,
+    ) -> GraphClientConfiguration {
+        self.config.pipeline.push(policy);
+        self
+    }
+
+    pub fn add_default_pipeline_retries_policies(mut self) -> GraphClientConfiguration {
+        self.config
+            .pipeline
+            .push(Arc::new(ExponentialBackoffRetryPolicy::default()));
+        self.config.pipeline.push(Arc::new(ThrottleRetryPolicy {}));
+        self
+    }
+
     pub fn build(self) -> Client {
         let config = self.clone();
         let headers = self.config.headers.clone();
@@ -155,16 +181,15 @@ impl GraphClientConfiguration {
             builder = builder.connect_timeout(connect_timeout);
         }
 
+        let mut pipeline = self.config.pipeline;
+        pipeline.push(Arc::new(TransportPolicy {}));
+
         Client {
             access_token: self.config.access_token.unwrap_or_default(),
             inner: builder.build().unwrap(),
             headers,
             builder: config,
-            pipeline: vec![
-                Arc::new(ExponentialBackoffRetryPolicy::default()),
-                Arc::new(ThrottleRetryPolicy {}),
-                Arc::new(TransportPolicy {}),
-            ],
+            pipeline,
         }
     }
 

graph-http/src/http_pipeline.rs

@@ -63,7 +63,7 @@ pub struct ExponentialBackoffRetryPolicy {
 }
 
 impl ExponentialBackoffRetryPolicy {
-    pub(crate) fn get_exponential_backoff_with_max_retries(&self) -> ExponentialBackoffWithMaxRetries {
+    fn get_exponential_backoff_with_max_retries(&self) -> ExponentialBackoffWithMaxRetries {
         ExponentialBackoffWithMaxRetries {
             exp: ExponentialBackoffBuilder::new()
                 .with_initial_interval(self.initial_interval)
@@ -86,7 +86,7 @@ impl Default for ExponentialBackoffRetryPolicy {
             multiplier: 1.5,
             max_interval: Duration::from_secs(450),
             max_elapsed_time: Some(Duration::from_secs(450)),
-            max_retries: 10,
+            max_retries: 5,
         }
     }
 }
@@ -138,7 +138,7 @@ impl HttpPipelinePolicy for ExponentialBackoffRetryPolicy {
 pub struct ThrottleRetryPolicy {}
 
 impl ThrottleRetryPolicy {
-    pub(crate) fn get_retries_zero_backoff(&self) -> ExponentialBackoff {
+    fn get_retries_zero_backoff(&self) -> ExponentialBackoff {
         ExponentialBackoffBuilder::new()
             .with_initial_interval(Duration::ZERO)
             .with_multiplier(0.0)

graph-http/src/lib.rs

@@ -41,5 +41,6 @@ pub mod api_impl {
     pub use crate::resource_identifier::{ResourceConfig, ResourceIdentifier};
     pub use crate::traits::{BodyExt, ODataQuery};
     pub use crate::upload_session::UploadSession;
+    pub use crate::http_pipeline::{HttpPipelinePolicy, ThrottleRetryPolicy, ExponentialBackoffRetryPolicy};
     pub use graph_error::{GraphFailure, GraphResult};
 }

src/lib.rs

@@ -315,6 +315,7 @@ pub mod http {
         ODataMetadataLink, ODataNextLink, ODataQuery, ResponseBlockingExt, ResponseExt,
         UploadSessionLink,
     };
+    pub use graph_http::api_impl::{HttpPipelinePolicy, ThrottleRetryPolicy, ExponentialBackoffRetryPolicy};
     pub use reqwest::tls::Version;
     pub use reqwest::{Body, Method};