快捷導(dǎo)航

使用Go語言中的Context取消協(xié)程執(zhí)行的操作代碼

更新時(shí)間：2024年11月29日 11:08:23 作者：軟件架構(gòu)師筆記

在 Go 語言中,協(xié)程（goroutine）是一種輕量級的線程,非常適合處理并發(fā)任務(wù),然而,如何優(yōu)雅地取消正在運(yùn)行的協(xié)程是一個(gè)常見的問題,本文將通過一個(gè)具體的例子來展示如何使用 context 包來取消協(xié)程的執(zhí)行,需要的朋友可以參考下

使用 Go 語言中的 Context 取消協(xié)程執(zhí)行

在 Go 語言中，協(xié)程（goroutine）是一種輕量級的線程，非常適合處理并發(fā)任務(wù)。然而，如何優(yōu)雅地取消正在運(yùn)行的協(xié)程是一個(gè)常見的問題。本文將通過一個(gè)具體的例子來展示如何使用 context 包來取消協(xié)程的執(zhí)行，特別是處理嵌套任務(wù)中的取消問題。

問題描述

假設(shè)我們有一個(gè)長時(shí)間運(yùn)行的任務(wù)，該任務(wù)包含一個(gè)外層循環(huán)和一個(gè)內(nèi)層任務(wù)。我們需要在外層循環(huán)接收到取消信號時(shí)，能夠立即終止內(nèi)層任務(wù)。以下是一個(gè)示例代碼：

package main

import (
	"context"
	"fmt"
	"time"
)

// longRunningTask 是一個(gè)模擬長時(shí)間運(yùn)行的任務(wù)。
func longRunningTask(ctx context.Context) {
	for {
		select {
		case <-ctx.Done(): // 監(jiān)聽 ctx.Done() 以獲取取消信號
			fmt.Println("任務(wù)被取消:", ctx.Err())
			return // 接收到取消信號后退出
		default:
			currentTime := time.Now().Format("2006-01-02 15:04:05") // 獲取并格式化當(dāng)前時(shí)間
			fmt.Printf("任務(wù)進(jìn)行中... 當(dāng)前時(shí)間：%s\n", currentTime)
			for {
				fmt.Printf("111")
				time.Sleep(1 * time.Second) //
			}
		}
	}
}

func main() {
	// 創(chuàng)建一個(gè)可以取消的 context
	ctx, cancel := context.WithCancel(context.Background())

	// 啟動(dòng)一個(gè)新的 goroutine 執(zhí)行任務(wù)
	go longRunningTask(ctx)

	// 模擬一段時(shí)間后取消任務(wù)
	time.Sleep(3 * time.Second)
	fmt.Println("取消任務(wù)...")
	cancel() // 發(fā)送取消信號

	// 等待一段時(shí)間讓任務(wù)有時(shí)間處理取消信號并退出
	time.Sleep(10 * time.Second)
}

在這個(gè)示例中，當(dāng)我們?nèi)∠蝿?wù)時(shí)，外層循環(huán)會(huì)接收到取消信號并退出，但內(nèi)層循環(huán)會(huì)繼續(xù)運(yùn)行，因?yàn)槲覀儧]有在內(nèi)層循環(huán)中檢查取消信號。

解決方案

為了確保內(nèi)層任務(wù)也能響應(yīng)取消信號，我們需要在內(nèi)層任務(wù)中也檢查 ctx.Done() 通道。以下是修改后的代碼：

package main

import (
	"context"
	"fmt"
	"time"
)

// longRunningTask 是一個(gè)模擬長時(shí)間運(yùn)行的任務(wù)。
func longRunningTask(ctx context.Context) {
	for {
		select {
		case <-ctx.Done(): // 監(jiān)聽 ctx.Done() 以獲取取消信號
			fmt.Println("任務(wù)被取消:", ctx.Err())
			return // 接收到取消信號后退出
		default:
			currentTime := time.Now().Format("2006-01-02 15:04:05") // 獲取并格式化當(dāng)前時(shí)間
			fmt.Printf("任務(wù)進(jìn)行中... 當(dāng)前時(shí)間：%s\n", currentTime)
			// 啟動(dòng)內(nèi)層任務(wù)
			runInnerTask(ctx)
		}
	}
}

// runInnerTask 是一個(gè)模擬內(nèi)層長時(shí)間運(yùn)行的任務(wù)。
func runInnerTask(ctx context.Context) {
	for {
		select {
		case <-ctx.Done(): // 內(nèi)層任務(wù)也監(jiān)聽 ctx.Done()
			fmt.Println("內(nèi)層任務(wù)被取消:", ctx.Err())
			return // 接收到取消信號后退出
		default:
			fmt.Printf("111")
			time.Sleep(1 * time.Second)
		}
	}
}

func main() {
	// 創(chuàng)建一個(gè)可以取消的 context
	ctx, cancel := context.WithCancel(context.Background())

	// 啟動(dòng)一個(gè)新的 goroutine 執(zhí)行任務(wù)
	go longRunningTask(ctx)

	// 模擬一段時(shí)間后取消任務(wù)
	time.Sleep(3 * time.Second)
	fmt.Println("取消任務(wù)...")
	cancel() // 發(fā)送取消信號

	// 等待一段時(shí)間讓任務(wù)有時(shí)間處理取消信號并退出
	time.Sleep(10 * time.Second)
}

解釋

外層循環(huán)：

外層循環(huán)使用 select 語句來監(jiān)聽 ctx.Done() 通道。如果接收到取消信號，任務(wù)會(huì)打印一條消息并退出。

內(nèi)層任務(wù)：

內(nèi)層任務(wù)也使用 select 語句來監(jiān)聽 ctx.Done() 通道。如果接收到取消信號，內(nèi)層任務(wù)會(huì)打印一條消息并退出。

通過這種方式，我們可以確保無論是在外層循環(huán)還是內(nèi)層任務(wù)中，任務(wù)都能響應(yīng)取消信號并優(yōu)雅地退出。

總結(jié)

在 Go 語言中，使用 context 包來管理協(xié)程的生命周期是非常重要的。通過在每個(gè)需要響應(yīng)取消信號的地方檢查 ctx.Done() 通道，我們可以確保任務(wù)能夠及時(shí)響應(yīng)取消信號并優(yōu)雅地退出。這對于構(gòu)建健壯和可靠的并發(fā)應(yīng)用程序至關(guān)重要。

到此這篇關(guān)于使用Go語言中的Context取消協(xié)程執(zhí)行的操作代碼的文章就介紹到這了,更多相關(guān)Go Context取消協(xié)程執(zhí)行內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: