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,我们可以确保所有的读和写操作都完成后,再继续执行后续代码。