首頁 > 後端開發 > Golang > Go 語言中的並發安全如何實現?

Go 語言中的並發安全如何實現?

WBOY
發布: 2023-06-10 08:13:36
原創
938 人瀏覽過

隨著電腦科技的不斷發展,我們必須從單執行緒轉向多執行緒來針對程式的處理。相對於傳統並發處理模式,Go語言強大的並發處理機制吸引了許多開發者的注意。 Go語言提供了一種輕量級的實作機制,使得地道的並發程式碼可以更容易地編寫。

然而,不可避免的是,多執行緒環境會帶來很多競爭條件(Race Condition)。當多個執行緒嘗試同時讀取和寫入同一個共享資源時,由於執行的順序的不確定性,可能會出現意外的結果。 Race Condition 是開發者最害怕的潛在問題之一。

為了避免在同時處理中出現潛在的問題,Go語言提供了種類豐富的標準函式庫:sync。本文將介紹透過 sync 函式庫實現並發安全的機制。

Mutex 和 RWMutex

mutex 是最常用的機制。在任意時刻,只能有一個協程能夠獲得 mutex 對象,而其他協程需要等待前一個協程釋放鎖之後才能夠繼續執行。 Mutex 可以用於保護共享資源,程式碼安全且穩定地運作。

RWMutex 是另一種互斥鎖類型,相當於 mutex 在讀寫領域的擴充。 RWMutex 包含了兩個計數器:讀取計數器和寫入計數器。

  • 當讀協程還在進行讀取操作時,寫入操作會被鎖住,並等待讀取操作結束。
  • 當寫協程在呼叫鎖定操作時,所有正在進行的協程的讀取和寫入操作都會被鎖定。

這種機制保證同時有多個協程能夠進行讀取操作,並且只有單一協程能夠進行寫入操作。

var rwMutex sync.RWMutex
var count int
func read() {
    rwMutex.RLock()
    defer rwMutex.RUnlock()

    fmt.Println(count)
}

func write() {
    rwMutex.Lock()
    defer rwMutex.Unlock()

    count++
}
登入後複製

在上述範例程式碼中,我們使用了 RWMutex 類型的鎖定來保護 count 變數的讀寫操作。當一個執行緒呼叫 write() 函數時,將會鎖定寫入計數器,並阻止所有其它協程進行讀寫操作。而當一個執行緒呼叫 read() 函數時,將會鎖定讀取計數器,並允許其它協程進行讀取操作。

WaitGroup

WaitGroup 用來等待一組協程執行完成。假設我們有 n 個協程需要執行,那麼在主協程中,我們需要呼叫 waitGroup.Add(n)。在每個協程執行完成後呼叫 waitGroup.Done()。

func main() {
    var wg sync.WaitGroup

    for i := 0; i < 5; i++ {
        wg.Add(1)
        go func(n int) {
            fmt.Println("goroutine ", n)
            wg.Done()
        }(i)
    }
    wg.Wait()
}
登入後複製

在這個例子中,我們使用 WaitGroup 來等待每個 goroutine 的執行,最後等待所有 goroutine 完成之後,才結束 main 的執行過程。

Cond

當多個協程需要停止或執行某些特定操作時,我們可以使用 Cond。 Cond 與鎖和 WaitGroup 的結合使用是很常見的。它允許 goroutine 同時阻塞,直到一個條件變數發生變化。

var cond = sync.NewCond(&sync.RWMutex{})

func printOddNumbers() {
    for i := 0; i < 10; i++ {
        cond.L.Lock()
        if i%2 == 1 {
            fmt.Println(i)
            cond.Signal()
        } else {
            cond.Wait()
        }
        cond.L.Unlock()
    }
}

func printEvenNumbers() {
    for i := 0; i < 10; i++ {
        cond.L.Lock()
        if i%2 == 0 {
            fmt.Println(i)
            cond.Signal()
        } else {
            cond.Wait()
        }
        cond.L.Unlock()
    }
}
登入後複製

在上述程式碼範例中,我們使用了 Cond 來保證偶數和奇數的分別輸出。每個協程都使用 sync.Mutex 來鎖定 goroutine,並等待另一個協程首先存取共享變量,然後再監測獲得變量的值。

Once

在某些情況下,需要確保只執行一次某些操作,例如只讀取一次設定檔或只初始化一次全域狀態。 Go語言的 sync.Once 類型正是為此而生。當第一次呼叫函數時它會執行其內部的程式碼,之後呼叫都不會再執行。

var once sync.Once

func doSomething() {
    once.Do(func() {
        fmt.Println("Do something")
    })
}
登入後複製

在上述範例中,我們使用 sync.Once 安全地執行了 doSomething 函數。在第一次呼叫 doSomething 時,函數會使用 once.Do() 只執行一次。

Conclusion

在本篇文章中,我們介紹了 Go 語言中常用的鎖定和機制,以確保並發程式碼的安全。使用 sync 函式庫的 Mutex、RWMutex、WaitGroup、Cond 和 Once 類型都非常強大,可用於設計安全且高效的並發程式。隨著並發機制的不斷發展,了解並發程式設計的最新進展是保持開發技能競爭力的關鍵。

以上是Go 語言中的並發安全如何實現?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

相關標籤:
來源:php.cn
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
最新問題
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板