cache: make it concurrency safe

1 files changed, 43 insertions, 0 deletions

diff --git a/cache/cache_test.go b/cache/cache_test.go
index 4751260..c663abf 100644
--- a/cache/cache_test.go
+++ b/cache/cache_test.go
@@ -5,6 +5,7 @@ package cache
 
 import (
 	"bytes"
+	"fmt"
 	"testing"
 	"time"
 )
@@ -71,3 +72,45 @@ func TestCacheGet(t *testing.T) {
 		return
 	}
 }
+
+func TestConcurrentSets(t *testing.T) {
+	// Expiration time for all keys.
+	exp := time.Now().Add(time.Second * 120)
+
+	// Generate some keys.
+	keys := make([]string, 0)
+	maxKeys := 1000
+	for i := 0; i < maxKeys; i++ {
+		keys = append(keys, fmt.Sprintf("key-%d", i))
+	}
+
+	// Go routing for adding keys to cache.
+	addToCache := func(c *Cache, keys []string, donec chan int) {
+		for i := 0; i < len(keys); i++ {
+			c.Set(keys[i], []byte(fmt.Sprintf("val-%d", i)), exp)
+		}
+		donec <- 1
+	}
+
+	// Init. cache.
+	c := NewCache()
+	if c == nil {
+		t.Errorf("cache is nil")
+		return
+	}
+	donec := make(chan int)
+
+	// Add keys to cache concurrently.
+	go addToCache(c, keys, donec)
+	go addToCache(c, keys, donec)
+	go addToCache(c, keys, donec)
+	completed := 0
+	for completed < 3 {
+		<-donec
+		completed += 1
+	}
+
+	if len(c.store) != maxKeys {
+		t.Errorf("number of keys in store != %d: %v", maxKeys, c.store)
+	}
+}