GoLang使用Context怎么控制請求超時

本篇內容主要講解“GoLang使用Context怎么控制請求超時”，感興趣的朋友不妨來看看。本文介紹的方法操作簡單快捷，實用性強。下面就讓小編來帶大家學習“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實現超時控制的原理。

到此，相信大家對“GoLang使用Context怎么控制請求超時”有了更深的了解，不妨來實際操作一番吧！這里是億速云網站，更多相關內容可以進入相關頻道進行查詢，關注我們，繼續學習！