Golang context 包详解
context
包在 Go 语言中用于传递上下文信息,常用于管理 goroutine 的生命周期、请求的处理 deadline、cancelation 信号、k-v 形式的值传递等。
以下是一些常用的 context
包的函数和方法:
context.Background()
: 返回一个非 nil 的空上下文。它通常作为处理最顶层函数的入参,例如命令行程序的 main 函数。context.TODO()
: 返回一个非 nil 的空上下文。它用于那些无法归类到其他类型的上下文中的函数调用。context.WithCancel(parent Context) (ctx Context, cancel CancelFunc)
: 返回一个带有 cancel 功能的上下文。当调用 cancel 函数时,所有依赖于该上下文的 goroutine 都会接收到取消信号。context.WithDeadline(parent Context, deadline time.Time) (Context, CancelFunc)
: 返回一个将在指定时间点自动取消的上下文。context.WithTimeout(parent Context, timeout time.Duration) (Context, CancelFunc)
: 返回一个在指定时间后自动取消的上下文。context.WithValue(parent Context, key interface{}, val interface{}) Context
: 返回一个可以携带键值对的上下文。type Context interface { Deadline() (deadline time.Time, ok bool); Done() <-chan struct{}; Err() error; Value(key interface{}) interface{} }
:Context 接口定义了上下文应该实现的方法。
示例代码:
package main
import (
"context"
"fmt"
"time"
)
func main() {
// 创建一个带有取消功能的上下文
ctx, cancel := context.WithCancel(context.Background())
// 启动一个 goroutine
go func(ctx context.Context) {
select {
case <-ctx.Done():
fmt.Println("Goroutine is canceled.")
return
case <-time.After(5 * time.Second):
fmt.Println("Goroutine is done.")
}
}(ctx)
// 取消上下文
cancel()
// 等待 goroutine 结束
time.Sleep(1 * time.Second)
}
在这个例子中,我们创建了一个带有取消功能的上下文,并启动了一个 goroutine。我们通过调用 cancel 函数来取消上下文,这会导致 goroutine 接收到取消信号并返回。
评论已关闭