詳解Golang中Context的三個(gè)常見應(yīng)用場(chǎng)景
超時(shí)取消
假設(shè)我們希望HTTP請(qǐng)求在給定時(shí)間內(nèi)完成,超時(shí)自動(dòng)取消。
首先定義超時(shí)上下文,設(shè)定時(shí)間返回取消函數(shù)(一旦超時(shí)用于清理資源)。調(diào)用取消函數(shù)取消后續(xù)操作,刪除子上下文對(duì)父的引用。
ctx, cancel := context.WithTimeout(context.Background(), time.Millisecond*80)
defer cancel()
req = req.WithContext(ctx)
還可以通過特定時(shí)間進(jìn)行設(shè)定:
/ The context will be cancelled after 3 seconds // If it needs to be cancelled earlier, the `cancel` function can // be used, like before ctx, cancel := context.WithTimeout(ctx, 3*time.Second) ???????// Setting a context deadline is similar to setting a timeout, except // you specify a time when you want the context to cancel, rather than a duration. // Here, the context will be cancelled on 2022-11-10 23:00:00 ctx, cancel := context.WithDeadline(ctx, time.Date(2022, time.November, 10, 23, 0, 0, 0, time.UTC))
完整實(shí)例如下:
func main() {
//定義請(qǐng)求
req, err := http.NewRequest(http.MethodGet, "https://www.baidu.com", nil)
if err != nil {
log.Fatal(err)
}
// 定義上下文
ctx, cancel := context.WithTimeout(context.Background(), time.Millisecond*80)
defer cancel()
req = req.WithContext(ctx)
// 執(zhí)行請(qǐng)求
c := &http.Client{}
res, err := c.Do(req)
if err != nil {
log.Fatal(err)
}
defer res.Body.Close()
// 輸出日志
out, err := io.ReadAll(res.Body)
if err != nil {
log.Fatal(err)
}
log.Println(string(out))
}超時(shí)輸出結(jié)果:
2022/12/27 14:36:00 Get "https://www.baidu.com": context deadline exceeded
我們可以調(diào)大超時(shí)時(shí)間,則能正常看到輸出結(jié)果。
取消后續(xù)操作
有時(shí)請(qǐng)求被取消后,需要阻止系統(tǒng)繼續(xù)做后續(xù)比必要的工作。請(qǐng)看下面用戶發(fā)起的http請(qǐng)求,應(yīng)用程序接收請(qǐng)求后查詢數(shù)據(jù)庫并返回查詢結(jié)果:
正常流程如下:

但如果客戶端取消了請(qǐng)求,如果沒有取消,應(yīng)用服務(wù)和數(shù)據(jù)庫仍然繼續(xù)工作,然后結(jié)果卻不能反饋給客戶端。理想狀況為所有下游過程停止,如圖所示:

考慮有兩個(gè)相關(guān)操作的情況,“相關(guān)”的意思是如果一個(gè)失敗了,另一個(gè)即使完成也沒有意義了。如果已經(jīng)知道前一個(gè)操作失敗了,則希望取消所有相關(guān)的操作。請(qǐng)看示例:
func operation1(ctx context.Context) error {
// 假設(shè)該操作因某種原因而失敗
// 下面模擬業(yè)務(wù)執(zhí)行一定時(shí)間
time.Sleep(100 * time.Millisecond)
return errors.New("failed")
}
func operation2(ctx context.Context) {
// 該方法要么正常執(zhí)行完成
// 要么取消,不再繼續(xù)執(zhí)行
select {
case <-time.After(500 * time.Millisecond):
fmt.Println("done")
case <-ctx.Done():
fmt.Println("halted operation2")
}
}
func main() {
// 創(chuàng)建上下文
ctx := context.Background()
// 基于上下文創(chuàng)建需求上下文
ctx, cancel := context.WithCancel(ctx)
// 在不同協(xié)程中執(zhí)行兩個(gè)操作
go func() {
err := operation1(ctx)
// 如果該方法返回錯(cuò)誤,則取消該上下文中的后續(xù)操作
if err != nil {
cancel()
}
}()
// 實(shí)用相同上下文執(zhí)行操作2
operation2(ctx)
}
由于我們?cè)O(shè)置操作2執(zhí)行時(shí)間較長,而操作1很快就報(bào)錯(cuò),因此輸出結(jié)果為操作2被取消:
halted operation2
上下文傳值
我們可以實(shí)用上下文變量在不同協(xié)程中傳遞值。

