揭秘 Go 语言中 runtime.Map 与 sync.Map 的真面目

后端

2024-01-01 02:58:44

runtime.Map 和 sync.Map 都提供了类似的 API，但它们在底层实现和使用场景上却有着天壤之别。

runtime.Map：纯粹的速度机器

runtime.Map 的设计理念是极致的速度。它利用 Go 语言的底层机制，直接在底层内存上操作，省去了锁机制的开销。这意味着 runtime.Map 在高并发场景下拥有无与伦比的性能。

sync.Map：安全第一的守护者

与 runtime.Map 相比，sync.Map 更加注重数据一致性和安全性。它在底层使用了读写锁机制，保证并发读写时的正确性和一致性。sync.Map 虽然牺牲了一定的性能，但它在并发安全方面提供了坚实的保障。

如何选择合适的武器

选择 runtime.Map 还是 sync.Map 取决于你的场景需求：

高并发、低锁争用场景： runtime.Map 是不二之选，因为它能提供最快的读写性能。
并发读写、数据一致性至关重要场景： sync.Map 值得信赖，因为它能保证并发读写时的正确性和一致性。

使用建议

在使用 runtime.Map 时，需要注意以下几点：

由于缺乏锁机制，runtime.Map 无法保证并发安全。
runtime.Map 仅支持基本类型和指针，不支持复杂数据结构。

示例代码

以下示例展示了如何使用 runtime.Map 和 sync.Map：

// 使用 runtime.Map
package main

import (
	"runtime"
	"sync"
)

func main() {
	// 初始化一个 runtime.Map
	m := runtime.Map{}

	// 并发写入
	var wg sync.WaitGroup
	for i := 0; i < 1000; i++ {
		wg.Add(1)
		go func(i int) {
			defer wg.Done()
			m.Store(i, i*i)
		}(i)
	}
	wg.Wait()

	// 并发读取
	for i := 0; i < 1000; i++ {
		if v, ok := m.Load(i); ok {
			fmt.Println(v)
		}
	}
}

// 使用 sync.Map
package main

import (
	"sync"
)

func main() {
	// 初始化一个 sync.Map
	m := sync.Map{}

	// 并发写入
	var wg sync.WaitGroup
	for i := 0; i < 1000; i++ {
		wg.Add(1)
		go func(i int) {
			defer wg.Done()
			m.Store(i, i*i)
		}(i)
	}
	wg.Wait()

	// 并发读取
	for i := 0; i < 1000; i++ {
		if v, ok := m.Load(i); ok {
			fmt.Println(v)
		}
	}
}