重试 http 请求 RoundTrip

Golang不知道大家是否熟悉？今天我将给大家介绍《重试 http 请求 RoundTrip》，这篇文章主要会讲到等等知识点，如果你在看完本篇文章后，有更好的建议或者发现哪里有问题，希望大家都能积极评论指出，谢谢！希望我们能一起加油进步！

我制作了一个服务器，客户端通过 http 进行访问。我在客户端的 transport 的 roundtripper 方法中设置了重试机制。以下是每个服务器和客户端的工作代码示例：

服务器main.go

package main

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

func test(w http.responsewriter, req *http.request) {
    time.sleep(2 * time.second)
    fmt.fprintf(w, "hello\n")
}

func main() {
    http.handlefunc("/test", test)
    http.listenandserve(":8090", nil)
}

客户端main.go

package main

import (
    "context"
    "fmt"
    "log"
    "net/http"
    "time"
)

type Retry struct {
    nums      int
    transport http.RoundTripper
}

// to retry
func (r *Retry) RoundTrip(req *http.Request) (resp *http.Response, err error) {
    for i := 0; i < r.nums; i++ {
        log.Println("Attempt: ", i+1)
        resp, err = r.transport.RoundTrip(req)
        if resp != nil && err == nil {
            return
        }
        log.Println("Retrying...")
    }
    return
}

func main() {
    r := &Retry{
        nums:      5,
        transport: http.DefaultTransport,
    }

    c := &http.Client{Transport: r}
    // each request will be timeout in 1 second
    ctx, cancel := context.WithTimeout(context.Background(), 1*time.Second)
    defer cancel()
    req, err := http.NewRequestWithContext(ctx, http.MethodGet, "http://localhost:8090/test", nil)
    if err != nil {
        panic(err)
    }
    resp, err := c.Do(req)
    if err != nil {
        panic(err)
    }
    fmt.Println(resp.StatusCode)
}

发生的情况是重试似乎只适用于第一次迭代。对于后续迭代，它不会每次等待一秒，而是打印与重试次数一样多的调试消息。

我希望重试每次等待 1 秒，因为我在上下文中设置了 1 秒的超时。但整个重试似乎只需要等待1秒。我错过了什么？

此外，如何阻止服务器处理超时请求？，我看到 clos​​enotifier 已弃用。

解决方案

问题出在 context 上。一旦上下文完成，您就不能再重复使用相同的上下文。每次尝试都必须重新创建上下文。您可以从父上下文中获取超时，并使用它来创建新上下文。

func (r *retry) roundtrip(req *http.request) (resp *http.response, err error) {
    var (
        duration time.duration
        ctx      context.context
        cancel   func()
    )
    if deadline, ok := req.context().deadline(); ok {
        duration = time.until(deadline)
    }
    for i := 0; i < r.nums; i++ {
        if duration > 0 {
            ctx, cancel = context.withtimeout(context.background(), duration)
            req = req.withcontext(ctx)
        }
        resp, err = r.rt.roundtrip(req)
        ...
        // the rest of code
        ...
    }
    return
}

此代码将在每次尝试时使用其父级的超时创建新的上下文。

对于服务器，您可以使用 request.context() 来检查请求是否被取消.

在客户端，当上下文在 1 秒后超时时，请求就会超时。所以上下文不会触发往返周期。如果您希望请求在上下文完成之前重试，您应该更改您正在使用的传输的行为。您现在正在使用 http.defaulttransport，其定义如下：

var DefaultTransport RoundTripper = &Transport{
    Proxy: ProxyFromEnvironment,
    DialContext: (&net.Dialer{
        Timeout:   30 * time.Second,
        KeepAlive: 30 * time.Second,
        DualStack: true,
    }).DialContext,
    ForceAttemptHTTP2:     true,
    MaxIdleConns:          100,
    IdleConnTimeout:       90 * time.Second,
    TLSHandshakeTimeout:   10 * time.Second,
    ExpectContinueTimeout: 1 * time.Second,
}

transport 在此处设置了更多超时变量，因此根据您想要重试的时间，您应该设置适当的超时。例如，在您的情况下，您可以将 transport.responseheadertimeout 设置为 1 秒。当服务器在 1 秒内未回复响应标头时，客户端将重试。然后，您使上下文在 5（或更好 6）秒后超时，您应该会看到客户端重试您指定的次数（5 次）。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《重试 http 请求 RoundTrip》文章吧，也可关注golang学习网公众号了解相关技术文章。