feat(utils): 新增泛型SyncMap实现,支持线程安全的键值存储和操作
This commit is contained in:
64
common/utils/syncmap.go
Normal file
64
common/utils/syncmap.go
Normal file
@@ -0,0 +1,64 @@
|
|||||||
|
package utils
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"sync"
|
||||||
|
)
|
||||||
|
|
||||||
|
// 定义一个泛型的 SyncMap,支持任意类型的键和值
|
||||||
|
type SyncMap[K comparable, V any] struct {
|
||||||
|
m sync.Map
|
||||||
|
}
|
||||||
|
|
||||||
|
// 设置键值对
|
||||||
|
func (sm *SyncMap[K, V]) Store(key K, value V) {
|
||||||
|
sm.m.Store(key, value)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取键对应的值
|
||||||
|
func (sm *SyncMap[K, V]) Load(key K) (V, bool) {
|
||||||
|
val, ok := sm.m.Load(key)
|
||||||
|
if ok {
|
||||||
|
return val.(V), true
|
||||||
|
}
|
||||||
|
var zeroValue V
|
||||||
|
return zeroValue, false
|
||||||
|
}
|
||||||
|
|
||||||
|
// 删除键值对
|
||||||
|
func (sm *SyncMap[K, V]) Delete(key K) {
|
||||||
|
sm.m.Delete(key)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 遍历所有键值对
|
||||||
|
func (sm *SyncMap[K, V]) Range(f func(key K, value V) bool) {
|
||||||
|
sm.m.Range(func(key, value any) bool {
|
||||||
|
return f(key.(K), value.(V))
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
// 使用 SyncMap 存储字符串 -> int 类型的键值对
|
||||||
|
sm1 := &SyncMap[string, int]{}
|
||||||
|
sm1.Store("apple", 5)
|
||||||
|
sm1.Store("banana", 10)
|
||||||
|
|
||||||
|
// 加载并打印值
|
||||||
|
if value, ok := sm1.Load("apple"); ok {
|
||||||
|
fmt.Println("apple:", value)
|
||||||
|
}
|
||||||
|
if value, ok := sm1.Load("banana"); ok {
|
||||||
|
fmt.Println("banana:", value)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 使用 SyncMap 存储 int -> string 类型的键值对
|
||||||
|
sm2 := &SyncMap[int, string]{}
|
||||||
|
sm2.Store(1, "one")
|
||||||
|
sm2.Store(2, "two")
|
||||||
|
|
||||||
|
// 遍历所有键值对
|
||||||
|
sm2.Range(func(key int, value string) bool {
|
||||||
|
fmt.Printf("%d: %s\n", key, value)
|
||||||
|
return true
|
||||||
|
})
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user