From 11d671c637fea63b74d9082a907b0a97f424e6e0 Mon Sep 17 00:00:00 2001
From: bergquist <carl.bergquist@gmail.com>
Date: Sat, 23 Feb 2019 18:28:33 +0100
Subject: [PATCH] add support for memcached

---
 pkg/infra/distcache/distcache.go         |  8 +--
 pkg/infra/distcache/distcache_test.go    |  2 +-
 pkg/infra/distcache/memcached_storage.go | 62 ++++++++++++++++++++++++
 3 files changed, 67 insertions(+), 5 deletions(-)
 create mode 100644 pkg/infra/distcache/memcached_storage.go

diff --git a/pkg/infra/distcache/distcache.go b/pkg/infra/distcache/distcache.go
index a60b3d309c2..8a6f7daf90c 100644
--- a/pkg/infra/distcache/distcache.go
+++ b/pkg/infra/distcache/distcache.go
@@ -39,12 +39,12 @@ func createClient(opts CacheOpts, sqlstore *sqlstore.SqlStore) cacheStorage {
 	}
 
 	if opts.name == "memcache" {
-		return nil
+		return newMemcacheStorage("localhost:9090")
 	}
 
-	if opts.name == "memory" {
-		return nil
-	}
+	// if opts.name == "memory" {
+	// 	return nil
+	// }
 
 	return newDatabaseCache(sqlstore) //&databaseCache{SQLStore: sqlstore}
 }
diff --git a/pkg/infra/distcache/distcache_test.go b/pkg/infra/distcache/distcache_test.go
index dd35744506c..a04b5d0228f 100644
--- a/pkg/infra/distcache/distcache_test.go
+++ b/pkg/infra/distcache/distcache_test.go
@@ -27,7 +27,7 @@ func createTestClient(t *testing.T, name string) cacheStorage {
 }
 
 func TestAllCacheClients(t *testing.T) {
-	clients := []string{"database", "redis"} // add redis, memcache, memory
+	clients := []string{"database", "redis", "memcached"} // add redis, memcache, memory
 
 	for _, v := range clients {
 		client := createTestClient(t, v)
diff --git a/pkg/infra/distcache/memcached_storage.go b/pkg/infra/distcache/memcached_storage.go
new file mode 100644
index 00000000000..44fbbcc33c6
--- /dev/null
+++ b/pkg/infra/distcache/memcached_storage.go
@@ -0,0 +1,62 @@
+package distcache
+
+import (
+	"time"
+
+	"github.com/bradfitz/gomemcache/memcache"
+)
+
+type memcacheStorage struct {
+	c *memcache.Client
+}
+
+func newMemcacheStorage(connStr string) *memcacheStorage {
+	return &memcacheStorage{
+		c: memcache.New(connStr),
+	}
+}
+
+func NewItem(sid string, data []byte, expire int32) *memcache.Item {
+	return &memcache.Item{
+		Key:        sid,
+		Value:      data,
+		Expiration: expire,
+	}
+}
+
+// Set sets value to given key in the cache.
+func (s *memcacheStorage) Put(key string, val interface{}, expires time.Duration) error {
+	item := &Item{Val: val}
+
+	bytes, err := EncodeGob(item)
+	if err != nil {
+		return err
+	}
+
+	memcacheItem := NewItem(key, bytes, int32(expires))
+
+	s.c.Add(memcacheItem)
+	return nil
+}
+
+// Get gets value by given key in the cache.
+func (s *memcacheStorage) Get(key string) (interface{}, error) {
+	i, err := s.c.Get(key)
+	if err != nil {
+		return nil, err
+	}
+
+	item := &Item{}
+
+	err = DecodeGob(i.Value, item)
+	if err != nil {
+		return nil, err
+	}
+
+	return item.Val, nil
+}
+
+// Delete delete a key from the cache
+func (s *memcacheStorage) Delete(key string) error {
+	return s.c.Delete(key)
+}