Golang 解决 Map 并发读写安全问题

最新推荐文章于 2024-01-28 10:30:00 发布

追风2019

最新推荐文章于 2024-01-28 10:30:00 发布

阅读量2.8k

点赞数

分类专栏： Go 文章标签： go map

本文链接：https://blog.csdn.net/createNo_1/article/details/106729583

版权

Go 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

一、错误案例

package main

import (
	"fmt"
	"time"
)

var TestMap map[string]string

func init() {
	TestMap = make(map[string]string, 1)
}

func main() {
	for i := 0; i < 1000; i++ {
		go Write("aaa")
		go Read("aaa")
		go Write("bbb")
		go Read("bbb")
	}
	time.Sleep(5 * time.Second)
}

func Read(key string) {
	fmt.Println(TestMap[key])
}

func Write(key string) {
	TestMap[key] = key
}

上面代码执行大概率出现报错：fatal error: concurrent map writes

二、问题分析

网上关于 golang 编程中 map 并发读写相关的资料很多，但总是都说成并发读写造成上面的错误，到底是并发读还是并发写造成的，这个很多资料都没有说明。

我们把上面的案例分别在循环中注释 Read 和 Write 函数的调用，分别测试并发读和并发写；

循环次数分别测试了 100、1 w、100 w 次，并发读操作绝对不会报上面的错，而并发写基本都会报错。

因此，这个错误主要原因是：map 并发写。

三、问题原因

为什么 map 并发写会导致这个错误？网络上的相关文章也大都有说明。

因为 map 变量为指针类型变量，并发写时，多个协程同时操作一个内存，类似于多线程操作同一个资源会发生竞争关系，共享资源会遭到破坏，因此golang 出于安全的考虑，抛出致命错误：fatal error: concurrent map writes。

四、解决方案

网上各路资料解决方案较多，主要思路是通过加锁保证每个协程同步操作内存。

github 上找到一个 concurrentMap 包，案例代码修改如下：

package main

import (
	"fmt"
	cmap "github.com/orcaman/concurrent-map"
	"time"
)

var TestMap cmap.ConcurrentMap

func init() {
	TestMap = cmap.New()
}

func main() {
	for i := 0; i < 100; i++ {
		go Write("aaa", "111")
		go Read("aaa")
		go Write("bbb", "222")
		go Read("bbb")
	}
	time.Sleep(5 * time.Second)
}

func Read(key string) {
	if v, ok := TestMap.Get(key); ok {
		fmt.Printf("键值为 %s 的值为：%s", key, v)
	} else {
		fmt.Printf("键值不存在")
	}
}

func Write(key string, value string) {
	TestMap.Set(key, value)
}