summaryrefslogtreecommitdiff
path: root/map.go
diff options
context:
space:
mode:
Diffstat (limited to 'map.go')
-rw-r--r--map.go53
1 files changed, 53 insertions, 0 deletions
diff --git a/map.go b/map.go
new file mode 100644
index 0000000..5b53748
--- /dev/null
+++ b/map.go
@@ -0,0 +1,53 @@
+// Copyright (C) 2022-2023 Luke Shumaker <lukeshu@lukeshu.com>
+//
+// SPDX-License-Identifier: GPL-2.0-or-later
+
+package typedsync
+
+import (
+ "sync"
+)
+
+type Map[K comparable, V any] struct {
+ inner sync.Map
+}
+
+func (m *Map[K, V]) Delete(key K) {
+ m.inner.Delete(key)
+}
+
+func (m *Map[K, V]) Load(key K) (value V, ok bool) {
+ _value, ok := m.inner.Load(key)
+ if ok {
+ //nolint:forcetypeassert // Typed wrapper around untyped lib.
+ value = _value.(V)
+ }
+ return value, ok
+}
+
+func (m *Map[K, V]) LoadAndDelete(key K) (value V, loaded bool) {
+ _value, ok := m.inner.LoadAndDelete(key)
+ if ok {
+ //nolint:forcetypeassert // Typed wrapper around untyped lib.
+ value = _value.(V)
+ }
+ return value, ok
+}
+
+func (m *Map[K, V]) LoadOrStore(key K, value V) (actual V, loaded bool) {
+ _actual, loaded := m.inner.LoadOrStore(key, value)
+ //nolint:forcetypeassert // Typed wrapper around untyped lib.
+ actual = _actual.(V)
+ return actual, loaded
+}
+
+func (m *Map[K, V]) Range(f func(key K, value V) bool) {
+ m.inner.Range(func(key, value any) bool {
+ //nolint:forcetypeassert // Typed wrapper around untyped lib.
+ return f(key.(K), value.(V))
+ })
+}
+
+func (m *Map[K, V]) Store(key K, value V) {
+ m.inner.Store(key, value)
+}