GoLang之使用Context控制请求超时的实现

 更新时间:2020年04月02日 09:31:11   作者:MACdress  
这篇文章主要介绍了GoLang之使用Context控制请求超时的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧

起因

  之前接触了一个需求:提供一个接口,这个接口有一个超时时间,如果超时了返回超时异常;这个接口中调用其他的接口,如果调用超时了,所有请求全部结束。
  在这个接口中,我使用了go协程去调用其他接口,所以不仅涉及到请求的超时控制,而且还涉及到父协程对子协程的控制问题。在翻阅了一些资料之后,了解到了Context的基本知识。

Context

  Context是golang.org.pkg下的一个包,类型是接口类型。主要功能有

父协程控制所有的子协程

  Context可以通过context.Background()或者 context.TODO()创建一个空的context。两个区别在于TODO


context可以进行派生,创建出子context。context有四种不同的子context:
  (1)WithCancel:方法入参是一个context;返回值是一个带有新Done的父context的副本,以及cancel函数。当调用cancel函数时,通道将被关闭。关闭规则:会先关闭内部的接收通道;通道关闭了接收该通道的操作会立即返回(即done返回的通道),并且context会向它所有的子值传递信号,如果子context还有子context,那这个撤销信号就会一级一级传递下去。最后这个context会断开其与父context的连接。
  (2)WithDeadlineWithTimeout(本次问题解决就使用的是这个):WithDeadline或者WithTimeout的功能极为相似。都是返回可以被撤销的Context子值。它们不但可以被手动撤销,还会依据在生成是给定的过期时间,自动地进行定时撤销。
  WithDeadline是设置一个时间点,时间对上了就到期。WithTimeout是设置一段时间,比如几秒,过个这段时间,就超时。其实底层的WithTimeout也是通过WithDeadline实现的。WithTimeout的调用就等于WithDeadline(parent, time.Now().Add(timeout))(其中parent是父级context)
  (3)WithValue:入参是父级parent,存储的键和存储的值。返回的是一个带有数据的Context。这个Context是不能被撤销的。撤销的信号在传递的时候会跳过这个Context。

协程间共享数据

  协程间共享数据主要使用的就是WithValue生成的子Context,这个Context存的值在其他的协程中也能读取到。可以用做数据的共享。

超时取消协程

  主要用到的是WithDeadline生成的子Context以及Go中HttpClient请求中的context字段(下文会有描述)

取消超时请求的模型

调度模型


  

其中,对于超时的判断,是根据Context中的Done管道判断的。如果超时了,则Done管道可以拿到东西。

超时之后取消请求

  使用http.NewRequest方法获取到的req,可以调用WithContext将定义好的WithTimeout类型的context放进去,之后调用&htto.Client{}.Do()方法即可。网上有一些博客中让手动调用transport中的CancelRequest方法,但是这个方法已经不被建议使用了。因为它不能取消Http/2的请求。

 现在在代码中有一个私有化的roundTrip方法,会调用CancelRequest调用的cancelRequest方法。而这个roundTrip在transport中会在外面包一层RoundTrip方法,之后交给Client中的send方法进行调用。(具体可以进行源码的查阅)。所以现在通过Client的Do方法,可以自动完成请求的超时控制。

结论

  该调度模型亲测之后,确实可以实现请求的超时控制。只要在最外层设置超时时间时30s,只要过了30s,所有协程中的请求都会结束,对应的协程也会相应的结束;加上Client.Do方法,将超时控制变的更加简洁,后续会写专门写一篇关于http中Client的博客,详细讲解一下Client实现超时控制的原理。

到此这篇关于GoLang之使用Context控制请求超时的实现的文章就介绍到这了,更多相关GoLang Context控制请求超时 内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • golang网络通信超时设置方式

    golang网络通信超时设置方式

    这篇文章主要介绍了golang网络通信超时设置方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-12-12
  • Golang实现请求限流的几种办法(小结)

    Golang实现请求限流的几种办法(小结)

    这篇文章主要介绍了Golang实现请求限流的几种办法(小结),文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2019-10-10
  • go如何利用orm简单实现接口分布式锁

    go如何利用orm简单实现接口分布式锁

    本篇文章主要介绍了go如何利用orm简单实现接口分布式锁,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2018-07-07
  • golang日志包logger的用法详解

    golang日志包logger的用法详解

    这篇文章主要介绍了golang日志包logger的用法详解,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2021-05-05
  • 使用os包和flag包实现读取main命令入参

    使用os包和flag包实现读取main命令入参

    这篇文章主要介绍了使用os包和flag包实现读取main命令入参的相关资料,需要的朋友可以参考下
    2015-03-03
  • go for range坑和闭包坑的分析

    go for range坑和闭包坑的分析

    今天小编就为大家分享一篇关于go for range坑和闭包坑的分析,小编觉得内容挺不错的,现在分享给大家,具有很好的参考价值,需要的朋友一起跟随小编来看看吧
    2019-04-04
  • Go语言中io.Reader和io.Writer的详解与实现

    Go语言中io.Reader和io.Writer的详解与实现

    在Go语言的实际编程中,几乎所有的数据结构都围绕接口展开,接口是Go语言中所有数据结构的核心。在使用Go语言的过程中,无论你是实现web应用程序,还是控制台输入输出,又或者是网络操作,不可避免的会遇到IO操作,使用到io.Reader和io.Writer接口。下面来详细看看。
    2016-09-09
  • Go语言实现二维数组的2种遍历方式以及案例详解

    Go语言实现二维数组的2种遍历方式以及案例详解

    这篇文章主要介绍了Go语言实现二维数组的2种遍历方式以及案例详解,图文代码声情并茂,有感兴趣的可以学习下
    2021-03-03
  • Go语言使用HTTP包创建WEB服务器的方法

    Go语言使用HTTP包创建WEB服务器的方法

    这篇文章主要介绍了Go语言使用HTTP包创建WEB服务器的方法,结合实例形式分析了Go语言基于HTTP包创建WEB服务器客户端与服务器端的实现方法与相关注意事项,需要的朋友可以参考下
    2016-07-07
  • 浅谈Go语言中的结构体struct & 接口Interface & 反射

    浅谈Go语言中的结构体struct & 接口Interface & 反射

    下面小编就为大家带来一篇浅谈Go语言中的结构体struct & 接口Interface & 反射。小编觉得挺不错的,现在就分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2017-07-07

最新评论