Go context.WithCancel(ctx) 创建带有取消能力的上下文

╰半夏微凉° 2023-10-12 09:07 34阅读 0赞

#### 阅读目录 ####

*  context.WithCancel(ctx) 创建带有取消能力的上下文

## context.WithCancel(ctx) 创建带有取消能力的上下文 ##

Go 语言中，上下文（Context）用于传递请求范围的参数、取消信号和截止时间等信息。

context.WithCancel(ctx) 函数返回一个新的上下文和一个取消函数，可以用于取消衍生的上下文。

具体来说，context.WithCancel(ctx) 接受一个现有的上下文对象 ctx 作为参数，并返回一个新的上下文对象和一个 cancel 函数。

新的上下文对象继承了父上下文的截止时间、超时和其他值，但拥有自己的取消信号。

当调用返回的 cancel 函数时，会发送一个取消信号给该上下文及其所有的衍生上下文。

使用带有取消能力的上下文，可以在某些条件满足时，通过调用 cancel 函数来取消与该上下文相关的操作。这样，相关的 goroutine 或函数可以在接收到取消信号后进行清理操作并提前返回，避免不必要的等待或资源浪费。

package main
    
    import (
    	"context"
    	"fmt"
    	"time"
    )
    
    func main() {
        
    	// 创建根上下文
    	ctx := context.Background()
    
    	// 创建带有取消能力的上下文
    	ctx, cancel := context.WithCancel(ctx)
    
    	// 启动一个 goroutine，在一段时间后取消上下文
    	go func() {
        
    		time.Sleep(3 * time.Second)
    		cancel() // 取消上下文
    	}()
    
    	// 执行某些操作，检查上下文是否已被取消
    	for {
        
    		select {
        
    		case <-ctx.Done():
    			fmt.Println("上下文已取消")
    			return
    		default:
    			fmt.Println("继续执行操作")
    			time.Sleep(1 * time.Second)
    		}
    	}
    }

PS E:\TEXT\test> go run .\test.go
    继续执行操作
    继续执行操作
    继续执行操作
    上下文已取消
    PS E:\TEXT\test>

在这个示例中，我们首先创建了一个根上下文 ctx（使用 context.Background()）。

然后，使用 context.WithCancel(ctx) 创建了一个带有取消能力的上下文 ctx 和一个取消函数 cancel。

--------------------

在一个独立的 goroutine 中，我们通过调用 cancel() 函数来取消上下文，在 3 秒后发送取消信号。

在主 goroutine 中，我们通过检查上下文的 Done() 通道，判断上下文是否已被取消。如果上下文已被取消，我们打印出 “上下文已取消” 并提前返回。否则，我们打印出 “继续执行操作” 并继续执行某些操作，直到收到取消信号或其他条件满足。

使用 context.WithCancel(ctx) 创建带有取消能力的上下文，可以在需要时灵活地控制和取消与该上下文相关的操作。