如何在 Goroutine 中使用上下文取消功能？-Golang-PHP中文网

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

WBOY

发布： 2024-05-15 12:24:02

原创

1181人浏览过

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

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

如何在 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)
}

登录后复制

在这个示例中：