如何在 Goroutine 中使用上下文取消功能？

在 Go 中使用上下文取消功能可以优雅地取消正在进行的 Goroutine：使用 context 包创建带超时的上下文。在函数返回时使用 defer 取消上下文。在 Goroutine 中使用 select 语句监听取消事件。

如何在 Goroutine 中使用上下文取消功能？

在 Go 中，上下文取消机制允许我们在某些条件满足时优雅地取消正在进行的 Goroutine。这在需要在后台运行但可以在必要时取消的任务中非常有用。

使用场景

上下文取消特别适合以下场景：

• 用户取消操作
• 超时任务
• 资源争用时的错误处理

实现细节

要使用上下文取消，需要使用 context 包。如下所示：

package main

import (
    "context"
    "fmt"
    "time"
)

func main() {
    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
    defer cancel()

    go func() {
        for {
            select {
            case <-ctx.Done():
                fmt.Println("Context cancelled!")
                return
            default:
                fmt.Println("Working...")
                time.Sleep(1 * time.Second)
            }
        }
    }()

    time.Sleep(10 * time.Second)
}

在这个示例中：

• context.WithTimeout() 创建了一个新的上下文，它将超时 5 秒。
• defer cancel() 确保在 main 函数返回时取消上下文。
• Goroutine 将运行一个无限循环，在 ctx.Done() 通道收到信号时退出。
• 主流程将在 10 秒后退出，此时 Goroutine 将被取消。

实战案例

在真实的应用程序中，上下文取消可以用于以下任务：

• Web 服务器： 当用户取消请求时取消 HTTP 处理程序。
• 数据库连接： 当数据库连接不再需要时取消数据库查询。
• 文件 I/O： 当文件传输需要取消时取消 I/O 操作。

注意事项

请注意以下注意事项：

• 使用 context.Done() 通道来监听取消事件。
• 应该在 Goroutine 中使用 defer cancel() 来在函数返回时取消上下文。
• 在需要时使用 select 语句在 Goroutine 中监听取消事件。