使用golang

时间:2017-04-10 11:23:18

标签: go httprequest settimeout goroutine

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

enter image description here

流程概述:

转到服务器:基本上,充当[反向代理]。2

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

Application Server :核心请求处理逻辑。

现在,如果授权服务器无法在规定的时间内处理请求,那么我想从内存中关闭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"
}

在此处,如果请求未在规定时间内处理,则上下文将返回&#34; 截止日期超过&#34;但它继续处理该请求并在指定时间内返回响应。那么当超出时间时,如何停止请求流(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)的超时。

1 个答案:

答案 0 :(得分:15)

您的代码中发生的情况非常正确并且表现得如预期的那样。

您创建一个5 seconds超时的上下文。您将其传递给request并提出该请求。假设请求在2秒内返回。然后,您执行select并等待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 <-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错误。你是控制上下文的人。如果请求超时,您当然应该测试一个案例,那么将返回一个正确的错误供您验证。