context在Golang函数并发编程中的作用-Golang-PHP中文网

context在Golang函数并发编程中的作用

王林

发布： 2024-04-17 14:27:02

原创

966人浏览过

context包在golang函数并发编程中用于管理协程执行，它提供了以下作用：传播取消信号，允许协程在任务完成前中断其他协程。设置截止时间，如果协程在截止时间内未完成则自动取消。传递附加信息，允许协程之间传递键值对。

context在Golang函数并发编程中的作用

context在Golang函数并发编程中的作用

context包是Golang中用于管理函数并发执行的关键工具。它提供了在协程之间传递请求取消信号、截止时间和其他相关信息的能力。

作用

立即学习“go语言免费学习笔记（深入）”；

取消信号传播：context允许协程传递取消信号，从而允许发起者协程在任务完成之前中断正在运行的协程。
截止时间设置：context可以指定截止时间，如果协程在截止时间之前未完成，则自动取消。
值传递：context可以携带任何类型的键值对，允许协程之间传递附加信息。

使用

要创建context对象，可以使用context.Background()或context.WithCancel()。

// 创建一个新context，取消信号为默认
ctx := context.Background()

// 创建一个带有取消信号的新context
ctx, cancel := context.WithCancel()

登录后复制

取消协程

豆包AI编程

豆包推出的AI编程助手

483

查看详情

要取消协程，只需调用cancel()函数。这将发送取消信号到所有监听该context的协程。

// 取消协程
cancel()

登录后复制

监听取消信号

协程可以使用context.Done()通道监听取消信号。当通道关闭时，表明context已取消。

// 监听取消信号
select {
case <-ctx.Done():
    // 处理取消
}

登录后复制

实战案例

考虑以下对HTTP请求进行超时的协程：

func MakeRequest(ctx context.Context, url string) (*http.Response, error) {
    // 创建一个带有截止时间的context
    ctx, cancel := context.WithTimeout(ctx, 10*time.Second)
    defer cancel()

    // 发起HTTP请求
    req, err := http.NewRequest(http.MethodGet, url, nil)
    if err != nil {
        return nil, err
    }

    req = req.WithContext(ctx)
    resp, err := http.DefaultClient.Do(req)
    if err != nil {
        return nil, err
    }

    return resp, nil
}

登录后复制

在此示例中：