迹忆客 专注技术分享

我们如何在 Go (Golang) 中实现具有给定超时的上下文？ 值得庆幸的是，Go 在标准库中内置了许多功能，在本文中，我们将了解如何使用 context 包定义超时。

带上下文的 Golang HTTP 请求

让我们首先探索一个场景，我们有一个 HTTP 请求，我们想要控制这种请求的超时。

package main

import "net/http"
import "log"
import "io/ioutil"

func main () {
  req, err := http.NewRequest(http.MethodGet, "http://jiyik.com/get", nil)
  if err != nil {
    log.Fatal(err)
  }
  c := &http.Client{}
  res, err := c.Do(req)
  if err != nil {
    log.Fatal(err)
  }
  defer res.Body.Close()
  out, err := ioutil.ReadAll(res.Body)
  if err != nil {
    log.Fatal(err)
  }
  log.Println(string(out))
}

好的，如果我们想确保这个请求在 80 毫秒内完成怎么办？ 我们只需要使用 context.WithTimeout！

这是我们需要对代码示例做的补充

ctx, cancel := context.WithTimeout(context.Background(), time.Duration(time.Millisecond*80))
defer cancel()
req = req.WithContext(ctx)

我们首先定义一个指定超时的新上下文（使用 time.Duration）。 然后我们使用 WithContext 将上下文添加到我们的请求中。 当创建带有超时的新上下文时，我们会收到一个 CancelFunc，它用于在上下文过期后清理资源。 这是有关 CancelFunc 的更多信息

调用 CancelFunc 会取消子项及其子项，删除父项对子项的引用，并停止任何关联的计时器。 未能调用 CancelFunc 会泄漏子级及其子级，直到父级被取消或计时器触发。 go vet 工具检查是否在所有控制流路径上使用了 CancelFuncs。

我们还可以使用 NewRequestWithContext 定义带有上下文的 http 请求

ctx, cancel := context.WithTimeout(context.Background(), time.Duration(time.Millisecond*80))
defer cancel()
req, err := http.NewRequestWithContext(ctx, http.MethodGet, "http://jiyik.com/get", nil)

下面是最后一个例子

package main

import "net/http"
import "log"
import "io/ioutil"
import "context"
import "time"

func main () {
  req, err := http.NewRequest(http.MethodGet, "http://jiyik.com/get", nil)
  if err != nil {
    log.Fatal(err)
  }
  ctx, cancel := context.WithTimeout(context.Background(), time.Duration(time.Millisecond*80))
  defer cancel()
  req = req.WithContext(ctx)
  c := &http.Client{}
  res, err := c.Do(req)
  if err != nil {
    log.Fatal(err)
  }
  defer res.Body.Close()
  out, err := ioutil.ReadAll(res.Body)
  if err != nil {
    log.Fatal(err)
  }
  log.Println(string(out))
}

这将在 80 毫秒后自动超时并取消请求