Add timeout to HTTP client

This commit is contained in:
Andrew E. Bruno 2016-03-18 23:20:58 -04:00
parent 1de97a90da
commit 9f8d5e4076

View file

@ -7,6 +7,7 @@ import (
"fmt" "fmt"
"net/http" "net/http"
"strconv" "strconv"
"time"
"github.com/xenolf/lego/acme" "github.com/xenolf/lego/acme"
) )
@ -64,7 +65,8 @@ func (d *DNSProvider) sendRequest(method, resource string, payload interface{})
req.Header.Set("Auth-Token", d.token) req.Header.Set("Auth-Token", d.token)
} }
resp, err := http.DefaultClient.Do(req) client := &http.Client{Timeout: time.Duration(10 * time.Second)}
resp, err := client.Do(req)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -136,7 +138,9 @@ func (d *DNSProvider) logout() error {
} }
req.Header.Set("Content-Type", "application/json") req.Header.Set("Content-Type", "application/json")
req.Header.Set("Auth-Token", d.token) req.Header.Set("Auth-Token", d.token)
resp, err := http.DefaultClient.Do(req)
client := &http.Client{Timeout: time.Duration(10 * time.Second)}
resp, err := client.Do(req)
if err != nil { if err != nil {
return err return err
} }
@ -219,7 +223,9 @@ func (d *DNSProvider) CleanUp(domain, token, keyAuth string) error {
} }
req.Header.Set("Content-Type", "application/json") req.Header.Set("Content-Type", "application/json")
req.Header.Set("Auth-Token", d.token) req.Header.Set("Auth-Token", d.token)
resp, err := http.DefaultClient.Do(req)
client := &http.Client{Timeout: time.Duration(10 * time.Second)}
resp, err := client.Do(req)
if err != nil { if err != nil {
return err return err
} }