在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()
}
這些方法都可以有效地解決并發資源競爭的問題,具體選擇使用哪種方法取決于實際需求和場景。