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）WithDeadline和WithTimeout（本次問題解決就使用的是這個）：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實現超時控制的原理。
  若是大佬們發現這個調度模型有什麼問題，或者我用的方法有什麼BUG，歡迎留言指正。（來自Go萌新心裏的渴望）