Go 创建一个带有取消功能的上下文（context）

旧城等待， 2023-10-11 22:36 96阅读 0赞

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

*  Go 创建一个带有取消功能的上下文（context）

## Go 创建一个带有取消功能的上下文（context） ##

package main
    
    import (
    	"context"
    	"fmt"
    	"time"
    )
    
    func main() {
        
    	// 创建一个带有取消功能的上下文
    	ctx, cancel := context.WithCancel(context.Background())
    
    	// 在后台执行一些任务
    	go doSomeWork(ctx)
    
    	// 模拟等待一段时间后取消任务
    	time.Sleep(3 * time.Second)
    	cancel() // 调用取消函数，取消上下文
    
    	// 等待一段时间，以便观察任务是否已取消
    	time.Sleep(20 * time.Second)
    	fmt.Println("Done")
    }
    
    func doSomeWork(ctx context.Context) {
        
    	for {
        
    		select {
        
    		case <-ctx.Done():
    			fmt.Println("Work canceled")
    			return
    		default:
    			fmt.Println("Doing some work")
    			time.Sleep(500 * time.Millisecond)
    		}
    	}
    }

PS E:\TEXT\test> go run .\test.go
    Doing some work
    Doing some work
    Doing some work
    Doing some work
    Doing some work
    Doing some work
    Work canceled
    Done
    PS E:\TEXT\test>

WithCancel 返回带有新 Done 通道的 parent 的副本。

当返回的取消函数被调用或父上下文的 Done 通道关闭时，返回的上下文的 Done 通道被关闭，以先发生者为准。

取消此上下文会释放与其关联的资源，因此代码应在此上下文中运行的操作完成后立即调用取消。

> 使用带有取消功能的上下文可以更好地控制并发任务的执行，并在需要时及时终止任务。

1 首先，context.Background() 返回一个空的根上下文（background context），它是所有上下文的起点。

2 context.WithCancel() 函数接受一个上下文作为参数，并返回一个新的上下文和一个取消函数（cancel function）。取消函数用于取消上下文的执行，即停止与该上下文相关的操作。

3 context.WithCancel(context.Background()) 中，我们将空的根上下文作为参数传递给 context.WithCancel()，这将创建一个新的上下文和一个取消函数。这个新的上下文可以在需要的地方传递和使用，而取消函数可以在需要取消上下文的时候调用。

4 示例中，我们创建了一个带有取消功能的上下文 ctx 和取消函数 cancel。然后，我们在后台执行一个任务 doSomeWork()，该任务会定期打印消息。在主函数中，我们等待一段时间后调用取消函数 cancel()，以取消任务的执行。然后，我们等待一段时间观察任务是否已被取消。