go?sync包中的互斥鎖Mutex和等待組WaitGroup使用詳解

更新時間：2023年08月02日 11:55:57 作者：移動安全星球

這篇文章主要為大家介紹了go?sync包中的互斥鎖Mutex和等待組WaitGroup使用詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪

Go 語言的sync包

Go 語言的 sync 包提供了一些基本的同步原語，如互斥鎖（Mutex）和等待組（WaitGroup）。這些原語可以幫助你在多個 Go Routine 之間同步狀態(tài)。

Mutex

互斥鎖是一種同步工具，用于保證多個 Go Routine 在訪問共享資源時的互斥性。

創(chuàng)建 Mutex：

var m sync.Mutex

使用 Mutex：

m.Lock()
// 臨界區(qū)，只有獲得鎖的 Go Routine 才能訪問
m.Unlock()

例如，假設(shè)我們有一個計數(shù)器## sync 包的使用

Go 語言的 sync 包提供了一些基本的同步原語，如互斥鎖（Mutex）和等待組（WaitGroup）。這些原語可以幫助你在多個 Go Routine 之間同步狀態(tài)。

Mutex

互斥鎖是一種同步工具，用于保證多個 Go Routine 在訪問共享資源時的互斥性。

創(chuàng)建 Mutex：

var m sync.Mutex

使用 Mutex：

m.Lock()
// 臨界區(qū)，只有獲得鎖的 Go Routine 才能訪問
m.Unlock()

例如，假設(shè)我們有一個計數(shù)器，我們希望它能在多個 Go Routine 之間安全地使用：

package main
import (
    "fmt"
    "sync"
    "time"
)
type SafeCounter struct {
    v   map[string]int
    mux sync.Mutex
}
func (c *SafeCounter) Inc(key string) {
    c.mux.Lock()
    c.v[key]++
    c.mux.Unlock()
}
func (c *SafeCounter) Value(key string) int {
    c.mux.Lock()
    defer c.mux.Unlock()
    return c.v[key]
}
func main() {
    c := SafeCounter{v: make(map[string]int)}
    for i := 0; i < 1000; i++ {
        go c.Inc("somekey")
    }
    time.Sleep(time.Second)
    fmt.Println(c.Value("somekey"))
}

在這個例子中，SafeCounter 的每個方法在操作 v 之前都會鎖定 Mutex 來確保安全訪問。

WaitGroup

WaitGroup 是用于等待一組 Go Routines 完成的結(jié)構(gòu)。

使用 WaitGroup：

var wg sync.WaitGroup

每次一個 Go Routine 啟動時，我們調(diào)用 wg.Add(1)。每次一個 Go Routine 完成時，我們調(diào)用 wg.Done()。我們可以使用 wg.Wait() 來阻塞，直到所有的 Go Routines 完成。

例如，假設(shè)我們啟動了多個 Go Routine 來完成一些工作，我們希望等待它們?nèi)客瓿桑?/p>

package main
import (
    "fmt"
    "sync"
    "time"
)
func worker(id int, wg *sync.WaitGroup) {
    defer wg.Done()
    fmt.Printf("Worker %d starting\n", id)
    time.Sleep(time.Second)
    fmt.Printf("Worker %d done\n", id)
}
func main() {
    var wg sync.WaitGroup
    for i := 1; i <= 5; i++ {
        wg.Add(1)
        go worker(i, &wg)
    }
    wg.Wait()
}

在這個例子中，我們使用 WaitGroup 來等待 5 個 worker Go Routines 完成。