使用golang对每个请求执行上下文超时

前端之家收集整理的这篇文章主要介绍了使用golang对每个请求执行上下文超时前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
我正在尝试处理每个请求的上下文超时.我们有以下服务器结构:

流程概述:

Go Server:基本上,充当[反向代理] .2

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

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"
}

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

那么我还需要任何单独的上下文实现吗?在此先感谢您的帮助.

注意1:如果我们可以使用上下文管理HTTP服务器创建的每个请求(goroutine)的超时,那将是非常棒的.

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

您创建一个5秒超时的上下文.您将其传递给请求并发出请求.假设请求在2秒内返回.然后执行选择并等待10秒或等待上下文完成.上下文将始终在创建后的最初5秒内完成,并且每次到达结尾时也会发出错误.

上下文与请求无关,除非事先取消,否则它将达到截止日期.在函数完成使用延迟时取消请求.

在您的代码中,请求会考虑您的超时.但是ctx.Err()将在每次达到超时时返回截止日期.因为那是在上下文中发生的事情.多次调用ctx.Err()将返回相同的错误.

ctx,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,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秒通过或达到上下文超时.在您的情况下,始终达到上下文超时.

你应该检查客户端返回的错误.调用并不担心这里的上下文错误.你是控制上下文的人.如果请求超时,您当然应该测试一个案例,那么将返回一个正确的错误供您验证.

猜你在找的Go相关文章