map与并发
我们都知道golang最大的优势在于对并发场景的处理,那么map遇到并发场景会出现什么情况呢?我们先看一个例子
var wg sync.WaitGroup
// var lock sync.Mutex
func f1(k int, m map[int]int) {
// lock.Lock()
m[k] = k * k
fmt.Println(k, m[k])
// lock.Unlock()
wg.Done()
}
func main() {
square := make(map[int]int)
for i := 0; i < 20; i++ {
wg.Add(1)
go f1(i, square)
}
wg.Wait()
}
// output:
// E:\go\src\learngolang\syncMap>go run main.go
// fatal error: concurrent map writes6 36
// 4 16
// fatal error: concurrent map writes
上面的例子中,我们创建了20个goroutine去对一个map写入和读取,可以看到出现了 致命错误:concurrent map writes,这说明golang的基础类型map是不是并发安全的,那么我们想要实现并发map该怎么做呢?就如上面例子中注释掉的语句一样,我们可以在对map操作的时候给他加锁,这样就是实现了并发map。如果想要应对多读少写场景,可以选用我之前博客中介绍的
那么golang中还有没有更加方便的方法呢?当然是有的,在golang 1.9之后,golang中增加了sync.Map。
sync.Map 和 map 不同,不是以语言原生形态提供,而是在 sync 包下的特殊结构。
- 无须初始化,直接声明即可。
- sync.Map 不能使用 map 的方式进行取值和设置等操作,而是使用 sync.Map 的方法进行调用,Store 表示存储,Load 表示获取,Delete 表示删除。
- 使用 Range 配合一个回调函数进行遍历操作,通过回调函数返回内部遍历出来的值,Range 参数中回调函数的返回值在需要继续迭代遍历时,返回 true,终止迭代遍历时,返回 false。
基本api有这几个
- Store 写入
- Load 读取,返回值有两个,第一个是value,第二个是bool变量表示key是否存在
- Delete 删除
- LoadOrStore 存在就读,不存在就写
- Range 遍历,Range() 方法需要提供一个匿名函数,参数为 k、v,类型为 interface{},每次 Range() 在遍历一个元素时,都会调用这个匿名函数把结果返回。
sync map底层使用map[interface{}]* entry来做存储,所以无论key还是value都是支持多种数据类型。
下面我们用一个例子看一下它的使用方法
func main() {
var square sync.Map
for i := 0; i < 10; i++ {
square.Store(i, i*i)
}
square.Store("age", 99)
age, ok := square.Load("age")
if ok {
fmt.Printf("age: %d\n", age)
}
square.Delete("age")
f := func(k, v interface{}) bool {
fmt.Println(k, v)
return true
}
square.Range(f)
}
// age: 99
// 5 25
// 6 36
// 7 49
// 9 81
// 2 4
// 1 1
// 3 9
// 4 16
// 8 64
// 0 0
有疑问加站长微信联系(非本文作者)