Adds ItemCount()

Signed-off-by: Elena Kolevska <[email protected]>

concurrency/mutexmap.go

@@ -24,6 +24,7 @@ type MutexMap[T comparable] interface {
 	RUnlock(key T)
 	Delete(key T)
 	Clear()
+	ItemCount() int
 }
 
 type mutexMap[T comparable] struct {
@@ -98,3 +99,9 @@ func (a *mutexMap[T]) Clear() {
 	a.items = make(map[T]*sync.RWMutex)
 	a.lock.Unlock()
 }
+
+func (a *mutexMap[T]) ItemCount() int {
+	a.lock.Lock()
+	defer a.lock.Unlock()
+	return len(a.items)
+}

concurrency/mutexmap_test.go

@@ -93,11 +93,14 @@ func TestNewMutexMap_Add_Delete(t *testing.T) {
 		require.False(t, ok)
 	})
 
-	t.Run("Clear all mutexes", func(t *testing.T) {
+	t.Run("Clear all mutexes, and check item count", func(t *testing.T) {
 		mm.Lock("key1")
 		mm.Unlock("key1")
 		mm.Lock("key2")
 		mm.Unlock("key2")
+
+		require.Equal(t, 2, mm.ItemCount())
+
 		mm.Clear()
 		require.Empty(t, mm.items)
 	})