Go使用context控制協(xié)程取消的實戰(zhàn)案例
在并發(fā)編程中，合理地控制協(xié)程（goroutine）的生命周期是保證程序穩(wěn)定性和資源可控使用的關(guān)鍵。Go語言標準庫中的 context 包正是為了解決這一問題而生。它為我們提供了取消信號、超時控制、請求作用域的值傳遞等功能。
本文將通過一個實際案例，演示如何使用 context 控制協(xié)程的取消，避免資源泄露，實現(xiàn)優(yōu)雅退出。
一、什么是 context
context 是 Go 1.7 起加入標準庫的一個重要包，用于跨 API 邊界傳遞取消信號、超時時間、截止時間等信息。
主要接口定義如下：
type Context interface {
Deadline() (deadline time.Time, ok bool)
Done() <-chan struct{}
Err() error
Value(key any) any
}
其中最關(guān)鍵的是：
Done()：返回一個 channel，當 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ù)每秒輸出一次“正在處理”，但當主程序在某個時機需要終止它（比如點擊“停止按鈕”或超時），我們要優(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é)程”。
六、最佳實踐建議
- 永遠使用
context.WithCancel/WithTimeout返回的cancel()函數(shù)，不要忘記defer cancel()； - 在需要可控中止的任務(wù)（如網(wǎng)絡(luò)請求、數(shù)據(jù)庫操作、循環(huán)處理）中傳入 context；
- 對
ctx.Done()的監(jiān)聽要放在協(xié)程內(nèi)部適當位置，防止資源泄露； - 在請求鏈中傳遞 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)案例的詳細內(nèi)容，更多關(guān)于Go context控制協(xié)程取消的資料請關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
備份mysql數(shù)據(jù)庫的方法有很多,常用的可以登錄mysql活著使用mysqldump命令進行備份,本文提供golang的方式來備份mysql數(shù)據(jù)庫,下面就來詳細的介紹一下2026-06-06
golang通過反射設(shè)置結(jié)構(gòu)體變量的值
這篇文章主要介紹了golang通過反射設(shè)置結(jié)構(gòu)體變量的值操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-04-04
這篇文章主要介紹了Go語言運算符案例講解,本篇文章通過簡要的案例,講解了該項技術(shù)的了解與使用,以下就是詳細內(nèi)容,需要的朋友可以參考下2021-07-07
Swagger是一個基于OpenAPI規(guī)范的API文檔生成工具,它能夠自動生成、描述、調(diào)試和可視化RESTful API文檔,下面小編就為大家詳細講講Gin框架整合Swagger生成接口文檔的完整方法吧2025-09-09
Go語言模型:string的底層數(shù)據(jù)結(jié)構(gòu)與高效操作詳解
這篇文章主要介紹了Go語言模型:string的底層數(shù)據(jù)結(jié)構(gòu)與高效操作詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-12-12
本文主要介紹了Go語言copy()實現(xiàn)切片復(fù)制，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-04-04
這篇文章主要介紹了詳解golang 模板(template)的常用基本語法，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-08-08
這篇文章主要介紹了golang post請求常用的幾種方式小結(jié)，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-04-04
本文主要介紹了Go語言中Pool對象復(fù)用的實現(xiàn),文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2026-04-04
這篇文章主要給大家介紹了關(guān)于go實現(xiàn)圖片拼接與文字書寫的相關(guān)資料,文中通過實例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2022-01-01