golang map 并发安全

admin

0
文章

0
评论

2024-10-09 10:52:53 编程来源：ZONE.CI 全球网 0 阅读模式

并发安全的Golang Map

在Go语言中，map是一种非常常用的数据结构，用于存储键值对。然而，当多个goroutine同时对map进行读写操作时，就会出现并发安全的问题。本文将介绍如何使用sync包中的锁机制来解决这个问题。

什么是并发安全？

并发安全是指多个并发执行的线程或进程对共享数据的访问不会产生冲突或破坏数据的完整性。在Golang中，当多个goroutine对同一个map进行读写操作时，会导致数据的不一致性，甚至可能引发panic错误。

使用Mutex实现并发安全的Map

sync包中提供了一种称为Mutex（互斥锁）的机制，用于保护对共享资源的访问。我们可以使用Mutex来保护map的读写操作，从而实现并发安全。

import (
	"sync"
)

type SafeMap struct {
	mu sync.Mutex
	data map[string]interface{}
}

func (sm *SafeMap) Get(key string) interface{} {
	sm.mu.Lock()
	defer sm.mu.Unlock()
	return sm.data[key]
}

func (sm *SafeMap) Set(key string, value interface{}) {
	sm.mu.Lock()
	defer sm.mu.Unlock()
	sm.data[key] = value
}

func (sm *SafeMap) Delete(key string) {
	sm.mu.Lock()
	defer sm.mu.Unlock()
	delete(sm.data, key)
}

在上述代码中，我们定义了一个SafeMap结构体，其中包含一个Mutex和一个map类型的data字段。Get、Set和Delete方法分别用于对map进行读取、写入和删除操作。在每个方法中，我们首先使用Lock方法获取锁，在操作完成后使用Unlock方法释放锁。

使用RWMutex实现并发安全的Map

除了互斥锁之外，sync包中还提供了一种称为RWMutex（读写锁）的机制，用于实现更高效的并发访问。RWMutex允许多个goroutine同时对共享资源进行读取操作，但只允许一个goroutine进行写入操作。

import (
	"sync"
)

type SafeMap struct {
	mu sync.RWMutex
	data map[string]interface{}
}

func (sm *SafeMap) Get(key string) interface{} {
	sm.mu.RLock()
	defer sm.mu.RUnlock()
	return sm.data[key]
}

func (sm *SafeMap) Set(key string, value interface{}) {
	sm.mu.Lock()
	defer sm.mu.Unlock()
	sm.data[key] = value
}

func (sm *SafeMap) Delete(key string) {
	sm.mu.Lock()
	defer sm.mu.Unlock()
	delete(sm.data, key)
}

上述代码中，我们将Mutex替换为RWMutex，并分别使用RLock和RUnlock方法进行读取操作；而对于写入和删除操作，则仍然使用Lock和Unlock方法来进行。