详解Golang中Context的三个常见应用场景
超时取消
假设我们希望HTTP请求在给定时间内完成,超时自动取消。
首先定义超时上下文,设定时间返回取消函数(一旦超时用于清理资源)。调用取消函数取消后续操作,删除子上下文对父的引用。
ctx, cancel := context.WithTimeout(context.Background(), time.Millisecond*80) defer cancel() req = req.WithContext(ctx)
还可以通过特定时间进行设定:
/ The context will be cancelled after 3 seconds // If it needs to be cancelled earlier, the `cancel` function can // be used, like before ctx, cancel := context.WithTimeout(ctx, 3*time.Second) // Setting a context deadline is similar to setting a timeout, except // you specify a time when you want the context to cancel, rather than a duration. // Here, the context will be cancelled on 2022-11-10 23:00:00 ctx, cancel := context.WithDeadline(ctx, time.Date(2022, time.November, 10, 23, 0, 0, 0, time.UTC))
完整实例如下:
func main() { //定义请求 req, err := http.NewRequest(http.MethodGet, "https://www.baidu.com", nil) if err != nil { log.Fatal(err) } // 定义上下文 ctx, cancel := context.WithTimeout(context.Background(), time.Millisecond*80) defer cancel() req = req.WithContext(ctx) // 执行请求 c := &http.Client{} res, err := c.Do(req) if err != nil { log.Fatal(err) } defer res.Body.Close() // 输出日志 out, err := io.ReadAll(res.Body) if err != nil { log.Fatal(err) } log.Println(string(out)) }
超时输出结果:
2022/12/27 14:36:00 Get "https://www.baidu.com": context deadline exceeded
我们可以调大超时时间,则能正常看到输出结果。
取消后续操作
有时请求被取消后,需要阻止系统继续做后续比必要的工作。请看下面用户发起的http请求,应用程序接收请求后查询数据库并返回查询结果:
正常流程如下:
但如果客户端取消了请求,如果没有取消,应用服务和数据库仍然继续工作,然后结果却不能反馈给客户端。理想状况为所有下游过程停止,如图所示:
考虑有两个相关操作的情况,“相关”的意思是如果一个失败了,另一个即使完成也没有意义了。如果已经知道前一个操作失败了,则希望取消所有相关的操作。请看示例:
func operation1(ctx context.Context) error { // 假设该操作因某种原因而失败 // 下面模拟业务执行一定时间 time.Sleep(100 * time.Millisecond) return errors.New("failed") } func operation2(ctx context.Context) { // 该方法要么正常执行完成 // 要么取消,不再继续执行 select { case <-time.After(500 * time.Millisecond): fmt.Println("done") case <-ctx.Done(): fmt.Println("halted operation2") } } func main() { // 创建上下文 ctx := context.Background() // 基于上下文创建需求上下文 ctx, cancel := context.WithCancel(ctx) // 在不同协程中执行两个操作 go func() { err := operation1(ctx) // 如果该方法返回错误,则取消该上下文中的后续操作 if err != nil { cancel() } }() // 实用相同上下文执行操作2 operation2(ctx) }
由于我们设置操作2执行时间较长,而操作1很快就报错,因此输出结果为操作2被取消:
halted operation2
上下文传值
我们可以实用上下文变量在不同协程中传递值。
假设一个操作需要调用函数多次,其中用于标识的公共ID需要被 日志记录,请看示例:
// 定义key,用于保存上下文值的键 const keyID = "id" func main() { // 定义上下文值 rand.Seed(time.Now().Unix()) ctx := context.WithValue(context.Background(), keyID, rand.Int()) operation1(ctx) } func operation1(ctx context.Context) { // do some work // we can get the value from the context by passing in the key log.Println("operation1 for id:", ctx.Value(keyID), " completed") operation2(ctx) } func operation2(ctx context.Context) { // do some work // this way, the same ID is passed from one function call to the next log.Println("operation2 for id:", ctx.Value(keyID), " completed") }
这里在main函数中创建上下文,并采用键值对方式存储id值,从而后续函数调用时可以从上下文中获取该值。如图所示:
使用context变量在不同操作中传递信息非常有用,主要原因包括:
- 线程安全: 一旦设置了上下文键,就不能修改它的值,可以使用context.WithValue方法可以设置新的值
- 通用方法: 在Go的官方库和应用程序中大量使用上下文传递数据,我们当然最好也使用这种模式
到此这篇关于详解Golang中Context的三个常见应用场景的文章就介绍到这了,更多相关Golang Context内容请搜索猪先飞以前的文章或继续浏览下面的相关文章希望大家以后多多支持猪先飞!
原文出处:https://blog.csdn.net/neweastsun/article/details/128458477
相关文章
- 这篇文章主要介绍了golang 调用 php7详解及实例的相关资料,需要的朋友可以参考下...2017-01-15
- 这篇文章主要介绍了用golang实现替换某个文件中的字符串操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-04-25
- 这篇文章主要介绍了golang在GRPC中设置client的超时时间,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-04-27
- 这篇文章主要介绍了解决Golang json序列化字符串时多了\的情况,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-12-24
Golang中的自定义类型之间的转换的实现(type conversion)
这篇文章主要介绍了Golang中的自定义类型之间的转换的实现(type conversion),文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-02-21- 这篇文章主要介绍了golang与php实现计算两个经纬度之间距离的方法,结合实例形式对比分析了Go语言与php进行经纬度计算的相关数学运算技巧,需要的朋友可以参考下...2016-07-29
解决golang处理http response碰到的问题和需要注意的点
这篇文章主要介绍了解决golang处理http response碰到的问题和需要注意的点,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-12-16- 这篇文章主要介绍了golang http使用踩过的坑与填坑指南,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-04-27
- 这篇文章主要介绍了golang文件读取-按指定BUFF大小读取方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-12-22
- Go(又称Golang)是Google开发的一种静态强类型、编译型、并发型,并具有垃圾回收功能的编程语言。这篇文章给大家介绍golang 正则表达式的相关知识,感兴趣的朋友跟随小编一起看看吧...2021-05-07
- 这次文章为大家带来的是一个比较实用的示例:利用Golang生成整数随机数,对此感兴趣的可以一起来看看。 php随机数生成一个给定范围的随机数,用 PHP 就太简单不过了,而...2017-07-06
- 这篇文章主要介绍了golang DNS服务器的简单实现操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-05-01
- 这篇文章主要介绍了golang中json和struct的使用说明,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-05-01
Fedora14 Linux系统安装Golang开发环境笔记
这篇文章主要介绍了Fedora14 Linux系统安装Golang开发环境笔记,本文讲解了2种安装方法,需要的朋友可以参考下...2020-05-01- stream_context_create创建并返回一个文本数据流并应用各种选项,可用于fopen(),file_get_contents()等过程的超时设置、代理服务器、请求方式、头信息设置的特殊过程。...2016-11-25
- 这篇文章主要介绍了golang去除多余空白字符(含制表符)的操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-04-25
golang elasticsearch Client的使用详解
这篇文章主要介绍了golang elasticsearch Client的使用详解,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-05-04- 这篇文章主要介绍了解决golang json解析出现值为空的问题,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-12-24
- 这篇文章主要介绍了golang中的空接口使用,本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下...2021-03-30
- 这篇文章主要介绍了Golang Cron 定时任务的实现示例,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2020-05-11