如何實現子goroutine與主線程的同步
1、第一種方式：time.sleep():
package main
import (
"fmt"
"time"
)
func printCount() {
for i := 0; i < 10; i++ {
fmt.Println(i)
}
}
func main() {
go printCount()
time.Sleep(time.Millisecond * 100)
}
這種方式沒法正確預估goroutine執行的時間，很死板，不推薦使用。
2、第二種方式：使用channel機制，每個goroutine傳一個channel進去然后往里寫數據，在再主線程中讀取這些channel，直到全部讀到數據了子goroutine也就全部運行完了，那么主goroutine也就可以結束了。這種模式是子線程去通知主線程結束。
package main
import (
"fmt"
)
func printCount(rs chan int) {
defer close(rs)
for i := 0; i < 10; i++ {
rs <- i
}
}
func main() {
rs := make(chan int, 10)
go printCount(rs)
for v := range rs {
fmt.Println(v)
}
}
3、第三種方式：使用context中cancel函數，這種模式是主線程去通知子線程結束。
package main
import (
"context"
"fmt"
)
func printCount(ctx context.Context) chan int {
dst := make(chan int)
n := 0
go func() {
for {
select {
case <-ctx.Done():
return
case dst <- n:
n++
}
}
}()
return dst
}
func main() {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
intChan := printCount(ctx)
for n := range intChan {
if n == 9 {
break
}
fmt.Println(n)
}
}
4、第四種方式：sync.WaitGroup模式，Add方法設置等待子goroutine的數量，使用Done方法設置等待子goroutine的數量減1，當等待的數量等于0時，Wait函數返回。
package main
import (
"fmt"
"sync"
)
func printCount(wgp *sync.WaitGroup) {
defer wgp.Done()
for i := 0; i < 10; i++ {
fmt.Println(i)
}
}
func main() {
wgp := &sync.WaitGroup{}
wgp.Add(1)
go printCount(wgp)
wgp.Wait()
}
案例一:
- 問題描述 使用兩個 goroutine 交替打印序列，一個 goroutinue 打印數字， 另外一個goroutine打印字母， 最終效果如下 12AB34CD56EF78GH910IJ 。
- 解題思路 問題很簡單，使用 channel來控制打印的進度。使用兩個channel ，來分別控制數字和字母的打印序列， 數字打印完成后通過 channel 通知字母打印, 字母打印完成后通知數字打印，然后周而復始的工作。
- 代碼實現
package main
import (
"fmt"
"runtime"
)
func main() {
runtime.GOMAXPROCS(runtime.NumCPU())
chan_n := make(chan bool)
chan_c := make(chan bool, 1)
done := make(chan struct{})
go func() {
for i := 1; i < 11; i += 2 {
<-chan_c
fmt.Printf("%d", i)
fmt.Printf("%d", i+1)
chan_n <- true
}
}()
go func() {
char_seq := [...]string{"A", "B", "C", "D", "E", "F", "G", "H", "I", "J"}
for i := 0; i < 10; i += 2 {
<-chan_n
fmt.Printf("%s", char_seq[i])
fmt.Printf("%s", char_seq[i+1])
chan_c <- true
}
done <- struct{}{}
}()
chan_c <- true
<-done
}
代碼執行結果:
12AB34CD56EF78GH910IJ
聲明：文章來源多來自互聯網，如果您感覺文章侵犯您的個人著作權，請聯系我們，會在第一時間刪除。
來源：https://studygolang.com/articles/10786
關注小編微信：grey0805，了解更多哦！
