在Go语言中,可以使用以下几种方法来解决并发资源竞争的问题:
sync.Mutex
类型来创建一个互斥锁对象,通过调用Lock()
和Unlock()
方法来保护临界区代码,确保同一时间只有一个goroutine可以访问共享资源。package main
import (
"fmt"
"sync"
)
var counter int
var mutex sync.Mutex
func main() {
var wg sync.WaitGroup
for i := 0; i < 1000; i++ {
wg.Add(1)
go increment(&wg)
}
wg.Wait()
fmt.Println(counter)
}
func increment(wg *sync.WaitGroup) {
mutex.Lock()
counter++
mutex.Unlock()
wg.Done()
}
sync.RWMutex
类型提供了更灵活的锁定机制,允许多个goroutine同时读取资源,但只允许一个goroutine写入资源。通过调用RLock()
和RUnlock()
方法来进行读取操作,调用Lock()
和Unlock()
方法来进行写入操作。package main
import (
"fmt"
"sync"
)
var counter int
var rwMutex sync.RWMutex
func main() {
var wg sync.WaitGroup
for i := 0; i < 1000; i++ {
wg.Add(1)
go increment(&wg)
}
wg.Wait()
fmt.Println(counter)
}
func increment(wg *sync.WaitGroup) {
rwMutex.Lock()
counter++
rwMutex.Unlock()
wg.Done()
}
sync/atomic
包提供的原子操作函数来执行对共享资源的操作,保证操作的原子性。常用的原子操作函数有AddInt32()
、AddInt64()
、AddUint32()
、AddUint64()
、CompareAndSwapInt32()
、CompareAndSwapInt64()
等。package main
import (
"fmt"
"sync"
"sync/atomic"
)
var counter int32
func main() {
var wg sync.WaitGroup
for i := 0; i < 1000; i++ {
wg.Add(1)
go increment(&wg)
}
wg.Wait()
fmt.Println(atomic.LoadInt32(&counter))
}
func increment(wg *sync.WaitGroup) {
atomic.AddInt32(&counter, 1)
wg.Done()
}
这些方法都可以有效地解决并发资源竞争的问题,具体选择使用哪种方法取决于实际需求和场景。