Go使用context控制協(xié)程取消的實戰(zhàn)案例
在并發(fā)編程中,合理地控制協(xié)程(goroutine)的生命周期是保證程序穩(wěn)定性和資源可控使用的關(guān)鍵。Go語言標(biāo)準(zhǔn)庫中的 context 包正是為了解決這一問題而生。它為我們提供了取消信號、超時控制、請求作用域的值傳遞等功能。
本文將通過一個實際案例,演示如何使用 context 控制協(xié)程的取消,避免資源泄露,實現(xiàn)優(yōu)雅退出。
一、什么是 context
context 是 Go 1.7 起加入標(biāo)準(zhǔn)庫的一個重要包,用于跨 API 邊界傳遞取消信號、超時時間、截止時間等信息。
主要接口定義如下:
type Context interface {
Deadline() (deadline time.Time, ok bool)
Done() <-chan struct{}
Err() error
Value(key any) any
}
其中最關(guān)鍵的是:
Done():返回一個 channel,當(dāng) context 被取消或超時關(guān)閉時,該 channel 會被關(guān)閉;Err():返回取消的原因,例如context.Canceled或context.DeadlineExceeded。
二、常見創(chuàng)建方式
Go 提供了以下常用方式創(chuàng)建 context:
ctx := context.Background() // 最頂層、永不取消的 context ctx, cancel := context.WithCancel(parent) // 手動調(diào)用 cancel() 取消 ctx, cancel := context.WithTimeout(parent, 3*time.Second) // 指定超時時間 ctx, cancel := context.WithDeadline(parent, time.Now().Add(3*time.Second)) // 到期時間點
這些 context 都可以傳遞到協(xié)程中,通過 ctx.Done() 控制協(xié)程的停止。
三、實戰(zhàn)案例:使用 context 控制任務(wù)協(xié)程
場景描述
假設(shè)我們要運行一個任務(wù),該任務(wù)每秒輸出一次“正在處理”,但當(dāng)主程序在某個時機需要終止它(比如點擊“停止按鈕”或超時),我們要優(yōu)雅地通知協(xié)程退出。
示例代碼
package main
import (
"context"
"fmt"
"time"
)
func worker(ctx context.Context) {
for {
select {
case <-ctx.Done():
fmt.Println("worker 任務(wù)被取消:", ctx.Err())
return
default:
fmt.Println("worker 正在處理任務(wù)...")
time.Sleep(1 * time.Second)
}
}
}
func main() {
ctx, cancel := context.WithCancel(context.Background())
go worker(ctx)
// 主線程運行5秒后取消任務(wù)
time.Sleep(5 * time.Second)
cancel()
// 等待協(xié)程打印結(jié)束語
time.Sleep(1 * time.Second)
fmt.Println("主程序退出")
}
輸出結(jié)果
worker 正在處理任務(wù)... worker 正在處理任務(wù)... worker 正在處理任務(wù)... worker 正在處理任務(wù)... worker 正在處理任務(wù)... worker 任務(wù)被取消: context canceled 主程序退出
可以看到,主程序通過 cancel() 取消了 context,協(xié)程立即響應(yīng)退出了。
四、使用 context.WithTimeout 實現(xiàn)超時控制
除了手動調(diào)用 cancel,我們還可以通過設(shè)定超時時間自動取消任務(wù)。
func main() {
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
defer cancel()
go worker(ctx)
time.Sleep(5 * time.Second) // 主線程等待更久,看任務(wù)是否能自動終止
fmt.Println("主程序退出")
}
運行結(jié)果類似:
worker 正在處理任務(wù)... worker 正在處理任務(wù)... worker 正在處理任務(wù)... worker 任務(wù)被取消: context deadline exceeded 主程序退出
這表示:即使主程序沒有主動調(diào)用 cancel(),協(xié)程也在 3 秒后收到 context 超時通知并正常退出。
五、多個協(xié)程共享一個 context
我們可以啟動多個協(xié)程,并使用同一個 context 控制它們:
func main() {
ctx, cancel := context.WithCancel(context.Background())
for i := 1; i <= 3; i++ {
go func(id int) {
for {
select {
case <-ctx.Done():
fmt.Printf("協(xié)程 %d 接收到取消信號\n", id)
return
default:
fmt.Printf("協(xié)程 %d 正在工作\n", id)
time.Sleep(1 * time.Second)
}
}
}(i)
}
time.Sleep(4 * time.Second)
cancel()
time.Sleep(1 * time.Second)
fmt.Println("主程序退出")
}
輸出:
協(xié)程 1 正在工作 協(xié)程 2 正在工作 協(xié)程 3 正在工作 ...(多次輸出) 協(xié)程 1 接收到取消信號 協(xié)程 2 接收到取消信號 協(xié)程 3 接收到取消信號 主程序退出
這樣我們實現(xiàn)了“一鍵終止所有協(xié)程”。
六、最佳實踐建議
- 永遠(yuǎn)使用
context.WithCancel/WithTimeout返回的cancel()函數(shù),不要忘記defer cancel(); - 在需要可控中止的任務(wù)(如網(wǎng)絡(luò)請求、數(shù)據(jù)庫操作、循環(huán)處理)中傳入 context;
- 對
ctx.Done()的監(jiān)聽要放在協(xié)程內(nèi)部適當(dāng)位置,防止資源泄露; - 在請求鏈中傳遞 context,以實現(xiàn)鏈路級別的取消與超時控制。
七、結(jié)語
context 是 Go 并發(fā)控制中不可或缺的利器。它不僅解決了協(xié)程取消的痛點,還能為任務(wù)設(shè)置統(tǒng)一的生命周期控制邏輯,是構(gòu)建高可靠網(wǎng)絡(luò)服務(wù)、后臺任務(wù)系統(tǒng)、爬蟲等并發(fā)程序的基礎(chǔ)設(shè)施。
如果你還沒有在項目中大量使用它,不妨從今天開始重構(gòu)代碼,使用 context 管理協(xié)程生命周期,讓你的 Go 程序更穩(wěn)定、更可控!
以上就是Go使用context控制協(xié)程取消的實戰(zhàn)案例的詳細(xì)內(nèi)容,更多關(guān)于Go context控制協(xié)程取消的資料請關(guān)注腳本之家其它相關(guān)文章!
相關(guān)文章
為什么Go里值為nil可以調(diào)用函數(shù)原理分析
這篇文章主要為大家介紹了為什么Go里值為nil可以調(diào)用函數(shù)原理分析,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2023-08-08
golang標(biāo)準(zhǔn)庫crc32的使用樣例
根據(jù)實驗我們知道crc32算法比md5算法快4倍左右,所以研究了下golang的crc32使用,這篇文章主要給大家介紹了關(guān)于golang標(biāo)準(zhǔn)庫crc32使用的相關(guān)資料,需要的朋友可以參考下2024-03-03
Golang創(chuàng)建第一個web項目(Gin+Gorm)
本文主要介紹了Golang創(chuàng)建第一個web項目(Gin+Gorm),文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2024-06-06
Go語言Elasticsearch數(shù)據(jù)清理工具思路詳解
這篇文章主要介紹了Go語言Elasticsearch數(shù)據(jù)清理工具思路詳解,本文給大家介紹的非常詳細(xì),對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友可以參考下2021-10-10
go 類型轉(zhuǎn)換方式(interface 類型的轉(zhuǎn)換)
這篇文章主要介紹了go 類型轉(zhuǎn)換方式(interface 類型的轉(zhuǎn)換),具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2021-05-05

