From ddb488cb477b243467fe4b44d3bfb28f075f7fcf Mon Sep 17 00:00:00 2001 From: Yota Toyama Date: Mon, 21 Sep 2020 22:15:12 -0700 Subject: [PATCH] Increase default TCP timeout (#92) --- configuration.go | 1 + fasthttp_http_client_factory.go | 5 ++++- 2 files changed, 5 insertions(+), 1 deletion(-) 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, )