简单理解GO CONTEXT机制

网上关于golang context的介绍很多，讲了一大堆，代码写的也很复杂，对于新手来说看的晕乎乎的，这里举个简单的例子来介绍下。直接上代码：

package main

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

func doWork(ctx context.Context, a int) {
	for {
		fmt.Println("loop %d", a)
		a++
		time.Sleep(1 * time.Second)
		select {
		case <-ctx.Done():
			fmt.Println("工作结束")
			return
		default:
		}
	}
}

func main() {
	a := 1
	ctx, cancelCtx := context.WithCancel(context.Background())
	go func() {
		time.Sleep(5 * time.Second)
		cancelCtx() // 在调用处主动取消
	}()
	go doWork(ctx, a)
	fmt.Println("end of do work")

	time.Sleep(10 * time.Second)

	fmt.Println("end of main")
}

执行结果如下：

C:/go/bin/go.exe build [D:/PROGRAM/Golang/src/study/context-test]
成功: 进程退出代码 0.
D:/PROGRAM/Golang/src/study/context-test/context-test.exe  [D:/PROGRAM/Golang/src/study/context-test]
end of do work
loop %d 1
loop %d 2
loop %d 3
loop %d 4
loop %d 5
工作结束
end of main
成功: 进程退出代码 0.

代码解读：

上述代码执行5s后调用CancelFunc（），触发取消ctx，从而达到取消doWork协程的目的。main协程在程序执行10秒后退出。

详细分析：

func WithCancel(parent Context) (ctx Context, cancel CancelFunc)

函数WithCancel接收一个父ctx，生成一个子ctx和一个CancelFunc。ctx.Done()方法返回一个信道（channel），当Context被撤销或过期时，该信道是关闭的，即它是一个表示Context是否已关闭的信号，调用CancelFunc（）表示信道关闭。

调用CancelFunc对象将撤销对应的Context对象，这就是主动撤销Context的方法。在父节点的Context所对应的环境中，通过WithCancel函数不仅可创建子节点的Context，同时也获得了该节点Context的控制权，一旦执行该函数，则该节点Context就结束了，则子节点需要类似如下代码来判断是否已结束，并退出该Goroutine：

select {
    case <-cxt.Done():
        // do some clean...
        return // exit goroutine
    default:
        ...
}

简单理解GO CONTEXT机制

猜你喜欢