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

起因

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

Context

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

父协程控制所有的子协程

context.Background() context.TODO()

WithCancelWithDeadlineWithTimeoutWithValue

协程间共享数据

  协程间共享数据主要使用的就是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实现超时控制的原理。