Go语言sync.Map

Go语言中的 map 在并发情况下,只读是线程安全的,同时读写是线程不安全的。
如果想实现并发线程安全有两种方法:

  • map加互斥锁或读写锁
  • 标准库sync.map(Go1.19+新特性)

sync.map源码

https://github.com/golang/go/...

sync.map 实现原理及优化

  • 利用map只读不用锁,通过冗余 read 和 dirty 两个字段将读写分离,读的数据存在只读字段 read 上,将最新写入的数据则存在 dirty 字段上,只在dirty读写上加锁,提高程序只读效率。
  • 读取时会先查询 read,不存在再查询 dirty,写入时则只写入 dirty
  • 读取 read 并不需要加锁,而读或写 dirty 都需要加锁
  • 另外有 misses 字段来统计 read 被穿透的次数(被穿透指需要读 dirty 的情况),超过一定次数则将 dirty 数据同步到 read 上
  • 对于删除数据则直接通过标记来延迟删除

具体数据结构可参考:
https://blog.csdn.net/u010853...
https://www.haohongfan.com/do...

sync.map 使用场景

map+Mutex: 通过Mutex互斥锁来实现多个goroutine对map的串行化访问,读写都需要通过Mutex加锁和释放锁,适用于读写比接近的场景
map+RWMutex:通过RWMutex来实现对map的读写进行读写锁分离加锁,从而实现读的并发性能提高,同Mutex相比适用于读多写少的场景
sync.Map:底层通分离读写map和原子指令来实现读的近似无锁,并通过延迟更新的方式来保证读的无锁化。读多修改少,元素增加删除频率不高的情况,在大多数情况下替代上述两种实现

sync.map 使用方法如下

package main

import (
    "fmt"
    "sync"
)

func main() {
    var m sync.Map
    //添加一个元素
    m.Store(1, "a")
    m.Store("a", 2)

    //读取一个元素
    fmt.Println(m.Load(1))   //a true
    fmt.Println(m.Load("a")) //2 true

    //读取不存在的元素
    fmt.Println(m.Load(2)) // false

    //存在就返回,不存在就插入
    fmt.Println(m.LoadOrStore("3", 33)) //33 false
    fmt.Println(m.Load("3"))            //33 true

    //如果存在的话,同时删除这个 key
    fmt.Println(m.LoadAndDelete("3")) // 33 true

    //删除某个元素
    m.Delete("3")
    // 遍历所有sync.Map中的键值对
    m.Range(func(k, v interface{}) bool {
        fmt.Println("iterate:", k, v)
        return true
    })
}

通过以上示例可以看到sync.map具有以下特性:

  • 可以存储不同的数据类型在一起,这有别于map只能存储声明好的数据类型,且相同的。
  • 无须初始化,直接声明即可。
  • sync.Map 不能使用map的方式进行取值和设置等操作,而是使用 sync.Map 的方法进行调用,Store 表示存储,Load 表示获取,Delete 表示删除
  • 增加了两个特殊方法LoadOrStore 存在就返回,不存在就插入,LoadAndDelete 如果存在的话,同时删除这个 key
  • 使用 Range 配合一个回调函数进行遍历操作,通过回调函数返回内部遍历出来的值,Range 参数中回调函数的返回值在需要继续迭代遍历时,返回 true,终止迭代遍历时,返回 false。

参考资料

https://juejin.cn/post/684490...
https://pkg.go.dev/sync#Map
https://blog.csdn.net/u010853...
https://medium.com/@deckarep/...
https://www.haohongfan.com/do...

你可能感兴趣的:(gomapsync)