利用上下文对象实现超时控制

哈喽！今天心血来潮给大家带来了《利用上下文对象实现超时控制》，想必大家应该对Golang都不陌生吧，那么阅读本文就都不会很困难，以下内容主要涉及到，若是你正在学习Golang，千万别错过这篇文章~希望能帮助到你！

假设我有一个将 web 请求发送到 api 端点的函数，我想向客户端添加一个超时，这样如果调用花费的时间太长，操作就会通过返回错误或使当前的恐慌而中断。线程。

另一个假设是，客户端函数（发送 web 请求的函数）来自库，并且以同步方式实现。

让我们看一下客户端函数的签名：

func send(params map[string]string) (*http.response, error)

我想围绕这个函数编写一个包装器来添加超时机制。为此，我可以这样做：

func SendWithTimeout(ctx context.Context, params map[string]string) (*http.Response, error) {
    completed := make(chan bool)

    go func() {
        res, err := Send(params)
        _ = res
        _ = err
        completed <- true
    }()

    for {
        select {
        case <-ctx.Done():
            {
                return nil, errors.New("Cancelled")
            }
        case <-completed:
            {
                return nil, nil // just to test how this method works
            }
        }
    }
}

现在，当我调用新函数并传递可取消上下文时，我成功收到取消错误，但运行原始 send 函数的 goroutine 继续运行到最后。

由于该函数进行 api 调用，这意味着建立套接字/tcp 连接实际上是在后台进行的，因此将长时间运行的 api 留在后台并不是一个好的做法。

当 context.done() 被击中时，是否有任何标准方法可以中断原始 send 函数？

正确答案

这是一个“糟糕”的设计选择，将上下文支持添加到之前不支持的现有 API/实现中。应将上下文支持添加到使用它/监视它的现有 Send() 实现中，将其重命名为 SendWithTimeout()，并提供一个新的 Send() 函数，该函数不需要上下文，并使用 context.TODO() 调用 SendWithTimeout()或149​​669778。

例如，如果您的 Send() 函数发出传出 HTTP 调用，则可以通过使用 http.NewRequest() 后跟 Client.Do() 来实现。在新的上下文感知版本中，使用 http.NewRequestWithContext()。

如果您有一个无法更改的 Send() 函数，那么您就“运气不好”。函数本身必须支持上下文或取消。您无法从外部中止它。

查看相关内容：

Terminating function execution if a context is cancelled

Is it possible to cancel unfinished goroutines?

Stopping running function using context timeout in Golang

cancel a blocking operation in Go

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