Golang

关注公众号 jb51net

关闭
首页 > 脚本专栏 > Golang > Go线程安全Map

Go语言如何实现线程安全的Map

作者:熬了夜的程序员

Go语言内置的map虽然高效,但并不是线程安全的,若在多线程环境中直接操作map,可能会引发并发写入的错误,下面我们就来看看如何实现线程安全的Map吧

在并发编程中,数据共享和访问是一个重要的主题。Go语言内置的map虽然高效,但并不是线程安全的。若在多线程环境中直接操作map,可能会引发并发写入的错误(fatal error: concurrent map writes)。因此,在需要并发访问map时,必须采取措施确保线程安全。

本文将介绍如何使用Go语言的泛型和sync.RWMutex实现一个线程安全的Map,同时支持常见的操作,例如增删改查、遍历和转化为普通的Map。

1. 为什么需要线程安全的Map

Go语言内置的map在多线程环境中并不安全。例如,以下代码可能引发崩溃:

package main

import (
	"fmt"
	"sync"
)

func main() {
	m := make(map[int]int)
	var wg sync.WaitGroup

	for i := 0; i < 10; i++ {
		wg.Add(1)
		go func(i int) {
			defer wg.Done()
			m[i] = i
		}(i)
	}

	wg.Wait()
	fmt.Println(m)
}

运行上述代码可能会报错:fatal error: concurrent map writes。这是因为map的写操作没有加锁,在多线程中引发了竞态条件。

2. 如何实现线程安全的Map

Go标准库提供了sync.Map,它是线程安全的。但它的API相对简单,缺乏泛型支持且性能在某些场景下并不理想。因此,我们可以基于sync.RWMutex和泛型封装一个自定义的线程安全Map。

2.1 基本实现

以下是线程安全SyncMap的完整实现:

package syncmap

import (
	"sync"
)

// SyncMap 定义了一个线程安全的泛型Map
type SyncMap[K comparable, V any] struct {
	mu sync.RWMutex
	m  map[K]V
}

// NewSyncMap 创建一个新的线程安全的SyncMap
func NewSyncMap[K comparable, V any]() *SyncMap[K, V] {
	return &SyncMap[K, V]{
		m: make(map[K]V),
	}
}

// Load 获取指定key的值,如果存在返回值和true,否则返回零值和false
func (s *SyncMap[K, V]) Load(key K) (V, bool) {
	s.mu.RLock()
	defer s.mu.RUnlock()
	val, ok := s.m[key]
	return val, ok
}

// Store 设置指定key的值,如果key已存在会覆盖旧值
func (s *SyncMap[K, V]) Store(key K, value V) {
	s.mu.Lock()
	defer s.mu.Unlock()
	s.m[key] = value
}

// Has returns true if the key exists in the map.
func (s *SyncMap[K, V]) Has(key K) bool {
	s.mu.RLock()
	defer s.mu.RUnlock()

	_, ok := s.m[key]
	return ok
}


// Delete 删除指定key的值
func (s *SyncMap[K, V]) Delete(key K) {
	s.mu.Lock()
	defer s.mu.Unlock()
	delete(s.m, key)
}

// Range 遍历所有的键值对,callback函数返回false时停止遍历
func (s *SyncMap[K, V]) Range(callback func(key K, value V) bool) {
	s.mu.RLock()
	defer s.mu.RUnlock()
	for k, v := range s.m {
		if !callback(k, v) {
			break
		}
	}
}

// Len returns the length of the map.
func (s *SyncMap[K, V]) Len() int {
	s.mu.RLock()
	defer s.mu.RUnlock()
	return len(s.m)
}

// ToMap 转化为普通的map,返回一个线程安全的副本
func (s *SyncMap[K, V]) ToMap() map[K]V {
	s.mu.RLock()
	defer s.mu.RUnlock()

	copyMap := make(map[K]V, len(s.m))
	for k, v := range s.m {
		copyMap[k] = v
	}
	return copyMap
}

2.2 关键功能说明

线程安全:

支持泛型:

通过K和V泛型参数支持任意键值类型,其中K必须是可比较的。

基本操作:

3. 使用示例

以下代码演示了SyncMap的基本用法:

package main

import (
	"fmt"
	"syncmap"
)

func main() {
	// 创建一个线程安全的Map
	m := syncmap.NewSyncMap[string, int]()

	// 添加值
	m.Store("one", 1)
	m.Store("two", 2)

	// 获取值
	if val, ok := m.Load("one"); ok {
		fmt.Println("Key 'one':", val)
	} else {
		fmt.Println("Key 'one' not found")
	}

	// 删除值
	m.Delete("one")

	// 遍历所有键值对
	m.Range(func(key string, value int) bool {
		fmt.Printf("Key: %s, Value: %d
", key, value)
		return true
	})

	// 转化为普通map
	ordinaryMap := m.ToMap()
	fmt.Println("Ordinary map:", ordinaryMap)
}

运行结果:

Key 'one': 1
Key: two, Value: 2
Ordinary map: map[two:2]

4. 总结

自定义线程安全的SyncMap具备以下优点:

通过本文的实现与示例,希望您能更好地理解和应用线程安全Map,构建健壮的并发应用。

到此这篇关于Go语言如何实现线程安全的Map的文章就介绍到这了,更多相关Go线程安全Map内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文