GoLang之使用Context控制請(qǐng)求超時(shí)的實(shí)現(xiàn)

起因

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

Context

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

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

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

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

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

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

超時(shí)取消協(xié)程

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

取消超時(shí)請(qǐng)求的模型

調(diào)度模型

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

超時(shí)之后取消請(qǐng)求

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

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

結(jié)論

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

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