exposes surf's with_http_client

Author: ctrlaltf24

influxdb/src/client/mod.rs

@@ -20,7 +20,7 @@ use http::StatusCode;
 #[cfg(feature = "reqwest")]
 use reqwest::{Client as HttpClient, Response as HttpResponse};
 #[cfg(feature = "surf")]
-use surf::{Client as HttpClient, Response as HttpResponse};
+use surf::{Client as HttpClient, HttpClient as SurfHttpClient, Response as HttpResponse};
 
 use crate::query::QueryType;
 use crate::Error;
@@ -91,6 +91,13 @@ impl Client {
         self
     }
 
+    /// Allows creation of custom http clients
+    #[cfg(feature = "surf")]
+    pub fn with_http_client<T: SurfHttpClient>(mut self, http_client: T) -> Self {
+        self.client = HttpClient::with_http_client(http_client);
+        self
+    }
+
     /// Returns the name of the database the client is using
     pub fn database_name(&self) -> &str {
         // safe to unwrap: we always set the database name in `Self::new`