Fix removal with copy-on-write

forbearing · Apr 19, 2022 · f091512 · f091512
1 parent 2d74338
commit f091512
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/hashmap/map.go b/hashmap/map.go
@@ -146,6 +146,13 @@ func (m *Map[K, V]) Remove(key K) {
 		return
 	}
 
+	if m.readonly {
+		entries := make([]entry[K, V], len(m.entries), cap(m.entries))
+		copy(entries, m.entries)
+		m.entries = entries
+		m.readonly = false
+	}
+
 	m.remove(idx)
 
 	idx = (idx + 1) & (m.capacity - 1)