golang withcancel

2024-10-16 19:41:44 编程来源：ZONE.CI 全球网 0 阅读模式

开头

在golang开发中，我们常常需要处理并发任务的取消操作。使用context包下的WithCancel函数可以创建一个新的上下文，该上下文在调用其cancel函数时会关闭，从而实现任务的取消。本文将介绍如何使用golang的WithCancel函数来处理并发任务的取消。

使用WithCancel函数

WithCancel函数用于创建一个带有取消功能的上下文。它返回一个上下文对象和一个可以调用的cancel函数。当我们想要取消某个任务时，只需要调用cancel函数即可。

首先，我们需要导入"context"包，然后使用WithCancel函数创建一个新的上下文：

import "context"

func main() {
    ctx, cancel := context.WithCancel(context.Background())
    defer cancel()
    // 其他代码
}

在这个例子中，我们使用了background上下文作为基础上下文。在Golang中，background上下文是默认的顶层上下文。我们可以使用WithCancel函数来派生出一个新的上下文。

取消并发任务

一旦我们创建了具有取消功能的上下文，我们就可以将其传递给并发任务。当我们想要取消这些任务时，只需在需要的地方调用cancel函数。

例如，我们将使用goroutine来执行一个模拟的并发任务：

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

func main() {
    ctx, cancel := context.WithCancel(context.Background())
    defer cancel()

    go func() {
        for {
            select {
            case <-ctx.done(): return="" 取消任务="" default:="" 执行任务逻辑="" fmt.println("处理并发任务")="" time.sleep(time.second)="" }="" }="" }()="" time.sleep(3="" *="" time.second)="" cancel()="" 取消任务="" time.sleep(time.second)="" }="">

在这个例子中，我们创建了一个匿名的goroutine来执行并发任务。在goroutine中，我们使用select语句来监听取消信号。当调用cancel函数时，ctx.Done()会接收到信号，并执行对应的取消逻辑。

取消任务的传播

一个上下文可以用于派生出其他的上下文。当一个上下文被取消时，其派生的所有上下文也会被自动取消。这种机制可以确保任务的取消操作能够正确地传播到所有相关的并发任务。

例如，我们可以创建一个父级上下文，并从中派生出两个子级上下文。当我们取消父级上下文时，两个子级上下文也会被取消：

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

func main() {
    parentCtx, cancel := context.WithCancel(context.Background())
    defer cancel()

    childCtx1, _ := context.WithCancel(parentCtx)
    childCtx2, _ := context.WithCancel(parentCtx)

    go func() {
        for {
            select {
            case <-childctx1.done(): fmt.println("子级上下文1已被取消")="" return="" default:="" fmt.println("处理子级任务1")="" time.sleep(time.second)="" }="" }="" }()="" go="" func()="" {="" for="" {="" select="" {="" case=""><-childctx2.done(): fmt.println("子级上下文2已被取消")="" return="" default:="" fmt.println("处理子级任务2")="" time.sleep(time.second)="" }="" }="" }()="" time.sleep(3="" *="" time.second)="" cancel()="" 取消父级任务="" time.sleep(time.second)="" }="">

在这个例子中，我们创建了一个父级上下文parentCtx，并从中派生出两个子级上下文childCtx1和childCtx2。当我们取消父级上下文时，两个子级上下文也会相应地被取消。

通过这种机制，我们可以确保并发任务的取消操作能够正确地传播到所有相关的任务，避免出现资源泄露或未经授权的操作。