Store reported piped stats to redis

pkg/cache/cache.go

@@ -19,17 +19,20 @@ import (
 )
 
 var (
-	ErrNotFound = errors.New("not found")
+	ErrNotFound      = errors.New("not found")
+	ErrUnimplemented = errors.New("unimplemented")
 )
 
 // Getter wraps a method to read from cache.
 type Getter interface {
 	Get(key interface{}) (interface{}, error)
+	GetAll() ([]interface{}, error)
 }
 
 // Putter wraps a method to write to cache.
 type Putter interface {
 	Put(key interface{}, value interface{}) error
+	PutHash(key interface{}, value interface{}) error
 }
 
 // Deleter wraps a method to delete from cache.
@@ -82,3 +85,11 @@ func (mg *multiGetter) Get(key interface{}) (interface{}, error) {
 	}
 	return nil, firstErr
 }
+
+func (mg *multiGetter) GetAll() ([]interface{}, error) {
+	return nil, ErrUnimplemented
+}
+
+func (mg *multiGetter) PutHash(k interface{}, v interface{}) error {
+	return ErrUnimplemented
+}

pkg/cache/cache_test.go

@@ -27,6 +27,10 @@ func (f getterFunc) Get(key interface{}) (interface{}, error) {
 	return f(key)
 }
 
+func (f getterFunc) GetAll() ([]interface{}, error) {
+	return nil, ErrUnimplemented
+}
+
 func TestMultiGetter(t *testing.T) {
 	value := "ok"
 	err := errors.New("err")

pkg/cache/memorycache/cache.go

@@ -54,3 +54,11 @@ func (c *Cache) Delete(key interface{}) error {
 	c.values.Delete(key)
 	return nil
 }
+
+func (c *Cache) GetAll() ([]interface{}, error) {
+	return nil, cache.ErrUnimplemented
+}
+
+func (c *Cache) PutHash(k interface{}, v interface{}) error {
+	return cache.ErrUnimplemented
+}

pkg/cache/memorycache/lru_cache.go

@@ -60,3 +60,11 @@ func (c *LRUCache) Delete(key interface{}) error {
 	c.cache.Remove(key)
 	return nil
 }
+
+func (c *LRUCache) GetAll() ([]interface{}, error) {
+	return nil, cache.ErrUnimplemented
+}
+
+func (c *LRUCache) PutHash(k interface{}, v interface{}) error {
+	return cache.ErrUnimplemented
+}

pkg/cache/memorycache/ttl_cache.go

@@ -97,3 +97,11 @@ func (c *TTLCache) Delete(key interface{}) error {
 	c.entries.Delete(key)
 	return nil
 }
+
+func (c *TTLCache) GetAll() ([]interface{}, error) {
+	return nil, cache.ErrUnimplemented
+}
+
+func (c *TTLCache) PutHash(k interface{}, v interface{}) error {
+	return cache.ErrUnimplemented
+}

pkg/cache/rediscache/BUILD.bazel

@@ -2,7 +2,10 @@ load("@io_bazel_rules_go//go:def.bzl", "go_library")
 
 go_library(
     name = "go_default_library",
-    srcs = ["cache.go"],
+    srcs = [
+        "cache.go",
+        "hashcache.go",
+    ],
     importpath = "github.com/pipe-cd/pipe/pkg/cache/rediscache",
     visibility = ["//visibility:public"],
     deps = [

pkg/cache/rediscache/cache.go

@@ -95,3 +95,11 @@ func (c *RedisCache) Delete(k interface{}) error {
 	_, err := conn.Do("DEL", k)
 	return err
 }
+
+func (c *RedisCache) GetAll() ([]interface{}, error) {
+	return nil, cache.ErrUnimplemented
+}
+
+func (c *RedisCache) PutHash(k interface{}, v interface{}) error {
+	return cache.ErrUnimplemented
+}

pkg/cache/rediscache/hashcache.go

@@ -0,0 +1,79 @@
+// Copyright 2021 The PipeCD Authors.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package rediscache
+
+import (
+	"time"
+
+	redigo "github.com/gomodule/redigo/redis"
+
+	"github.com/pipe-cd/pipe/pkg/cache"
+	"github.com/pipe-cd/pipe/pkg/redis"
+)
+
+type RedisHashCache struct {
+	RedisCache
+	key string
+}
+
+func NewHashCache(redis redis.Redis, ttl time.Duration, key string) *RedisHashCache {
+	return &RedisHashCache{
+		RedisCache: RedisCache{
+			redis: redis,
+		},
+		key: key,
+	}
+}
+
+func NewTTLHashCache(redis redis.Redis, ttl time.Duration, key string) *RedisHashCache {
+	return &RedisHashCache{
+		RedisCache: RedisCache{
+			redis: redis,
+			ttl:   uint(ttl.Seconds()),
+		},
+		key: key,
+	}
+}
+
+func (r *RedisHashCache) PutHash(k interface{}, v interface{}) error {
+	conn := r.redis.Get()
+	defer conn.Close()
+	_, err := conn.Do("HSET", r.key, k, v)
+	if r.ttl != 0 {
+		_, err = conn.Do("EXPIRE", r.key, r.ttl)
+	}
+	return err
+}
+
+func (r *RedisHashCache) GetAll() ([]interface{}, error) {
+	conn := r.redis.Get()
+	defer conn.Close()
+	reply, err := redigo.StringMap(conn.Do("HGETALL", r.key))
+	if err != nil {
+		if err == redigo.ErrNil {
+			return nil, cache.ErrNotFound
+		}
+		return nil, err
+	}
+	if len(reply) == 0 {
+		return nil, cache.ErrNotFound
+	}
+
+	out := make([]interface{}, 0, len(reply))
+	for _, v := range reply {
+		out = append(out, v)
+	}
+	return out, nil
+}