weed/util/http/client/http_client_opt.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

package client

import (
	"net"
	"time"
)

type HttpClientOpt = func(clientCfg *HTTPClient)

func AddDialContext(httpClient *HTTPClient) {
	dialContext := (&net.Dialer{
		Timeout:   10 * time.Second,
		KeepAlive: 10 * time.Second,
	}).DialContext

	httpClient.Transport.DialContext = dialContext
	httpClient.Client.Transport = httpClient.Transport
}