more fuse

1 files changed, 32 insertions, 0 deletions

diff --git a/pkg/util/generic.go b/pkg/util/generic.go
index dbe077f..6882724 100644
--- a/pkg/util/generic.go
+++ b/pkg/util/generic.go
@@ -2,6 +2,7 @@ package util
 
 import (
 	"sort"
+	"sync"
 
 	"golang.org/x/exp/constraints"
 )
@@ -88,3 +89,34 @@ func CmpUint[T constraints.Unsigned](a, b T) int {
 		return 1
 	}
 }
+
+type SyncMap[K comparable, V any] struct {
+	inner sync.Map
+}
+
+func (m *SyncMap[K, V]) Delete(key K) { m.inner.Delete(key) }
+func (m *SyncMap[K, V]) Load(key K) (value V, ok bool) {
+	_value, ok := m.inner.Load(key)
+	if ok {
+		value = _value.(V)
+	}
+	return value, ok
+}
+func (m *SyncMap[K, V]) LoadAndDelete(key K) (value V, loaded bool) {
+	_value, ok := m.inner.LoadAndDelete(key)
+	if ok {
+		value = _value.(V)
+	}
+	return value, ok
+}
+func (m *SyncMap[K, V]) LoadOrStore(key K, value V) (actual V, loaded bool) {
+	_actual, loaded := m.inner.LoadOrStore(key, value)
+	actual = _actual.(V)
+	return actual, loaded
+}
+func (m *SyncMap[K, V]) Range(f func(key K, value V) bool) {
+	m.inner.Range(func(key, value any) bool {
+		return f(key.(K), value.(V))
+	})
+}
+func (m *SyncMap[K, V]) Store(key K, value V) { m.inner.Store(key, value) }