all

Downloads/Voltaserve/api/cache/file_cache.go

@@ -0,0 +1,64 @@
+package cache
+
+import (
+	"encoding/json"
+	"voltaserve/infra"
+	"voltaserve/model"
+	"voltaserve/repo"
+)
+
+type FileCache struct {
+	redis     *infra.RedisManager
+	fileRepo  repo.FileRepo
+	keyPrefix string
+}
+
+func NewFileCache() *FileCache {
+	return &FileCache{
+		fileRepo:  repo.NewFileRepo(),
+		redis:     infra.NewRedisManager(),
+		keyPrefix: "file:",
+	}
+}
+
+func (c *FileCache) Set(file model.File) error {
+	b, err := json.Marshal(file)
+	if err != nil {
+		return err
+	}
+	err = c.redis.Set(c.keyPrefix+file.GetID(), string(b))
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (c *FileCache) Get(id string) (model.File, error) {
+	value, err := c.redis.Get(c.keyPrefix + id)
+	if err != nil {
+		return c.Refresh(id)
+	}
+	file := repo.NewFile()
+	if err = json.Unmarshal([]byte(value), &file); err != nil {
+		return nil, err
+	}
+	return file, nil
+}
+
+func (c *FileCache) Refresh(id string) (model.File, error) {
+	res, err := c.fileRepo.Find(id)
+	if err != nil {
+		return nil, err
+	}
+	if err = c.Set(res); err != nil {
+		return nil, err
+	}
+	return res, nil
+}
+
+func (c *FileCache) Delete(id string) error {
+	if err := c.redis.Delete(c.keyPrefix + id); err != nil {
+		return nil
+	}
+	return nil
+}