golang超时控制
前言
日常开发中我们大概率会遇到超时控制的场景,比如一个批量耗时任务、网络请求等;一个良好的超时控制可以有效的避免一些问题(比如 goroutine
泄露、资源不释放等)。
Timer
在 go 中实现超时控制的方法非常简单,首先第一种方案是 Time.After(d Duration)
:
func main() {fmt.Println(time.Now())x := <-time.After(3 * time.Second)fmt.Println(x)
}
output:
2021-10-27 23:06:04.304596 +0800 CST m=+0.000085653
2021-10-27 23:06:07.306311 +0800 CST m=+3.001711390
time.After()
会返回一个 Channel
,该 Channel
会在延时 d 段时间后写入数据。
有了这个特性就可以实现一些异步控制超时的场景:
func main() {ch := make(chan struct{}, 1)go func() {fmt.Println("do something...")time.Sleep(4*time.Second)ch<- struct{}{}}()select {case <-ch:fmt.Println("done")case <-time.After(3*time.Second):fmt.Println("timeout")}
}
这里假设有一个 goroutine
在跑一个耗时任务,利用 select 有一个 channel
获取到数据便退出的特性,当 goroutine
没有在有限时间内完成任务时,主 goroutine
便会退出,也就达到了超时的目的。
output:
do something...
timeout
timer.After 取消,同时 Channel 发出消息,也可以关闭通道等通知方式。
注意 Channel 最好是有大小,防止阻塞 goroutine ,导致泄露。
Context
第二种方案是利用 context,go 的 context 功能强大;
利用 context.WithTimeout()
方法会返回一个具有超时功能的上下文。
ch := make(chan string)timeout, cancel := context.WithTimeout(context.Background(), 3*time.Second)defer cancel()go func() {time.Sleep(time.Second * 4)ch <- "done"}()select {case res := <-ch:fmt.Println(res)case <-timeout.Done():fmt.Println("timout", timeout.Err())}
同样的用法,context
的 Done()
函数会返回一个 channel
,该 channel
会在当前工作完成或者是上下文取消生效。
timout context deadline exceeded
通过 timeout.Err()
也能知道当前 context
关闭的原因。
goroutine 传递 context
使用 context
还有一个好处是,可以利用其天然在多个 goroutine 中传递的特性,让所有传递了该 context 的 goroutine 同时接收到取消通知,这点在多 go 中应用非常广泛。
func main() {total := 12var num int32log.Println("begin")ctx, cancelFunc := context.WithTimeout(context.Background(), 3*time.Second)for i := 0; i < total; i++ {go func() {//time.Sleep(3 * time.Second)atomic.AddInt32(&num, 1)if atomic.LoadInt32(&num) == 10 {cancelFunc()}}()}for i := 0; i < 5; i++ {go func() {select {case <-ctx.Done():log.Println("ctx1 done", ctx.Err())}for i := 0; i < 2; i++ {go func() {select {case <-ctx.Done():log.Println("ctx2 done", ctx.Err())}}()}}()}time.Sleep(time.Second*5)log.Println("end", ctx.Err())fmt.Printf("执行完毕 %v", num)
}
在以上例子中,无论 goroutine
嵌套了多少层,都是可以在 context
取消时获得消息(当然前提是 context
得传递走)
某些特殊情况需要提前取消 context 时,也可以手动调用 cancelFunc()
函数。
Gin 中的案例
Gin 提供的 Shutdown(ctx)
函数也充分使用了 context
。
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)defer cancel()if err := srv.Shutdown(ctx); err != nil {log.Fatal("Server Shutdown:", err)}log.Println("Server exiting")
比如以上代码便是超时等待 10s 进行 Gin
的资源释放,实现的原理也和上文的例子相同。
总结
因为写 go 的时间不长,所以自己写了一个练手的项目:一个接口压力测试工具。
其中一个很常见的需求就是压测 N 秒后退出,这里正好就应用到了相关知识点,同样是初学 go
的小伙伴可以参考。
ptg/duration.go at d0781fcb5551281cf6d90a86b70130149e1525a6 · crossoverJie/ptg · GitHub
作者: crossoverJie
出处: https://crossoverjie.top
欢迎关注博主公众号与我交流。
本文版权归作者所有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出, 如有问题, 可邮件(crossoverJie#gmail.com)咨询。
golang超时控制相关推荐
- golang基础-chan的select操作、定时器操作、超时控制、goroutine中使用recover
chan的只读和只写 a.只读chan的声明 Var 变量的名字 <-chan int Var readChan <- chan int b. 只写chan的声明 Var 变量的名字 ch ...
- golang通过itemid获取zabbix graph监控图
2019独角兽企业重金招聘Python工程师标准>>> #简述 本文将使用golang和第三方http client 库gorequest编写.如需要只使用golang 标准库可以参 ...
- golang context 父子任务同步取消信号 协程调度 简介
目录 前言 为什么需要context context是什么 context的使用 总结 前言 这篇文章将介绍Golang并发编程中常用到一种编程模式:context.本文将从为什么需要context出 ...
- golang net/http 超时机制完全手册
目录 SetDeadline 服务器端超时设置 客户端超时设置 Cancel 和 Context 英文原始出处: The complete guide to Go net/http timeouts, ...
- Golang 并发编程指南
分享 Golang 并发基础库,扩展以及三方库的一些常见问题.使用介绍和技巧,以及对一些并发库的选择和优化探讨. go 原生/扩展库 提倡的原则 不要通过共享内存进行通信;相反,通过通信来共享内存. ...
- 深入理解Golang之context
深入理解Golang之context context是Go并发编程中常用到一种编程模式.本文将从为什么需要context,深入了解context的实现原理,以了解如何使用context. 作者:Tur ...
- 微服务高可用之熔断器实现原理与 Golang 实践
I.内容提要 在微服务架构中,经常会碰到服务超时或通讯失败的问题,由于服务间层层依赖,很可能由于某个服务出现问题,不合理的重试和超时设置,导致问题层层传递引发雪崩现象,而限流和熔断是解决这个问题重要的 ...
- Golang 须知知识点
(1)同级目录不能放多个包,否则报编译错误: getLocalIP.go:8:5: found packages test (test.go) and test1 (test1.go) in /dat ...
- golang基础-etcd介绍与使用、etcd存取值、etcd监测数据写入
关注公众号"风色年代"订阅更多精彩文章,本博大部分文章为转载并已标明原文出处,如有再转敬请保留,请自觉尊重原创作者的劳动成果! golang基础-etcd介绍与使用.etcd存取值 ...
最新文章
- 深入了解WM_SIZE
- Win2K下关联进程/端口之代码初步分析
- 面试官:说说 Java 中的 Unsafe 和 CAS
- IntelliJ IDEA快捷键总结
- 内存泄露检测工具比较
- python爬虫常见报错_对Python爬虫常见工具总结,欢迎补充
- centos7 网卡配置vlan_centos 7 下多网卡绑定+ vlan 网卡配置
- 基于JAVA+Servlet+JSP+MYSQL的人事管理系统
- 基于c#的windows基础设计(学习日记1)【关于异或运算】
- [导入]在没有 IIS 的条件下运行 ASMX
- 西威变频器avo下载调试资料_变频器设置面板及参数设置方法
- 转载:Java导出数据到Excel
- [人工智能-深度学习-4]:数据流图与正向传播、动态图与静态图
- 区块链 技术 基本概念
- Java实现XML格式化输出
- UE Gameplay Learning Record
- 2022引流新玩法,异业联盟打造商业闭环
- 服务器上Ubuntu系统的java依赖安装
- 如何在业务开发中实现自我成长
- Android:三种Adapter的使用方法
热门文章
- 机器人环境感知算法之算法分析阶段
- linux向上翻页_Linux下vim编辑器命令大全
- 浅谈js与jq给新增元素动态绑定事件(事件委托)
- 实测解决:在此系统上禁止运行脚本。有关详细信息,请参阅 httpsgo.microsoft.comfwlinkLinkID=135170 中的 about_Execution_Policies。
- 文本属性之文本缩进(CSS、HTML)
- 实验2-2-7 整数四则运算 (10 分)
- 微信小程序API之map
- mfc之CPtrArray数组
- c语言3到7位水仙花数流程图_C语言入门基础整理
- 使用SQL语句获得服务器名称和IP 地址