go context之WithTimeout的使用

最新推荐文章于 2025-07-03 11:34:09 发布

YZF_Kevin

最新推荐文章于 2025-07-03 11:34:09 发布

阅读量3.4w

点赞数 14

CC 4.0 BY-SA版权

分类专栏： Go语言文章标签： go WithTimeout

本文链接：https://blog.csdn.net/yzf279533105/article/details/107292247

Go语言专栏收录该内容

56 篇文章

订阅专栏

本文详细解析了Go语言中context包的WithTimeout()函数如何在协程中实现超时控制。通过创建子Context并监控其Done通道，演示了在超时或手动调用cancel函数时，如何使协程优雅地退出。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

简言

1. context包的WithTimeout()函数接受一个 Context 和超时时间作为参数，返回其子Context和取消函数cancel

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

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

4. 若不调用cancel函数，到了原先创建Contetx时的超时时间，它也会自动调用cancel()函数，即会往子Context的Done通道发送消息
实验步骤

1. 利用根Context创建一个父Context，使用父Context创建2个协程，超时时间设为3秒

2. 等待8秒钟，再调用cancel函数，其实这个时候会发现在3秒钟的时候两个协程已经收到退出信号了
实验结果如下图

完整代码

package main

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

func main() {
	// 创建一个子节点的context,3秒后自动超时
	ctx, cancel := context.WithTimeout(context.Background(), time.Second*3)

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

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

	fmt.Println("等待8秒结束,准备调用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)
		}
	}
}