golang中使用sync.Map

Z时代
2024-01-10
分类：综合

背景

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 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
	})
}