起因

  之前接觸了一個需求：提供一個接口，這個接口有一個超時時間，如果超時了返回超時異常；這個接口中調(diào)用其他的接口，如果調(diào)用超時了，所有請求全部結(jié)束。
  在這個接口中，我使用了go協(xié)程去調(diào)用其他接口，所以不僅涉及到請求的超時控制，而且還涉及到父協(xié)程對子協(xié)程的控制問題。在翻閱了一些資料之后，了解到了Context的基本知識。

Context

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

父協(xié)程控制所有的子協(xié)程

  Context可以通過context.Background()或者 context.TODO()創(chuàng)建一個空的context。兩個區(qū)別在于TODO

context可以進(jìn)行派生，創(chuàng)建出子context。context有四種不同的子context：
  （1）WithCancel：方法入?yún)⑹且粋€context；返回值是一個帶有新Done的父context的副本，以及cancel函數(shù)。當(dāng)調(diào)用cancel函數(shù)時，通道將被關(guān)閉。關(guān)閉規(guī)則：會先關(guān)閉內(nèi)部的接收通道；通道關(guān)閉了接收該通道的操作會立即返回（即done返回的通道），并且context會向它所有的子值傳遞信號，如果子context還有子context，那這個撤銷信號就會一級一級傳遞下去。最后這個context會斷開其與父context的連接。
  （2）WithDeadline和WithTimeout（本次問題解決就使用的是這個）：WithDeadline或者WithTimeout的功能極為相似。都是返回可以被撤銷的Context子值。它們不但可以被手動撤銷，還會依據(jù)在生成是給定的過期時間，自動地進(jìn)行定時撤銷。
  WithDeadline是設(shè)置一個時間點(diǎn)，時間對上了就到期。WithTimeout是設(shè)置一段時間，比如幾秒，過個這段時間，就超時。其實(shí)底層的WithTimeout也是通過WithDeadline實(shí)現(xiàn)的。WithTimeout的調(diào)用就等于WithDeadline(parent, time.Now().Add(timeout))（其中parent是父級context）
  （3）WithValue:入?yún)⑹歉讣塸arent，存儲的鍵和存儲的值。返回的是一個帶有數(shù)據(jù)的Context。這個Context是不能被撤銷的。撤銷的信號在傳遞的時候會跳過這個Context。

協(xié)程間共享數(shù)據(jù)

  協(xié)程間共享數(shù)據(jù)主要使用的就是WithValue生成的子Context，這個Context存的值在其他的協(xié)程中也能讀取到?？梢杂米鰯?shù)據(jù)的共享。

超時取消協(xié)程

  主要用到的是WithDeadline生成的子Context以及Go中HttpClient請求中的context字段（下文會有描述）

取消超時請求的模型

調(diào)度模型

其中，對于超時的判斷，是根據(jù)Context中的Done管道判斷的。如果超時了，則Done管道可以拿到東西。

超時之后取消請求

  使用http.NewRequest方法獲取到的req，可以調(diào)用WithContext將定義好的WithTimeout類型的context放進(jìn)去，之后調(diào)用&htto.Client{}.Do()方法即可。網(wǎng)上有一些博客中讓手動調(diào)用transport中的CancelRequest方法，但是這個方法已經(jīng)不被建議使用了。因?yàn)樗荒苋∠鸋ttp/2的請求。

 現(xiàn)在在代碼中有一個私有化的roundTrip方法，會調(diào)用CancelRequest調(diào)用的cancelRequest方法。而這個roundTrip在transport中會在外面包一層RoundTrip方法，之后交給Client中的send方法進(jìn)行調(diào)用。（具體可以進(jìn)行源碼的查閱）。所以現(xiàn)在通過Client的Do方法，可以自動完成請求的超時控制。

結(jié)論

  該調(diào)度模型親測之后，確實(shí)可以實(shí)現(xiàn)請求的超時控制。只要在最外層設(shè)置超時時間時30s，只要過了30s，所有協(xié)程中的請求都會結(jié)束，對應(yīng)的協(xié)程也會相應(yīng)的結(jié)束；加上Client.Do方法，將超時控制變的更加簡潔，后續(xù)會寫專門寫一篇關(guān)于http中Client的博客，詳細(xì)講解一下Client實(shí)現(xiàn)超時控制的原理。

到此這篇關(guān)于GoLang之使用Context控制請求超時的實(shí)現(xiàn)的文章就介紹到這了,更多相關(guān)GoLang Context控制請求超時 內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評論