channel 在 goroutine 中互相通信是非常合適且方便的送膳。如果,我們不需要互相通信,只需要保證同一時刻只能有一個 goroutine 訪問共享變量惰瓜,以免沖突。我們該怎么做呢汉矿?
這就需要用到 互斥和互斥鎖崎坊。
go 語言標(biāo)準(zhǔn)庫中提供了 sync.Mutex 類型及兩個方法: Lock 和 Unlock
我們可以通過在代碼前加 Lock,在代碼后加 Unlock 的方法洲拇,保證代碼執(zhí)行時的互斥性奈揍。
c.mux.Lock()
c.v[key]++ //Lock之后曲尸,同一時刻只有一個 goroutine 能訪問 c.v
c.mux.Unlock()
也可以用 defer 語句來保證互斥鎖一定會被解鎖。
c.mux.Lock()
defer c.mux.Unlock()
創(chuàng)建一個代碼示例男翰,先建立一個計數(shù)器結(jié)構(gòu)體
//安全計數(shù)器另患,使用 sync.Mutex 保證安全
type SafeCounter struct {
v map[string]int
mux sync.Mutex
}
v 是計數(shù)器map,再加上一個 sync.Mutex 類型蛾绎。
給 SafeCounter 增加兩個方法昆箕,一個 inc 來實現(xiàn)計數(shù)器值累加,一個 Value 來實現(xiàn)取得當(dāng)前計數(shù)器值租冠。
//增加計數(shù)器 key 的計數(shù)值
func (c *SafeCounter) Inc(key string) {
c.mux.Lock()
//Lock之后鹏倘,同一時刻只有一個 goroutine 能訪問 c.v
c.v[key]++
c.mux.Unlock()
}
//取得計數(shù)器當(dāng)前值
func (c *SafeCounter) Value(key string) int {
c.mux.Lock()
defer c.mux.Unlock()
return c.v[key]
}
主函數(shù)循環(huán) 100 次,使用 goroutine 讓一個計數(shù)器去增加值顽爹。然后取得當(dāng)前計數(shù)器的值纤泵。
為了避免運行太快,看不到效果话原,每次循環(huán)要間隔 100 毫秒夕吻。
time.Sleep(100*time.Millisecond)
完整代碼示例
package main
import (
"sync"
"fmt"
"time"
)
//安全計數(shù)器,使用 sync.Mutex 保證安全
type SafeCounter struct {
v map[string]int
mux sync.Mutex
}
//增加計數(shù)器 key 的計數(shù)值
func (c *SafeCounter) Inc(key string) {
c.mux.Lock()
//Lock之后繁仁,同一時刻只有一個 goroutine 能訪問 c.v
c.v[key]++
c.mux.Unlock()
}
//取得計數(shù)器當(dāng)前值
func (c *SafeCounter) Value(key string) int {
c.mux.Lock()
defer c.mux.Unlock()
return c.v[key]
}
func main() {
c := SafeCounter{v:make(map[string]int)}
for i := 0; i < 100; i++{
go c.Inc("somekey")
time.Sleep(100*time.Millisecond)
fmt.Println(c.Value("somekey"))
}
fmt.Println(c.Value("somekey"))
}
運行結(jié)果太長涉馅,不做展示。請自行實踐黄虱。