Merge pull request #15 from agis/14-custom-timeout

Add support for setting custom timeout

Author: xinsnake

digest_auth_client.go

@@ -18,18 +18,21 @@ type DigestRequest struct {
 	Auth     *authorization
 	Wa       *wwwAuthenticate
 	CertVal  bool
+	Timeout  time.Duration
 }
 
 type DigestTransport struct {
 	Password string
 	Username string
+	Timeout  time.Duration
 }
 
 // NewRequest creates a new DigestRequest object
 func NewRequest(username, password, method, uri, body string) DigestRequest {
 	dr := DigestRequest{}
 	dr.UpdateRequest(username, password, method, uri, body)
 	dr.CertVal = true
+	dr.Timeout = 30 * time.Second
 	return dr
 }
 
@@ -38,6 +41,7 @@ func NewTransport(username, password string) DigestTransport {
 	dt := DigestTransport{}
 	dt.Password = password
 	dt.Username = username
+	dt.Timeout = 30 * time.Second
 	return dt
 }
 
@@ -68,6 +72,7 @@ func (dt *DigestTransport) RoundTrip(req *http.Request) (resp *http.Response, er
 	}
 
 	dr := NewRequest(username, password, method, uri, body)
+	dr.Timeout = dt.Timeout
 	return dr.Execute()
 }
 
@@ -85,7 +90,7 @@ func (dr *DigestRequest) Execute() (resp *http.Response, err error) {
 	req.Header = dr.Header
 
 	client := &http.Client{
-		Timeout: 30 * time.Second,
+		Timeout: dr.Timeout,
 	}
 
 	if !dr.CertVal {
@@ -156,7 +161,7 @@ func (dr *DigestRequest) executeRequest(authString string) (resp *http.Response,
 	req.Header.Add("Authorization", authString)
 
 	client := &http.Client{
-		Timeout: 30 * time.Second,
+		Timeout: dr.Timeout,
 	}
 
 	if !dr.CertVal {