Golang 函数如何使用上下文来取消并发任务？

在 go 中，可以使用 context 包的 context.context 来取消并发任务，它提供了一种在 goroutine 之间传递取消信号的机制：创建 context.context：使用 context.background() 或 context.withcancel() 创建一个表示取消上下文的 context.context 对象。取消上下文：调用 context.withcancel() 返回的 cancelfunc 来取消上下文。传递上下文：将上下文传递给 goroutine，以便它们可以通过 context.done() 检查取消状态。实战案例：在 http 请求超时时优雅取消请求和关闭连接。

Go 中如何使用上下文取消并发任务

在 Go 中，context 包提供了一个机制，用于在不同的 goroutine 之间传递取消信号。这对于在超时或应用程序关闭时优雅地取消并发任务非常有用。

使用 context.Context

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

首先，创建一个代表取消上下文的 context.Context。这是通过调用 context.Background() 或 context.WithCancel() 完成的：

Latent Labs

下载

ctx := context.Background()

要取消上下文，请调用 CancelFunc，它由 context.WithCancel() 返回：

cancelFunc := context.WithCancel(ctx).CancelFunc

传递上下文

将上下文传递给 goroutine，以便它们可以检查取消状态：

go func() {
    for {
        select {
        case <-ctx.Done():
            // 上下文已取消，退出 goroutine
            return
        default:
            // 执行任务
        }
    }
}()

实战案例

考虑一个获取 HTTP 请求响应的 goroutine。如果请求在指定的时间量内没有完成，我们希望取消请求并关闭连接：

import (
    "context"
    "net/http"
)

func main() {
    // 创建带有 5 秒超时的上下文
    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
    defer cancel()

    // 创建一个 HTTP 请求
    req, err := http.NewRequest(http.MethodGet, "https://example.com", nil)
    if err != nil {
        log.Fatal(err)
    }

    // 添加 ctx 到请求
    req = req.WithContext(ctx)

    // 执行请求
    resp, err := http.DefaultClient.Do(req)
    if err != nil {
        if err == context.Canceled {
            // 请求被取消
            log.Println("请求已取消")
        } else {
            log.Fatal(err)
        }
    }
    defer resp.Body.Close()
}