Go context.CancelFunc 取消一个 Context

以你之姓@ 2023-10-12 09:51 6阅读 0赞

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

*  Go context.CancelFunc 取消一个 Context

## Go context.CancelFunc 取消一个 Context ##

Go 语言中，context.CancelFunc 是一个函数类型，用于取消一个context.Context的操作。

当调用 CancelFunc 时，与之关联的 Context 会被取消，所有依赖于该 Context 的操作都会收到取消信号。

下面是一个示例，演示如何使用context.CancelFunc取消一个Context：

package main
    
    import (
    	"context"
    	"fmt"
    	"time"
    )
    
    type MiraMenu struct {
        
    	cancel context.CancelFunc
    }
    
    func main() {
        
    	// 创建一个父Context和取消函数
    	parentCtx, cancel := context.WithCancel(context.Background())
    
    	// 创建一个MiraMenu实例
    	menu := MiraMenu{
        
    		cancel: cancel,
    	}
    
    	// 启动一个goroutine来执行一些任务
    	go menu.doSomething(parentCtx)
    
    	// 等待一段时间后取消操作
    	time.Sleep(2 * time.Second)
    	menu.cancel()
    
    	// 等待一段时间让任务完成
    	time.Sleep(1 * time.Second)
    }
    
    func (m *MiraMenu) doSomething(ctx context.Context) {
        
    	for {
        
    		select {
        
    		case <-ctx.Done():
    			fmt.Println("任务已取消")
    			return
    		default:
    			fmt.Println("正在执行任务")
    			time.Sleep(500 * time.Millisecond)
    		}
    	}
    }

在上面的示例中，我们创建了一个MiraMenu结构体，其中包含一个cancel字段，该字段的类型为context.CancelFunc。

在main函数中，我们创建了一个父Context和相应的取消函数cancel，然后将取消函数赋值给MiraMenu实例的cancel字段。

接着，我们在一个单独的goroutine中执行menu.doSomething方法，该方法会持续执行一个任务。在任务执行过程中，我们使用select语句来检查Context的取消状态。如果取消函数被调用，ctx.Done()会接收到信号，我们会打印出"任务已取消"并返回函数，终止任务的执行。

在主goroutine中，我们等待2秒钟后调用menu.cancel()，这会触发取消操作。然后我们再等待1秒钟，以便任务有足够的时间来执行完成。

运行上述示例代码，你将会看到输出类似于以下内容：

PS D:\TEXT\test> go run .\hello.go
    正在执行任务
    正在执行任务
    正在执行任务
    正在执行任务
    任务已取消
    PS D:\TEXT\test>