go 線程同步

1.主協程中創建的協程執行完畢之後再結束主協程

package main

import (
    "fmt"
    "time"
)

func main() {
    go func() {
        fmt.Println("Goroutine 1")
    }()

    go func() {
        fmt.Println("Goroutine 2")
    }()

    time.Sleep(time.Second * 1) // 睡眠1秒,等待上面兩個協程結束
}

2.如果這兩個協程中包含複雜的操作,可能很耗時間,就無法確定需要睡眠多久,當然可以用管道實現同步:

package main

import (
    "fmt"
)

func main() {

    ch := make(chan struct{})
    count := 2 // count 表示活動的協程個數

    go func() {
        fmt.Println("Goroutine 1")
        ch <- struct{}{} // 協程結束,發出信號
    }()

    go func() {
        fmt.Println("Goroutine 2")
        ch <- struct{}{} // 協程結束,發出信號
    }()

    for range ch {
        // 每次從ch中接收數據,表明一個活動的協程結束
        count--
        // 當所有活動的協程都結束時,關閉管道
        if count == 0 {
            close(ch)
        }
    }
}

3.Go提供了更簡單的方法——使用sync.WaitGroup

WaitGroup內部實現了一個計數器,用來記錄未完成的操作個數,它提供了三個方法,Add()用來添加計數。Done()用來在操作結束時調用,使計數減一。Wait()用來等待所有的操作結束,即計數變爲0,該函數會在計數不爲0時等待,在計數爲0時立即返回。

package main

import (
	"fmt"
	"sync"
)

func main() {

	var wg sync.WaitGroup

	wg.Add(2) // 因爲有兩個動作,所以增加2個計數
	go func() {
		fmt.Println("Goroutine 1")
		wg.Done() // 操作完成,減少一個計數
	}()

	go func() {
		fmt.Println("Goroutine 2")
		wg.Done() // 操作完成,減少一個計數
	}()

	wg.Wait() // 等待,直到計數爲0
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章