diff --git a/configuration.go b/configuration.go index 081189eb..f6177dd3 100644 --- a/configuration.go +++ b/configuration.go @@ -9,4 +9,5 @@ const ( defaultMaxRedirections = 64 defaultTimeout = 10 * time.Second agentName = "muffet" + tcpTimeout = time.Minute ) diff --git a/fasthttp_http_client_factory.go b/fasthttp_http_client_factory.go index 20fd8ce2..b6b0987d 100644 --- a/fasthttp_http_client_factory.go +++ b/fasthttp_http_client_factory.go @@ -2,6 +2,7 @@ package main import ( "crypto/tls" + "net" "github.com/valyala/fasthttp" ) @@ -21,7 +22,9 @@ func (*fasthttpHTTPClientFactory) Create(o httpClientOptions) httpClient { TLSConfig: &tls.Config{ InsecureSkipVerify: o.SkipTLSVerification, }, - }, + Dial: func(addr string) (net.Conn, error) { + return fasthttp.DialTimeout(addr, tcpTimeout) + }}, o.MaxRedirections, o.Timeout, )