快捷導(dǎo)航

Go語言同步機(jī)制的幾種常用方法

更新時(shí)間：2025年11月02日 10:13:55 作者：li_zehao

本文介紹了如何在Go語言中通過通道、sync包中的WaitGroup和互斥鎖來保證多協(xié)程的有序執(zhí)行,以及它們各自在不同場景下的應(yīng)用,具有一定的參考價(jià)值,感興趣的可以了解一下

在 Go 語言中，保證多線程（或者更準(zhǔn)確地說，多協(xié)程）有序執(zhí)行，主要依賴于協(xié)程間的同步機(jī)制。Go 提供了幾種工具來幫助開發(fā)者控制協(xié)程（goroutine）的執(zhí)行順序，確保數(shù)據(jù)的一致性和操作的原子性。下面是一些常用的同步機(jī)制：

1. 使用通道（Channel）

通道是 Go 中協(xié)程間通信的主要方式，通過通道可以安全地在協(xié)程之間傳遞數(shù)據(jù)，從而控制執(zhí)行順序。

示例：使用無緩沖通道實(shí)現(xiàn)同步
無緩沖通道保證發(fā)送操作在接收準(zhǔn)備好之前是阻塞的，這可以用來控制協(xié)程的執(zhí)行順序。

package main

import (
    "fmt"
    "sync"
)

func main() {
    ch := make(chan bool)
    go worker(ch)
    ch <- true  // 主協(xié)程通過發(fā)送信號控制worker的開始執(zhí)行
    fmt.Println("sent true to worker")
}

func worker(ch chan bool) {
    <-ch  // 等待主協(xié)程的信號
    fmt.Println("Received true, start working")
}

2. 使用 sync 包

sync 包提供了多種同步工具，如互斥鎖（Mutex）、讀寫鎖（RWMutex）、WaitGroup 等。

示例：使用 WaitGroup 等待多個(gè)協(xié)程完成
WaitGroup 是用來等待一組協(xié)程執(zhí)行完成。每個(gè)協(xié)程在開始時(shí)調(diào)用 WaitGroup.Add(1)，完成時(shí)調(diào)用 WaitGroup.Done()，而 WaitGroup.Wait() 會阻塞直到所有協(xié)程報(bào)告完成。

package main

import (
    "fmt"
    "sync"
    "time"
)

func main() {
    var wg sync.WaitGroup
    wg.Add(2)
    
    go func() {
        defer wg.Done()
        time.Sleep(2 * time.Second)
        fmt.Println("Task 1 completed")
    }()
    
    go func() {
        defer wg.Done()
        fmt.Println("Task 2 completed")
    }()
    
    wg.Wait()
    fmt.Println("All tasks completed")
}

3. 使用互斥鎖（Mutex）

互斥鎖用于保護(hù)共享資源，在多個(gè)協(xié)程需要訪問同一資源時(shí)，使用 Mutex 來保證同一時(shí)刻只有一個(gè)協(xié)程可以訪問該資源。

package main

import (
    "fmt"
    "sync"
)

var (
    counter int
    lock    sync.Mutex
)

func main() {
    var wg sync.WaitGroup
    wg.Add(2)

    go updateCounter(&wg)
    go updateCounter(&wg)

    wg.Wait()
    fmt.Printf("Final Counter: %d\n", counter)
}

func updateCounter(wg *sync.WaitGroup) {
    defer wg.Done()
    lock.Lock()
    defer lock.Unlock()
    for i := 0; i < 1000; i++ {
        counter++
    }
}