gpt4 book ai didi

http - 重试 http 请求 RoundTrip

转载 作者:行者123 更新时间:2023-12-03 10:11:06 30 4
gpt4 key购买 nike

我制作了一个服务器,客户端通过 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 已经弃用了。

最佳答案

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

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/

30 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com