起因
之前接觸了一個需求:提供一個接口,這個接口有一個超時時間,如果超時了返回超時異常;這個接口中調用其他的接口,如果調用超時了,所有請求全部結束。
在這個接口中,我使用了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控制請求超時的實現的文章就介紹到這了,更多相關GoLang Context控制請求超時 內容請搜索腳本之家以前的文章或繼續瀏覽下面的相關文章希望大家以后多多支持腳本之家!
您可能感興趣的文章:- golang通過context控制并發的應用場景實現
- GOLANG使用Context實現傳值、超時和取消的方法
- GOLANG使用Context管理關聯goroutine的方法
- 深入Golang之context的用法詳解
- golang中context的作用詳解