Merge pull request #93 from pczajkowski/master

Added Clear and TestClear. As per #90.

Author: orcaman

concurrent_map.go

@@ -186,6 +186,13 @@ func (m ConcurrentMap) IterBuffered() <-chan Tuple {
 	return ch
 }
 
+// Clear removes all items from map.
+func (m ConcurrentMap) Clear() {
+	for item := range m.IterBuffered() {
+		m.Remove(item.Key)
+	}
+}
+
 // Returns a array of channels that contains elements in each shard,
 // which likely takes a snapshot of `m`.
 // It returns once the size of each buffered channel is determined,

concurrent_map_test.go

@@ -326,6 +326,21 @@ func TestBufferedIterator(t *testing.T) {
 	}
 }
 
+func TestClear(t *testing.T) {
+	m := New()
+
+	// Insert 100 elements.
+	for i := 0; i < 100; i++ {
+		m.Set(strconv.Itoa(i), Animal{strconv.Itoa(i)})
+	}
+
+	m.Clear()
+
+	if m.Count() != 0 {
+		t.Error("We should have 0 elements.")
+	}
+}
+
 func TestIterCb(t *testing.T) {
 	m := New()