在Go語言中,為了保證并發編程的一致性,我們可以采用以下方法:
sync
包提供了Mutex
結構體,可以用來實現互斥鎖。使用Lock()
方法加鎖,Unlock()
方法解鎖。package main
import (
"fmt"
"sync"
)
var counter int
var lock sync.Mutex
func increment() {
lock.Lock()
counter++
lock.Unlock()
}
func main() {
wg := sync.WaitGroup{}
for i := 0; i < 100; i++ {
wg.Add(1)
go func() {
defer wg.Done()
increment()
}()
}
wg.Wait()
fmt.Println("Counter:", counter)
}
sync
包提供了RWMutex
結構體,可以用來實現讀寫鎖。使用RLock()
方法加讀鎖,RUnlock()
方法解讀鎖,Lock()
方法加寫鎖,Unlock()
方法解寫鎖。package main
import (
"fmt"
"sync"
)
var data map[string]string
var rwLock sync.RWMutex
func readData(key string) {
rwLock.RLock()
value := data[key]
rwLock.RUnlock()
fmt.Println("Read:", value)
}
func writeData(key, value string) {
rwLock.Lock()
data[key] = value
rwLock.Unlock()
}
func main() {
data = make(map[string]string)
wg := sync.WaitGroup{}
for i := 0; i < 10; i++ {
wg.Add(1)
go func(i int) {
defer wg.Done()
key := fmt.Sprintf("key%d", i)
value := fmt.Sprintf("value%d", i)
writeData(key, value)
}(i)
}
for i := 0; i < 100; i++ {
wg.Add(1)
go func(i int) {
defer wg.Done()
key := fmt.Sprintf("key%d", i%10)
readData(key)
}(i)
}
wg.Wait()
}
sync/atomic
包提供了一組原子操作函數,如AddInt32()
、CompareAndSwapInt32()
等。package main
import (
"fmt"
"sync"
"sync/atomic"
)
var counter int32
var wg sync.WaitGroup
func increment() {
atomic.AddInt32(&counter, 1)
}
func main() {
for i := 0; i < 100; i++ {
wg.Add(1)
go func() {
defer wg.Done()
increment()
}()
}
wg.Wait()
fmt.Println("Counter:", counter)
}
package main
import (
"fmt"
"sync"
)
func worker(wg *sync.WaitGroup, ch chan int) {
defer wg.Done()
for i := range ch {
fmt.Println("Received:", i)
}
}
func main() {
ch := make(chan int, 100)
wg := sync.WaitGroup{}
for i := 0; i < 10; i++ {
wg.Add(1)
go worker(&wg, ch)
}
for i := 0; i < 100; i++ {
ch <- i
}
close(ch)
wg.Wait()
}
這些方法可以單獨使用,也可以組合使用,以滿足不同的并發編程需求。在實際編程中,需要根據具體場景選擇合適的同步策略。