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

今天golang学习网给大家带来了《使用 golang 对每个请求的上下文超时实现》，其中涉及到的知识点包括golang等等，无论你是小白还是老手，都适合看一看哦~有好的建议也欢迎大家在评论留言，若是看完有所收获，也希望大家能多多点赞支持呀！一起加油学习~

问题内容

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

流程概览：

Go Server ：基本上，充当[反向代理]。2

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

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

现在，如果授权服务器无法在规定的时间内处理请求，那么我想从内存中关闭 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 
<p>在这里，如果请求没有在规定的时间内处理，则上下文返回“ <strong>超出期限”。</strong>
但它会继续处理该请求并在指定时间内返回响应。那么，当超过时间时，如何停止请求流（goroutine）。</p>
<p>我还需要使用以下代码在 60 秒内处理完整的请求：</p>
<pre class="brush:go;toolbar:false">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
    },
}

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

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

正确答案

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

您创建一个带有5 seconds超时的上下文。您将其传递给request并提出该请求。假设请求在 2 秒内返回。然后，您执行 aselect并等待 10 秒或等待上下文完成。上下文将始终在创建后的最初 5 秒内完成，并且每次到达结束时也会给出该错误。

该context请求独立于请求，除非之前取消，否则它将达到其截止日期。当函数完成使用时取消请求defer。

在您的代码中，请求会考虑您的超时。但是每次到达超时ctx.Err()都会返回。deadline exceeded因为这就是context. ctx.Err()多次调用将返回相同的错误。

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

go func () {
    select {
    case 
<p>从上下文文档中：</p>
<pre class="brush:go;toolbar:false">// 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这里的错误。你是控制上下文的人。如果请求超时，当然是你应该测试的情况，然后会返回一个正确的错误供你验证。

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