Как установить тайм-аут для запросов http.Get () в Golang?

Я делаю сборщик URL-адресов в Go и имею список URL-адресов для получения. Я отправляю http.Get()запросы на каждый URL и получаю от них ответ.

resp,fetch_err := http.Get(url)

Как я могу установить индивидуальный тайм-аут для каждого запроса Get? (Время по умолчанию очень велико, и это делает мой сборщик очень медленным.) Я хочу, чтобы мой сборщик имел тайм-аут около 40-45 секунд, после чего он должен вернуть «запрос истек» и перейти к следующему URL-адресу.

Как я могу этого добиться?

Видимо в Go 1.3 http.Client есть поле Timeout

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

Это помогло мне.

Вам необходимо настроить собственный клиент с собственным транспортом, который использует настраиваемую функцию набора номера, которая оборачивается вокруг DialTimeout .

Что - то вроде (полностью непроверенным ) это :

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")
}

Чтобы добавить к ответу Волкера, если вы также хотите установить тайм-аут чтения / записи в дополнение к таймауту подключения, вы можете сделать что-то вроде следующего

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()

Если вы хотите делать это по запросу, обработка ошибок игнорируется для краткости:

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

Это мутирующее глобальное состояние без какой-либо координации. Тем не менее, это может быть нормально для вашего сборщика URL. В противном случае создайте частный экземпляр 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запускается только после того, как соединение установлено.