我正在尝试处理每个请求的上下文超时。我们有以下服务器结构:
流程概述:
转到服务器:基本上,充当[反向代理]。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)的超时。
答案 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
错误。你是控制上下文的人。如果请求超时,您当然应该测试一个案例,那么将返回一个正确的错误供您验证。