添加Keys 匹配key

Author: yumaojun03

cache/cache.go

@@ -24,6 +24,7 @@ func SetGlobal(c Cache) {
 
 // Cache provides the interface for cache implementations.
 type Cache interface {
+	Keys(pattern string) []string
 	// SetTTL set default ttl
 	SetDefaultTTL(ttl time.Duration)
 	// set cached value with key and expire time.

cache/memory/gocache.go

@@ -66,7 +66,6 @@ func (c *Cache) Get(key string, val interface{}) error {
 	}
 
 	return nil
-
 }
 
 // Delete 删除对象
@@ -104,6 +103,18 @@ func (c *Cache) Close() error {
 	return nil
 }
 
+// Keys todo
+func (c *Cache) Keys(pattern string) []string {
+	ks := []string{}
+	for _, k := range c.gc.Keys(true) {
+		if v, ok := k.(string); ok {
+			ks = append(ks, v)
+		}
+	}
+
+	return ks
+}
+
 // WithContext 携带上下文
 func (c *Cache) WithContext(ctx context.Context) cache.Cache {
 	clone := *c

cache/memory/gocache_test.go

@@ -34,6 +34,7 @@ func TestMemoryAdapterSuit(t *testing.T) {
 	t.Run("PutOK", testPutOK(suit))
 	t.Run("ExistOK", testExistOK(suit))
 	t.Run("ExistNotOK", testExistNotOK(suit))
+	t.Run("KeysOK", testKeysOK(suit))
 	t.Run("DelOK", testDelOK(suit))
 }
 
@@ -74,6 +75,15 @@ func testExistNotOK(a *adapterSuit) func(t *testing.T) {
 	}
 }
 
+func testKeysOK(a *adapterSuit) func(t *testing.T) {
+	return func(t *testing.T) {
+		should := require.New(t)
+
+		ks := a.adapter.Keys("*")
+		should.Equal([]string{a.testKey}, ks)
+	}
+}
+
 func testDelOK(a *adapterSuit) func(t *testing.T) {
 	return func(t *testing.T) {
 		should := require.New(t)

cache/redis/redis.go

@@ -101,6 +101,11 @@ func (c *Cache) ClearAll() error {
 	return c.client.FlushDB().Err()
 }
 
+// Keys todo
+func (c *Cache) Keys(pattern string) []string {
+	return c.client.Keys(pattern).Val()
+}
+
 // Decr todo
 func (c *Cache) Decr(key string) error {
 	return c.client.Decr(key).Err()

cache/redis/redis_test.go

@@ -36,6 +36,7 @@ func TestRedisAdapterSuit(t *testing.T) {
 	t.Run("ExistNotOK", testExistNotOK(suit))
 	t.Run("GetOK", testGetOK(suit))
 	t.Run("GetFailed", testGetFailed(suit))
+	t.Run("KeysOK", testKeysOK(suit))
 	t.Run("DelOK", testDelOK(suit))
 }
 
@@ -94,3 +95,12 @@ func testDelOK(a *adapterSuit) func(t *testing.T) {
 		should.NoError(err)
 	}
 }
+
+func testKeysOK(a *adapterSuit) func(t *testing.T) {
+	return func(t *testing.T) {
+		should := require.New(t)
+
+		ks := a.adapter.Keys("testkey*")
+		should.Equal([]string{a.testKey}, ks)
+	}
+}