如何在Golang中为http.Get（）请求设置超时？

我在Go中制作了一个URL提取程序，并具有要提取的URL列表。我将http.Get()请求发送到每个URL并获得他们的响应。

resp,fetch_err := http.Get(url)

如何为每个Get请求设置自定义超时？（默认时间很长，这会使我的提取程序非常慢。）我希望提取程序的超时时间为40-45秒左右，之后它应该返回“请求超时”并移至下一个URL。

我该如何实现？

显然在Go 1.3中，http.Client具有“超时”字段

client := http.Client{
    Timeout: 5 * time.Second,
}
client.Get(url)

这为我完成了窍门。

您需要使用自己的传输器设置自己的客户端，该传输器使用环绕DialTimeout的自定义Dial函数。

像（完全未经测试）这样的东西：

var timeout = time.Duration(2 * time.Second)

func dialTimeout(network, addr string) (net.Conn, error) {
    return net.DialTimeout(network, addr, timeout)
}

func main() {
    transport := http.Transport{
        Dial: dialTimeout,
    }

    client := http.Client{
        Transport: &transport,
    }

    resp, err := client.Get("http://some.url")
}

要添加到Volker的答案中，如果除了连接超时之外还希望设置读/写超时，则可以执行以下操作

package httpclient

import (
    "net"
    "net/http"
    "time"
)

func TimeoutDialer(cTimeout time.Duration, rwTimeout time.Duration) func(net, addr string) (c net.Conn, err error) {
    return func(netw, addr string) (net.Conn, error) {
        conn, err := net.DialTimeout(netw, addr, cTimeout)
        if err != nil {
            return nil, err
        }
        conn.SetDeadline(time.Now().Add(rwTimeout))
        return conn, nil
    }
}

func NewTimeoutClient(connectTimeout time.Duration, readWriteTimeout time.Duration) *http.Client {

    return &http.Client{
        Transport: &http.Transport{
            Dial: TimeoutDialer(connectTimeout, readWriteTimeout),
        },
    }
}

该代码已经过测试，正在生产中。完整的测试要点可以在这里 https://gist.github.com/dmichael/5710968

请注意，您将需要为每个请求创建一个新的客户端，因为conn.SetDeadline它将引用将来的某个点time.Now()

如果要根据请求执行此操作，为简便起见，忽略了err处理：

ctx, cncl := context.WithTimeout(context.Background(), time.Second*3)
defer cncl()

req, _ := http.NewRequestWithContext(ctx, http.MethodGet, "https://google.com", nil)

resp, _ := http.DefaultClient.Do(req)

快速而肮脏的方式：

http.DefaultTransport.(*http.Transport).ResponseHeaderTimeout = time.Second * 45

这正在改变全局状态而没有任何协调。但是，对于您的网址提取程序来说可能还可以。否则，创建一个私有实例http.RoundTripper：

var myTransport http.RoundTripper = &http.Transport{
        Proxy:                 http.ProxyFromEnvironment,
        ResponseHeaderTimeout: time.Second * 45,
}

var myClient = &http.Client{Transport: myTransport}

resp, err := myClient.Get(url)
...

上面没有测试。

您可以使用https://github.com/franela/goreq，它以一种简单的方式处理超时。

timeout := time.Duration(5 * time.Second)
transport := &http.Transport{Proxy: http.ProxyURL(proxyUrl), ResponseHeaderTimeout:timeout}

这可能会有所帮助，但请注意，ResponseHeaderTimeout只有在建立连接后才能开始。