发布时间:2024-12-22 23:07:15
import "sync"
接下来,我们定义一个互斥锁:var mutex sync.Mutex
在需要保护的临界区代码块前后,分别使用锁的Lock()和Unlock()方法来加锁和解锁:mutex.Lock()
// 这里是临界区代码块
mutex.Unlock()
下面是一个示例代码:package main
import (
"fmt"
"sync"
)
func main() {
var wg sync.WaitGroup
var mutex sync.Mutex
var counter int
for i := 0; i < 100; i++ {
wg.Add(1)
go func() {
mutex.Lock()
counter++
mutex.Unlock()
wg.Done()
}()
}
wg.Wait()
fmt.Println("Counter:", counter)
}
在上面的代码中,我们使用互斥锁来保护counter变量。每个goroutine在访问该变量之前都会先获取锁,然后进行+1操作后释放锁。这样可以确保counter变量的正确递增。import "sync"
然后,定义一个读写锁:var rwMutex sync.RWMutex
在需要保护的读操作代码块前后,分别使用锁的RLock()和RUnlock()方法来加锁和解锁:rwMutex.RLock()
// 这里是读操作代码块
rwMutex.RUnlock()
在需要保护的写操作代码块前后,分别使用锁的Lock()和Unlock()方法来加锁和解锁:rwMutex.Lock()
// 这里是写操作代码块
rwMutex.Unlock()
下面是一个示例代码:package main
import (
"fmt"
"sync"
)
func main() {
var wg sync.WaitGroup
var rwMutex sync.RWMutex
var counter int
for i := 0; i < 100; i++ {
wg.Add(1)
go func() {
rwMutex.Lock()
counter++
rwMutex.Unlock()
wg.Done()
}()
}
for i := 0; i < 10; i++ {
wg.Add(1)
go func() {
rwMutex.RLock()
fmt.Println("Counter:", counter)
rwMutex.RUnlock()
wg.Done()
}()
}
wg.Wait()
}
在上面的代码中,我们使用读写锁来保护counter变量。写操作需要先获取写锁,读操作需要先获取读锁。这样可以实现多个goroutine并发读取counter变量,而写操作会阻塞其他任何读写操作。参考资料:
https://golang.org/pkg/sync/
https://go101.org/article/basic-types-and-vars.html#basic-lock-types
https://go101.org/article/concurrent-atomic-object.html#Concurrent-References-Are-Always-Safe
https://go101.org/article/concurrent-control-primitives.html#sync.Mutex-And-Sync.RWMutex