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
}