43 lines
809 B
Go
43 lines
809 B
Go
package utils
|
||
|
||
import (
|
||
"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]) LoadOrStore(key K, value V) (V, bool) {
|
||
val, ok := sm.m.LoadOrStore(key, value)
|
||
return val.(V), ok
|
||
|
||
}
|
||
|
||
// 获取键对应的值
|
||
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))
|
||
})
|
||
}
|