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

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

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

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

作用

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

使用

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

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

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

登录后复制

取消协程

要取消协程，只需调用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
}

登录后复制

在此示例中：

• 使用context.WithTimeout()创建带有10秒截止时间的context。
• 请求被设置为此context，允许底层网络调用超时并取消请求。
• defer cancel()确保在函数退出时取消context，防止任何其他协程被阻塞。

通过使用context，我们可以控制协程的执行，避免资源泄漏和不必要的等待。

以上就是context在Golang函数并发编程中的作用的详细内容，更多请关注叮当号网其它相关文章！