diff options
author | Luke Shumaker <lukeshu@lukeshu.com> | 2023-01-09 14:23:23 -0700 |
---|---|---|
committer | Luke Shumaker <lukeshu@lukeshu.com> | 2023-02-12 16:16:53 -0700 |
commit | d91f8ce17a6fc165fafd9dc921911233a69c34d2 (patch) | |
tree | 808a86b8f2309fe344b4cb0af62101b8a3b8b59a /lib/containers/maputil.go | |
parent | 1d7f5446dc37687f078269af3c63af7d7ebbfab4 (diff) |
tree-wide: Migrate to the new ARCache
Diffstat (limited to 'lib/containers/maputil.go')
-rw-r--r-- | lib/containers/maputil.go | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/lib/containers/maputil.go b/lib/containers/maputil.go new file mode 100644 index 0000000..4d49d2a --- /dev/null +++ b/lib/containers/maputil.go @@ -0,0 +1,32 @@ +// Copyright (C) 2023 Luke Shumaker <lukeshu@lukeshu.com> +// +// SPDX-License-Identifier: GPL-2.0-or-later + +package containers + +type Map[K comparable, V any] interface { + Store(K, V) + Load(K) (V, bool) + Has(K) bool + Delete(K) + Len() int +} + +type RangeMap[K comparable, V any] interface { + Map[K, V] + Range(func(K, V) bool) +} + +type SubrangeMap[K comparable, V any] interface { + RangeMap[K, V] + Subrange(rangeFn func(K, V) int, handleFn func(K, V) bool) +} + +func LoadOrElse[K comparable, V any](m Map[K, V], k K, vFn func(K) V) V { + if v, ok := m.Load(k); ok { + return v + } + v := vFn(k) + m.Store(k, v) + return v +} |