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

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

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

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

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

示例：

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

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

返回語句

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

示例：

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

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

runtime.Goexit() 函數(shù)

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

示例：

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

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

panic() 和 recover()

如果在Goroutine中發(fā)生了恐慌（panic），默認情況下會終止當前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í)行其他任務
	time.Sleep(time.Second)
	fmt.Println("Main Goroutine execution complete")
}

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

context.Context 的取消

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

示例：

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

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

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

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

您可能感興趣的文章: