快捷導(dǎo)航

一文帶你搞懂Golang如何正確退出Goroutine

更新時(shí)間：2023年06月18日 10:45:43 作者：humblelei

在Go語言中，Goroutine是一種輕量級(jí)線程，它的退出機(jī)制對(duì)于并發(fā)編程至關(guān)重要，下午就來介紹幾種Goroutine的退出機(jī)制，希望對(duì)大家有所幫助

函數(shù)或方法執(zhí)行完畢

最常見的Goroutine退出方式是函數(shù)或方法執(zhí)行完畢。當(dāng)Goroutine中的函數(shù)或方法執(zhí)行完畢時(shí)，該Goroutine會(huì)自動(dòng)退出。

示例：

package main
import (
	"fmt"
	"time"
)
func main() {
	go func() {
		// Goroutine 中的函數(shù)執(zhí)行完畢后自動(dòng)退出
		fmt.Println("Goroutine execution complete")
	}()
	// 主 Goroutine 繼續(xù)執(zhí)行其他任務(wù)
	time.Sleep(time.Second)
	fmt.Println("Main Goroutine execution complete")
}

在上面的示例中，我們創(chuàng)建了一個(gè)匿名的Goroutine，在其中打印一條消息，然后Goroutine執(zhí)行完畢并自動(dòng)退出。主Goroutine繼續(xù)執(zhí)行其他任務(wù)。

返回語句

在Goroutine中，我們可以使用return語句顯式地退出。當(dāng)執(zhí)行到return語句時(shí)，該Goroutine將會(huì)終止并返回相應(yīng)的結(jié)果（如果有）。

示例：

package main
import (
	"fmt"
	"time"
)
func main() {
	go func() {
		// Goroutine 中使用 return 語句顯式退出
		fmt.Println("Goroutine execution")
		return
	}()
	// 主 Goroutine 繼續(xù)執(zhí)行其他任務(wù)
	time.Sleep(time.Second)
	fmt.Println("Main Goroutine execution complete")
}

在上面的示例中，我們創(chuàng)建了一個(gè)匿名的Goroutine，并在其中打印一條消息，然后使用return語句顯式地退出Goroutine。

runtime.Goexit() 函數(shù)

runtime.Goexit()函數(shù)可以用于直接終止當(dāng)前Goroutine的執(zhí)行，而不影響其他Goroutine。調(diào)用runtime.Goexit()會(huì)立即停止當(dāng)前Goroutine的執(zhí)行，但不會(huì)導(dǎo)致程序退出。

示例：

package main
import (
	"fmt"
	"runtime"
	"time"
)
func main() {
	go func() {
		// 使用 runtime.Goexit() 終止當(dāng)前 Goroutine 的執(zhí)行
		fmt.Println("Goroutine execution before Goexit")
		runtime.Goexit()
		fmt.Println("Goroutine execution after Goexit") // 不會(huì)執(zhí)行到這里
	}()
	// 主 Goroutine 繼續(xù)執(zhí)行其他任務(wù)
	time.Sleep(time.Second)
	fmt.Println("Main Goroutine execution complete")
}

在上面的示例中，我們創(chuàng)建了一個(gè)匿名的Goroutine，并在其中使用runtime.Goexit()函數(shù)終止Goroutine的執(zhí)行。

panic() 和 recover()

如果在Goroutine中發(fā)生了恐慌（panic），默認(rèn)情況下會(huì)終止當(dāng)前Goroutine的執(zhí)行，并向上傳播到調(diào)用棧的上層。我們可以在Goroutine中使用recover()函數(shù)來捕獲并處理恐慌，以避免程序崩潰。

示例：

package main
import (
	"fmt"
	"time"
)
func main() {
	go func() {
		defer func() {
			if err := recover(); err != nil {
				// 在 Goroutine 中捕獲并處理恐慌
				fmt.Println("Recovered from panic:", err)
			}
		}()
		// 引發(fā)恐慌
		panic("Goroutine panic")
	}()
	// 主 Goroutine 繼續(xù)執(zhí)行其他任務(wù)
	time.Sleep(time.Second)
	fmt.Println("Main Goroutine execution complete")
}

在上面的示例中，我們創(chuàng)建了一個(gè)匿名的Goroutine，在其中使用panic()引發(fā)恐慌。然后使用defer和recover()捕獲并處理恐慌，確保程序不會(huì)崩潰。

context.Context 的取消

Go語言的context包提供了一種機(jī)制來管理Goroutine的生命周期，包括取消正在運(yùn)行的Goroutine。我們可以使用context.Context的cancel()方法來取消Goroutine的執(zhí)行。一旦Goroutine接收到取消信號(hào)，它應(yīng)該盡快退出。

示例：

package main
import (
	"context"
	"fmt"
	"time"
)
func worker(ctx context.Context) {
	for {
		select {
		case <-ctx.Done():
			// 接收到取消信號(hào)后退出 Goroutine
			fmt.Println("Goroutine execution canceled")
			return
		default:
			// 執(zhí)行正常任務(wù)
			fmt.Println("Goroutine executing...")
			time.Sleep(time.Second)
		}
	}
}
func main() {
	ctx, cancel := context.WithCancel(context.Background())
	go worker(ctx)
	// 模擬取消任務(wù)
	time.Sleep(3 * time.Second)
	cancel()
	// 主 Goroutine 繼續(xù)執(zhí)行其他任務(wù)
	time.Sleep(time.Second)
	fmt.Println("Main Goroutine execution complete")
}

在上面的示例中，我們創(chuàng)建了一個(gè)worker函數(shù)，在其中使用select和ctx.Done()判斷是否接收到取消信號(hào)。當(dāng)取消信號(hào)到達(dá)時(shí)，Goroutine會(huì)及時(shí)退出。

這篇博客詳細(xì)介紹了Goroutine的退出機(jī)制，包括函數(shù)或方法執(zhí)行完畢、返回語句、runtime.Goexit()函數(shù)、panic()和recover()、以及context.Context的取消機(jī)制。了解和熟練應(yīng)用這些退出機(jī)制對(duì)于編寫可靠的并發(fā)代碼至關(guān)重要。

到此這篇關(guān)于一文帶你搞懂Golang如何正確退出Goroutine的文章就介紹到這了,更多相關(guān)Go正確退出Goroutine內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: