Golang context 包详解

context 包在 Go 语言中用于传递上下文信息,常用于管理 goroutine 的生命周期、请求的处理 deadline、cancelation 信号、k-v 形式的值传递等。

以下是一些常用的 context 包的函数和方法:

  1. context.Background(): 返回一个非 nil 的空上下文。它通常作为处理最顶层函数的入参,例如命令行程序的 main 函数。
  2. context.TODO(): 返回一个非 nil 的空上下文。它用于那些无法归类到其他类型的上下文中的函数调用。
  3. context.WithCancel(parent Context) (ctx Context, cancel CancelFunc): 返回一个带有 cancel 功能的上下文。当调用 cancel 函数时,所有依赖于该上下文的 goroutine 都会接收到取消信号。
  4. context.WithDeadline(parent Context, deadline time.Time) (Context, CancelFunc): 返回一个将在指定时间点自动取消的上下文。
  5. context.WithTimeout(parent Context, timeout time.Duration) (Context, CancelFunc): 返回一个在指定时间后自动取消的上下文。
  6. context.WithValue(parent Context, key interface{}, val interface{}) Context: 返回一个可以携带键值对的上下文。
  7. 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 接收到取消信号并返回。

最后修改于:2024年09月03日 18:16

评论已关闭

推荐阅读

DDPG 模型解析,附Pytorch完整代码
2024年11月24日
DQN 模型解析,附Pytorch完整代码
2024年11月24日
AIGC实战——Transformer模型
2024年12月01日
Socket TCP 和 UDP 编程基础(Python)
2024年11月30日
python , tcp , udp
如何使用 ChatGPT 进行学术润色?你需要这些指令
2024年12月01日
AI
最新 Python 调用 OpenAi 详细教程实现问答、图像合成、图像理解、语音合成、语音识别(详细教程)
2024年11月24日
ChatGPT 和 DALL·E 2 配合生成故事绘本
2024年12月01日
omegaconf,一个超强的 Python 库!
2024年11月24日
【视觉AIGC识别】误差特征、人脸伪造检测、其他类型假图检测
2024年12月01日
[超级详细]如何在深度学习训练模型过程中使用 GPU 加速
2024年11月29日
Python 物理引擎pymunk最完整教程
2024年11月27日
MediaPipe 人体姿态与手指关键点检测教程
2024年11月27日
深入了解 Taipy:Python 打造 Web 应用的全面教程
2024年11月26日
基于Transformer的时间序列预测模型
2024年11月25日
Python在金融大数据分析中的AI应用(股价分析、量化交易)实战
2024年11月25日
AIGC Gradio系列学习教程之Components
2024年12月01日
Python3 `asyncio` — 异步 I/O,事件循环和并发工具
2024年11月30日
llama-factory SFT系列教程:大模型在自定义数据集 LoRA 训练与部署
2024年12月01日
Python 多线程和多进程用法
2024年11月24日
Python socket详解,全网最全教程
2024年11月27日