Golang学习笔记-Context

有时候一个功能可能涉及多个goroutine的调用,当该功能中途需要取消时,需要通知其它goroutine,这个时候Context就可以派上用场了,Context主要用来处理退出通知以及上下文数据传递问题。

Context之间被设计为父子关系,要创建Context,首先要创建根节点,通常是context.Background(),该Context不能被取消,没有值,也没有过期时间。

有了根节点后,接下来就是创建子节点,子节点还可以继续创建子节点,有四种方式创建子节点:WithCancel/WithDeadline/WithTimeout/WithValue,前三种方法都会返回Context和CancelFunc,父节点通过调用CancelFunc通知子节点取消,子节点通过Context提供的Done方法检测是否被取消。WithValue将一对key-value关联到Context,通过Context提供的Value方法获取指定key对应的value。

WithDeadline/WithTimeout创建的Context在超时后会自动调用CancelFunc,如果调用没有超时,需要主动调用CancelFunc以避免不必要的资源浪费,所以通常在在调用WithDeadline/WithTimeout之后会使用defer cancel()确保资源尽快释放。

package main

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


func say(ctx context.Context){
	for{
		select {
		case <-ctx.Done():
			return
		default:
			fmt.Println(ctx.Value("test"))
			time.Sleep(time.Second)
		}
	}

}


func main(){
	//5秒后调用cancelFunc取消say
	ctx,cancelFunc:=context.WithCancel(context.Background())
	ctx=context.WithValue(ctx,"test","hello")
	go say(ctx)
	time.Sleep(time.Second*5)
	cancelFunc()



	//2秒后调用cancelFunc取消say
	ctx,_=context.WithTimeout(context.Background(),time.Second*2)
	ctx=context.WithValue(ctx,"test","world")
	go say(ctx)

	time.Sleep(time.Second*5)
}
©️2020 CSDN 皮肤主题: 编程工作室 设计师:CSDN官方博客 返回首页