golang中使用sync.Map的方法
背景
go中map数据结构不是线程安全的,即多个goroutine同时操作一个map,则会报错,因此go1.9之后诞生了sync.Map
sync.Map思路来自java的ConcurrentHashMap
接口
sync.map就是1.9版本带的线程安全map,主要有如下几种方法:
Load(key interface{}) (value interface{}, ok bool) //通过提供一个键key,查找对应的值value,如果不存在,则返回nil。ok的结果表示是否在map中找到值 Store(key, value interface{}) //这个相当于是写map(更新或新增),第一个参数是key,第二个参数是value LoadOrStore(key, value interface{}) (actual interface{}, loaded bool) //通过提供一个键key,查找对应的值value,如果存在返回键的现有值,否则存储并返回给定的值,如果是读取则返回true,如果是存储返回false Delete(key interface{}) //通过提供一个键key,删除键对应的值 Range(f func(key, value interface{}) bool) //循环读取map中的值。 //因为for ... range map是内置的语言特性,所以没有办法使用for range遍历sync.Map, 但是可以使用它的Range方法,通过回调的方式遍
实践
package main import ( "fmt" "sync" ) var num = 0 var addTest *AddTest func init() { addTest = &AddTest{} } type AddTest struct { m sync.Mutex } func (at *AddTest) increment(wg *sync.WaitGroup) { //互斥锁 at.m.Lock() //当有线程进去进行加锁 num++ at.m.Unlock() //出来后解锁,其他线程才可以进去 wg.Done() } func (at *AddTest) decrement(wg *sync.WaitGroup) { //互斥锁 at.m.Lock() //当有线程进去进行加锁 num-- at.m.Unlock() //出来后解锁,其他线程才可以进去 wg.Done() } var w sync.WaitGroup var aa map[int]int func main() { var bb sync.Map var wg sync.WaitGroup //aa = make(map[int]int) wg.Add(2) go func() { //wg.Add(1) for i:=0 ;i <100; i++{ //aa[i] = i+1 //fmt.Println("a") bb.Store(i, i+1) } wg.Done() }() go func() { for i:=0 ;i <100; i++{ //aa[i] = i+1 //fmt.Println("a") bb.Store(i, i+1) } wg.Done() }() wg.Wait() bb.Range(func(k, v interface{}) bool { fmt.Println("iterate:", k, v) return true } }
总结
- 读写锁和互斥锁 读写锁: 可以获取多个读锁,只有读写冲突(加了读锁的时候,其它线程不能写) 互斥锁:跟读写操作无关,加了锁,锁内的资源就线程独享
- 个人感觉使用起来不太方便,不如根据实际场景自己互斥锁。比如map都是可读的,只有写的时候需要串行执行,则写操作封装互斥锁即可
- sync.Map因为内部的操作较多等原因,并不适合大量写的场景(适合大量读,少量写)。
- sync.Map的原理详见:https://www.jb51.net/article/188788.htm
参考