Add timeout and send_null_body configurations to Builder

Author: spietika

src/lib.rs

@@ -115,6 +115,12 @@ pub enum Error {
 pub struct Builder {
     /// Number of DNS worker threads
     dns_workers: usize,
+
+    /// Request timeout
+    timeout: Duration,
+
+    /// Send null body
+    send_null_body: bool,
 }
 
 impl fmt::Display for Error {
@@ -154,7 +160,11 @@ impl error::Error for Error {
 
 impl Default for Builder {
     fn default() -> Self {
-        Self { dns_workers: 4 }
+        Self {
+            dns_workers: 4,
+            timeout: Duration::from_secs(std::u64::MAX),
+            send_null_body: true,
+        }
     }
 }
 
@@ -168,6 +178,24 @@ impl Builder {
         self
     }
 
+    /// Set request timeout
+    ///
+    /// Default is no timeout
+    #[inline]
+    pub fn timeout(&mut self, timeout: Duration) -> &mut Self {
+        self.timeout = timeout;
+        self
+    }
+
+    /// Send null body in POST/PUT
+    ///
+    /// Default is yes
+    #[inline]
+    pub fn send_null_body(&mut self, value: bool) -> &mut Self {
+        self.send_null_body = value;
+        self
+    }
+
     /// Create `RestClient` with the configuration in this builder
     pub fn build(&self, url: &str) -> Result<RestClient, Error> {
         RestClient::with_builder(url, self)
@@ -209,8 +237,8 @@ impl RestClient {
             baseurl,
             auth: None,
             headers: HeaderMap::new(),
-            timeout: Duration::from_secs(std::u64::MAX),
-            send_null_body: true,
+            timeout: builder.timeout,
+            send_null_body: builder.send_null_body,
         })
     }
 

tests/get.rs

@@ -4,6 +4,7 @@ extern crate restson;
 extern crate serde_derive;
 
 use restson::{Error, RestClient, RestPath};
+use std::time::Duration;
 
 #[derive(Deserialize)]
 struct HttpBinAnything {
@@ -50,6 +51,8 @@ fn basic_get_http() {
 fn basic_get_builder() {
     let mut client = RestClient::builder()
         .dns_workers(1)
+        .timeout(Duration::from_secs(10))
+        .send_null_body(false)
         .build("http://httpbin.org")
         .unwrap();