go context之WithDeadline的使用

简言

1. WithDeadline()函数接受一个 Context 和过期时间作为参数,返回其子Context和取消函数cancel

2. 新创建协程中传入子Context做参数,且需监控子Context的Done通道,若收到消息,则退出

3. 需要新协程结束时,在外面调用 cancel 函数,即会往子Context的Done通道发送消息

实验步骤

1. 利用根Context创建一个父Context,使用父Context创建2个协程,设定3秒过期

2. 等待5秒钟,再调用cancel函数,其实这个时候会发现在3秒钟的时候两个协程已经收到退出信号了

实验结果如下图

go context之WithDeadline的使用_第1张图片

完整代码

package main

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

func main() {
	// 创建一个子节点的context,3秒后自动取消
	ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(time.Second*3))

	go watch(ctx, "监控1")
	go watch(ctx, "监控2")

	fmt.Println("现在开始等待5秒,time=", time.Now().Unix())
	time.Sleep(5 * time.Second)

	fmt.Println("等待5秒结束,准备调用cancel()函数,发现两个子协程已经结束了,time=", time.Now().Unix())
	cancel()
}

// 单独的监控协程
func watch(ctx context.Context, name string) {
	for {
		select {
		case <-ctx.Done():
			fmt.Println(name, "收到信号,监控退出,time=", time.Now().Unix())
			return
		default:
			fmt.Println(name, "goroutine监控中,time=", time.Now().Unix())
			time.Sleep(1 * time.Second)
		}
	}
}

 

你可能感兴趣的:(go context之WithDeadline的使用)