http - 重试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秒钟即可进行整个重试。我想念什么？
另外，如何停止服务器处理超时请求？，我看到CloseNotifier已被弃用。

最佳答案

问题出在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
}

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

关于http - 重试http请求RoundTrip，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/64179218/