Go語言的sync.RWMutex
是一個讀寫鎖,用于在多個goroutine之間同步對共享資源的訪問。它允許多個讀操作同時進行,但在寫操作進行時,只允許一個goroutine進行寫操作。這樣可以提高并發性能,因為讀操作通常比寫操作更頻繁。
下面是一個簡單的示例,展示了如何使用sync.RWMutex
來管理資源:
package main
import (
"fmt"
"sync"
"time"
)
type Data struct {
value int
mu sync.RWMutex
}
func (d *Data) Read() int {
d.mu.RLock() // 加讀鎖
defer d.mu.RUnlock() // 讀鎖結束時釋放讀鎖
return d.value
}
func (d *Data) Write(value int) {
d.mu.Lock() // 加寫鎖
defer d.mu.Unlock() // 寫鎖結束時釋放寫鎖
d.value = value
}
func main() {
data := Data{value: 0}
var wg sync.WaitGroup
// 啟動10個讀操作
for i := 0; i < 10; i++ {
wg.Add(1)
go func() {
defer wg.Done()
fmt.Println("Read:", data.Read())
}()
}
// 等待所有讀操作完成
wg.Wait()
// 啟動1個寫操作
wg.Add(1)
go func() {
defer wg.Done()
data.Write(42)
}()
// 等待寫操作完成
wg.Wait()
// 再次啟動10個讀操作
for i := 0; i < 10; i++ {
wg.Add(1)
go func() {
defer wg.Done()
fmt.Println("Read:", data.Read())
}()
}
// 等待所有讀操作完成
wg.Wait()
}
在這個示例中,我們定義了一個Data
結構體,它包含一個整數值和一個sync.RWMutex
類型的字段。我們還定義了Read
和Write
方法,分別用于讀取和寫入數據。在讀取數據時,我們使用RLock
方法加讀鎖,并在讀取完成后使用RUnlock
方法釋放讀鎖。在寫入數據時,我們使用Lock
方法加寫鎖,并在寫入完成后使用Unlock
方法釋放寫鎖。
在main
函數中,我們創建了一個Data
實例,并啟動了多個讀和寫操作。通過使用sync.WaitGroup
,我們可以確保所有的讀和寫操作都完成后,再繼續執行后續代碼。