使用 golang 对每个请求实现上下文超时

我正在尝试处理每个请求的上下文超时。我们有以下服务器结构：

流程概述：

去服务器：基本上，充当[反向代理]。2 https://en.wikipedia.org/wiki/Reverse_proxy

认证服务器：检查请求身份验证。

应用服务器：核心请求处理逻辑。

现在，如果授权服务器无法在规定的时间内处理请求，那么我想从内存中关闭 goroutine。

这是我尝试过的：

ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
req, _ := http.NewRequest("GET", authorizationServer, nil)
req.Header = r.Header
req.WithContext(ctx)
res, error := client.Do(req)
select {
case <-time.After(10 * time.Second):
    fmt.Println("overslept")
case <-ctx.Done():
    fmt.Println(ctx.Err()) // prints "context deadline exceeded"
}

在这里，上下文返回为“超过最后期限“，如果请求没有在规定的时间内处理。但它继续处理该请求并在超过指定的时间内返回响应。那么，当超过时间时，如何停止请求流（goroutine）。

我还需要使用以下代码在 60 秒内处理完整的请求：

var netTransport = &http.Transport{
    Dial: (&net.Dialer{
        Timeout: 60 * time.Second,
    }).Dial,
    TLSHandshakeTimeout: 60 * time.Second,
}
client := &http.Client{
    Timeout:   time.Second * 60,
    Transport: netTransport,
    CheckRedirect: func(req *http.Request, via []*http.Request) error {
        return http.ErrUseLastResponse
    },
}

那么我还需要任何单独的上下文实现吗？

Note1：如果我们可以使用上下文来管理 HTTP 服务器创建的每个请求（goroutine）的超时，那就太棒了。

您的代码中发生的事情非常正确并且行为符合预期。

您创建一个上下文5 seconds暂停。你把它传递给request并提出该请求。假设请求在 2 秒内返回。然后你做一个select然后等待 10 秒或等待上下文完成。上下文将始终在创建后的最初 5 秒内完成，并且每次到达末尾时也会给出该错误。

The context与请求无关，除非事先取消，否则它将达到截止日期。当函数完成使用时，您取消请求defer.

在您的代码中，请求会考虑您的超时。但是ctx.Err()将返回deadline exceeded每次达到超时时间。因为这就是内部发生的事情context。呼叫ctx.Err()多次将返回相同的错误。

ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()

go func () {
    select {
    case <-time.After(10 * time.Second):
        fmt.Println("overslept")
    case <-ctx.Done():
        fmt.Println(ctx.Err()) // prints "context deadline exceeded"
    }
}()
req, _ := http.NewRequest("GET", authorizationServer, nil)
req.Header = r.Header
req = req.WithContext(ctx)
res, error := client.Do(req)

从上下文文档：

// Err returns a non-nil error value after Done is closed. Err returns
// Canceled if the context was canceled or DeadlineExceeded if the
// context's deadline passed. No other values for Err are defined.
// After Done is closed, successive calls to Err return the same value.

在您的代码中，超时将始终达到且不会取消，这就是您收到的原因DeadlineExceeeded。您的代码是正确的，除了选择部分，该部分将阻塞，直到 10 秒过去或达到上下文超时。在您的情况下，总是会达到上下文超时。

您应该检查error由返回client.Do打电话，不用担心context这里有错误。你是控制环境的人。如果请求超时，当然你应该测试这种情况，然后会返回一个正确的错误供你验证。