GoLang之使goroutine停止的5種方法

1.goroutine停止介紹

goroutine是Go語(yǔ)言實(shí)現(xiàn)并發(fā)編程的利器，簡(jiǎn)單的一個(gè)指令go function就能啟動(dòng)一個(gè)goroutine；

但是，Go語(yǔ)言并沒(méi)有提供終止goroutine的接口，也就是說(shuō)，我們不能從外部去停止一個(gè)goroutine，只能由goroutine內(nèi)部退出(main函數(shù)終止除外)；

我們有很多情況下需要主動(dòng)關(guān)閉goroutine，如需要實(shí)現(xiàn)一個(gè)系統(tǒng)自動(dòng)熔斷的功能就需要主動(dòng)關(guān)閉goroutine

2.goroutine停止的5種方法

2.1使用for-range

for-range從channel上接收值，直到channel關(guān)閉，該結(jié)構(gòu)在Go并發(fā)編程中很常用，這對(duì)于從單一通道上獲取數(shù)據(jù)去執(zhí)行某些任務(wù)是十分方便的

2.2使用for-select(向退出通道發(fā)出退出信號(hào))

當(dāng)channel比較多時(shí)，for-range結(jié)構(gòu)借不是很方便了；

Go語(yǔ)言提供了另外一種和channel相關(guān)的語(yǔ)法: select；

select能夠讓goroutine在多個(gè)通信操作上等待(可以理解為監(jiān)聽(tīng)多個(gè)channel)；

由于這個(gè)特性，for-select結(jié)構(gòu)在Go并發(fā)編程中使用的頻率很高；

我在使用Go的開(kāi)發(fā)中，這是我用的最多的一種組合形式:

for {<!--{C}%3C!%2D%2D%20%2D%2D%3E-->
select {<!--{C}%3C!%2D%2D%20%2D%2D%3E-->
}
}

對(duì)于for-select結(jié)構(gòu)，一般我會(huì)定義一個(gè)特定的退出通道，用于接收退出的信號(hào)，如quit

2.3使用for-select(關(guān)閉退出通道)

當(dāng)我們就需要向quit通道中發(fā)送100次數(shù)據(jù)，如果再用以上的代碼就很麻煩，有一個(gè)很簡(jiǎn)單的方法，關(guān)閉channel，這樣所有監(jiān)聽(tīng)quit channel的goroutine就都會(huì)收到關(guān)閉信號(hào),上面的代碼只要做一個(gè)很小的替換就能工作

2.4使用for-select(關(guān)閉多個(gè)channel)

如果select上監(jiān)聽(tīng)了多個(gè)通道，需要所有的通道都關(guān)閉后才能結(jié)束goroutine，這里就利用select的一個(gè)特性，select不會(huì)在nil的通道上進(jìn)行等待，因此將channel賦值為nil即可，此外，還需要利用channel的ok值

var wg sync.WaitGroup
func worker(in1, in2 <-chan int) {
	defer wg.Done()
	for {
		select {
		case v, ok := <-in1:
			if !ok {
				fmt.Println("收到退出信號(hào)")
				in1 = nil
			}
			// do something
			fmt.Println(v)
		case v, ok := <-in2:
			if !ok {
				fmt.Println("收到退出信號(hào)")
				in2 = nil
			}
			// do something
			fmt.Println(v)
		}
		// select已經(jīng)結(jié)束，我們需要判斷兩個(gè)通道的狀態(tài)
		// 都為nil則結(jié)束當(dāng)前goroutine
		if in1 == nil && in2 == nil {
			return
		}
	}
}
func main() {
	in1 := make(chan int) // 退出通道，接收
	in2 := make(chan int)
	wg.Add(2)
	go worker(in1, in2)
	go worker(in2, in2)
	for i := 0; i < 3; i++ {
		in1 <- i
		time.Sleep(1 * time.Second)
		in2 <- i
	}
	close(in1)
	close(in2)
	wg.Wait()
}

2.5使用context包

context包是官方提供的一個(gè)用于控制多個(gè)goroutine寫(xiě)作的包；

使用context的cancel信號(hào)，可以終止goroutine的運(yùn)行，context是可以向下傳遞的

總結(jié)

到此這篇關(guān)于GoLang使goroutine停止的五種方法的文章就介紹到這了,更多相關(guān)GoLang goroutine停止內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論