更新slicemap结构

pangdogs · Dec 3, 2024 · 3dd5440 · 3dd5440
1 parent 360296b
commit 3dd5440
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/utils/generic/slicemap.go b/utils/generic/slicemap.go
@@ -97,6 +97,13 @@ func (m *SliceMap[K, V]) Delete(k K) bool {
 	return ok
 }
 
+func (m SliceMap[K, V]) Index(k K) int {
+	idx, _ := slices.BinarySearchFunc(m, KV[K, V]{K: k}, func(a, b KV[K, V]) int {
+		return cmp.Compare(a.K, b.K)
+	})
+	return idx
+}
+
 func (m SliceMap[K, V]) Get(k K) (V, bool) {
 	idx, ok := slices.BinarySearchFunc(m, KV[K, V]{K: k}, func(a, b KV[K, V]) int {
 		return cmp.Compare(a.K, b.K)

diff --git a/utils/generic/slicemap_unordered.go b/utils/generic/slicemap_unordered.go
@@ -96,6 +96,12 @@ func (m *UnorderedSliceMap[K, V]) Delete(k K) bool {
 	return idx >= 0
 }
 
+func (m UnorderedSliceMap[K, V]) Index(k K) int {
+	return slices.IndexFunc(m, func(kv UnorderedKV[K, V]) bool {
+		return kv.K == k
+	})
+}
+
 func (m UnorderedSliceMap[K, V]) Get(k K) (V, bool) {
 	idx := slices.IndexFunc(m, func(kv UnorderedKV[K, V]) bool {
 		return kv.K == k