context经常发挥的作用:
- 多个goroutine之间的信息交互
- 进行超时控制
- 上下文控制
大概说明就是,进行进程之间的控制。
常用的方法有:Background(),WithCancel(),WithDeadline(),Background() - Background()方法,初始化一个空的context,没有过期时间,一般用在多线程的根部。
- WithCancel()
会继承一个context,如果父context被结束,或者超时,则子context也会被结束。
和Background()不同之处在于,这个方法在继承父context之后,会返回两个变量,第一个变量是context.Context,第二个变量是cancel函数,如果调用这个函数,就会导致这个context进程结束。
func main () {
ctx , cancel:= context.WithCancel(context.Background())
defer cancel()
t := gen(ctx)
for y := range t {
fmt.Println(y)
if y == 10 {
break
}
}
}
func gen(ctx context.Context) chan int{
context.Background()
dst := make(chan int , 1)
n := 1
go func() {
for {
select {
case <-ctx.Done():
return
case dst <- n:
n++
}
}
}()
return dst
}
- WithDeadline()
设置一个过期的时间,如果超过这个时间,此进程就会自动停止。
func main () {
ctx , cancel:= context.WithDeadline(context.Background() , time.Now().Add(time.Second))
defer cancel()
t := gen(ctx)
for y := range t {
fmt.Println(y)
}
}
func gen(ctx context.Context) chan int{
context.Background()
dst := make(chan int , 1)
n := 1
go func() {
for {
select {
case <-ctx.Done():
return
case dst <- n:
n++
}
}
}()
return dst
}
你会发现只执行1s就会结束程序
- WithTimeout()
这个方法与iWithDeadline不同的是此方法,第二个参数是Time.Duration,作用相似,会有一个计时器,如果超过这个时间,会自动停止
func main () {
ctx , cancel:= context.WithTimeout(context.Background() , time.Second)
defer cancel()
t := gen(ctx)
for y := range t {
fmt.Println(y)
}
}
func gen(ctx context.Context) chan int{
context.Background()
dst := make(chan int , 1)
n := 1
go func() {
for {
select {
case <-ctx.Done():
return
case dst <- n:
n++
}
}
}()
return dst
}
和上面的WithDeadline()相似,也是只执行1s中结束
- WithValue()
此方法是创建一个context会携带一个key,value值,并用于上下文的分享,也就是线程之间的通信,获得这个值,需要调用context.Value()根据key值返回value,如果没有返回nil
func main () {
ctx:= context.WithValue(context.Background() , "name" , "test")
fmt.Println(ctx.Value("name"))
}