假設(shè)一個(gè)操作需要調(diào)用函數(shù)多次,其中用于標(biāo)識(shí)的公共ID需要被 日志記錄,請(qǐng)看示例:
// 定義key,用于保存上下文值的鍵
const keyID = "id"
func main() {
// 定義上下文值
rand.Seed(time.Now().Unix())
ctx := context.WithValue(context.Background(), keyID, rand.Int())
operation1(ctx)
}
func operation1(ctx context.Context) {
// do some work
// we can get the value from the context by passing in the key
log.Println("operation1 for id:", ctx.Value(keyID), " completed")
operation2(ctx)
}
func operation2(ctx context.Context) {
// do some work
// this way, the same ID is passed from one function call to the next
log.Println("operation2 for id:", ctx.Value(keyID), " completed")
}
這里在main函數(shù)中創(chuàng)建上下文,并采用鍵值對(duì)方式存儲(chǔ)id值,從而后續(xù)函數(shù)調(diào)用時(shí)可以從上下文中獲取該值。如圖所示:
使用context變量在不同操作中傳遞信息非常有用,主要原因包括:
- 線程安全: 一旦設(shè)置了上下文鍵,就不能修改它的值,可以使用context.WithValue方法可以設(shè)置新的值
- 通用方法: 在Go的官方庫和應(yīng)用程序中大量使用上下文傳遞數(shù)據(jù),我們當(dāng)然最好也使用這種模式
到此這篇關(guān)于詳解Golang中Context的三個(gè)常見應(yīng)用場(chǎng)景的文章就介紹到這了,更多相關(guān)Golang Context內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
- 深入Golang之context的用法詳解
- GOLANG使用Context實(shí)現(xiàn)傳值、超時(shí)和取消的方法
- GoLang之使用Context控制請(qǐng)求超時(shí)的實(shí)現(xiàn)
- golang中context的作用詳解
- golang通過context控制并發(fā)的應(yīng)用場(chǎng)景實(shí)現(xiàn)
- 提升Golang應(yīng)用性能:深入理解Context的應(yīng)用
- golang context接口類型方法介紹
- golang中context使用小結(jié)
- 詳解Golang中Context的原理和使用技巧
- 深入理解Golang中Context的使用
相關(guān)文章
Go語言實(shí)現(xiàn)AzDG可逆加密算法實(shí)例
這篇文章主要介紹了Go語言實(shí)現(xiàn)AzDG可逆加密算法,實(shí)例分析了AzDG可逆加密算法的實(shí)現(xiàn)技巧,具有一定參考借鑒價(jià)值,需要的朋友可以參考下2015-02-02
go語言中的udp協(xié)議及TCP通訊實(shí)現(xiàn)示例
這篇文章主要為大家介紹了go語言中的udp協(xié)議及TCP通訊的實(shí)現(xiàn)示例,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步早日升職加薪2022-04-04
探索Golang實(shí)現(xiàn)Redis持久化AOF實(shí)例
這篇文章主要為大家介紹了Golang實(shí)現(xiàn)Redis持久化AOF實(shí)例探索,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2024-01-01
Goland調(diào)節(jié)字體大小的設(shè)置(編輯區(qū),terminal區(qū),頁面字體)
這篇文章主要介紹了Goland調(diào)節(jié)字體大小的設(shè)置(編輯區(qū),terminal區(qū),頁面字體),具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2020-12-12
Go標(biāo)準(zhǔn)庫bytes|strings使用解析
本文介紹了Go語言中`strings`和`bytes`庫的基本功能、設(shè)計(jì)思想、核心函數(shù)以及應(yīng)用場(chǎng)景,這兩個(gè)庫提供了豐富的工具函數(shù)來處理字符串和字節(jié)切片,適用于各種字符串操作需求,通過對(duì)比兩者的性能和適用場(chǎng)景,可以幫助開發(fā)者更好地選擇合適的工具庫來提高代碼效率和性能2026-01-01
Go?Map從數(shù)據(jù)結(jié)構(gòu)到核心機(jī)制實(shí)現(xiàn)原理解析
Go語言中的map是一種高效的內(nèi)置數(shù)據(jù)結(jié)構(gòu),用于存儲(chǔ)鍵值對(duì)(key-value?pairs),本文將深入解析?Go?map?的實(shí)現(xiàn)原理,涵蓋數(shù)據(jù)結(jié)構(gòu)、哈希沖突處理、負(fù)載因子、擴(kuò)容機(jī)制、查找和插入操作等關(guān)鍵技術(shù)細(xì)節(jié),感興趣的朋友跟隨小編一起看看吧2021-05-05

