网上关于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:
...
}