GoLang之使用Context控制請求超時的實(shí)現(xiàn)
起因
之前接觸了一個需求：提供一個接口，這個接口有一個超時時間，如果超時了返回超時異常；這個接口中調(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)文章希望大家以后多多支持腳本之家！
相關(guān)文章
這篇文章主要介紹了使用Gomock進(jìn)行單元測試的方法示例，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-11-11
Go語言常見錯誤之濫用getters/setters誤區(qū)實(shí)例探究
在Go語言編程中,恰如其分地使用getters和setters是至關(guān)重要的,過度和不適當(dāng)?shù)厥褂盟鼈兛赡軐?dǎo)致代碼冗余、可讀性差和封裝不當(dāng),在本文中,我們將深入探討如何識別濫用getter和setter的情況,以及如何采取最佳實(shí)踐來避免這些常見的Go錯誤2024-01-01
這篇文章主要為大家介紹了Go語言教程關(guān)于GC垃圾回收三色標(biāo)記的示例詳解，本篇文章是Go語言七篇入門教程系列文章，有需要的朋友可以借鑒參考下，希望能夠有所幫助2021-11-11
Golang設(shè)計(jì)模式之生成器模式講解和代碼示例
生成器是一種創(chuàng)建型設(shè)計(jì)模式，使你能夠分步驟創(chuàng)建復(fù)雜對象，與其他創(chuàng)建型模式不同，生成器不要求產(chǎn)品擁有通用接口，這使得用相同的創(chuàng)建過程生成不同的產(chǎn)品成為可能，本文就通過代碼示例為大家詳細(xì)介紹Golang生成器模式，感興趣的同學(xué)可以參考下2023-06-06
這篇文章主要介紹了詳解Golang使用MongoDB通用操作,本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2020-12-12
這篇文章主要介紹了go語言 bool類型的使用操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-12-12
Go語言實(shí)現(xiàn)一個Http Server框架(二) Server的抽象
上一篇文章對http庫的基本使用做了說明，這篇文章主要介紹了如何實(shí)現(xiàn)一個簡單地httpServer，文中代碼示例非常詳細(xì)，感興趣的朋友可以參考下2023-04-04
protobuf是Google公司提出的一種輕便高效的結(jié)構(gòu)化數(shù)據(jù)存儲格式,常用于結(jié)構(gòu)化數(shù)據(jù)的序列化,具有語言無關(guān)、平臺無關(guān)、可擴(kuò)展性特性,常用于通訊協(xié)議、服務(wù)端數(shù)據(jù)交換場景,下面我們就來看看golang中protobuf的具體使用吧2023-10-10
這篇文章主要介紹了深入Golang的接口interface，go不要求類型顯示地聲明實(shí)現(xiàn)了哪個接口，只要實(shí)現(xiàn)了相關(guān)的方法即可，編譯器就能檢測到，接下來關(guān)于接口interface的相關(guān)介紹需要的朋友可以參考下面文章內(nèi)容2022-06-06
Golang HTTP請求Json響應(yīng)解析方法以及解讀失敗的原因
這篇文章主要介紹了Golang HTTP請求Json響應(yīng)解析方法以及解讀失敗的原因,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-03-03