深入Golang之context的用法详解-编程学习网

这篇文章将为大家详细讲解有关深入Golang之context的用法详解，小编觉得挺实用的，因此分享给大家做个参考，希望大家阅读完这篇文章后可以有所收获。

Go 语言中 Context 的用法详解

简介

Context 是 Go 语言中用于传递请求相关信息的机制，它允许开发人员在代码中传递请求元数据，并在整个请求的生命周期中访问这些信息。Context 主要用于管理请求的超时、取消和传播值。

创建 Context

可以通过以下方法创建 Context：

context.Background()：创建一个新的顶级 Context。
context.TODO()：创建一个与当前环境无关的 Context。通常用于测试或仅需一个占位符 Context 的情况。
context.WithTimeout(parent, timeout)：从父 Context 创建一个带超时的 Context。
context.WithCancel(parent)：从父 Context 创建一个带取消功能的 Context。

访问 Context

可以使用以下方法访问 Context 中的信息：

ctx.Done()：返回一个 chan，当 Context 被取消时关闭。
ctx.Err()：当 Context 被取消时返回一个错误，否则返回 nil。
ctx.Deadline()：返回 Context 的截止时间。
ctx.Value(key)：使用给定的键从 Context 中获取值。

传递 Context

可以通过函数参数或中间件传递 Context。例如：

func MyFunction(ctx context.Context, input string) {
  // 使用 ctx
}

func MyMiddleware(next http.Handler) http.Handler {
  return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    // 从 r 中提取 ctx
    ctx := r.Context()

    // 使用 ctx
    next.ServeHTTP(w, r.WithContext(ctx))
  })
}

超时

使用 context.WithTimeout() 创建的 Context 会在指定的持续时间后超时。在超时的情况下，Context 的 Done() 通道将关闭。

取消

使用 context.WithCancel() 创建的 Context 可以通过调用 CancelFunc 来取消。在取消的情况下，Context 的 Done() 通道将关闭。

值

可以向 Context 中存储值，以便在整个请求生命周期中访问。使用 context.WithValue() 将值存储到 Context 中，并使用 context.Value() 检索值。

最佳实践