git.iko.soy/crush
on 0633988 help sign in
crush/internal/csync/versionedmap.go
51 lines 1.2 kB history raw
1package csync
3import (
4 "iter"
5 "sync/atomic"
6)
8// NewVersionedMap creates a new versioned, thread-safe map.
9func NewVersionedMap[K comparable, V any]() *VersionedMap[K, V] {
10 return &VersionedMap[K, V]{
11 m: NewMap[K, V](),
12 }
15// VersionedMap is a thread-safe map that keeps track of its version.
16type VersionedMap[K comparable, V any] struct {
17 m *Map[K, V]
18 v atomic.Uint64
21// Get gets the value for the specified key from the map.
22func (m *VersionedMap[K, V]) Get(key K) (V, bool) {
23 return m.m.Get(key)
26// Set sets the value for the specified key in the map and increments the version.
27func (m *VersionedMap[K, V]) Set(key K, value V) {
28 m.m.Set(key, value)
29 m.v.Add(1)
32// Del deletes the specified key from the map and increments the version.
33func (m *VersionedMap[K, V]) Del(key K) {
34 m.m.Del(key)
35 m.v.Add(1)
38// Seq2 returns an iter.Seq2 that yields key-value pairs from the map.
39func (m *VersionedMap[K, V]) Seq2() iter.Seq2[K, V] {
40 return m.m.Seq2()
43// Len returns the number of items in the map.
44func (m *VersionedMap[K, V]) Len() int {
45 return m.m.Len()
48// Version returns the current version of the map.
49func (m *VersionedMap[K, V]) Version() uint64 {
50 return m.v.Load()