all

Downloads/Voltaserve/api/cache/group_cache.go

@@ -0,0 +1,57 @@
+package cache
+
+import (
+	"encoding/json"
+	"voltaserve/infra"
+	"voltaserve/model"
+	"voltaserve/repo"
+)
+
+type GroupCache struct {
+	redis     *infra.RedisManager
+	groupRepo repo.GroupRepo
+	keyPrefix string
+}
+
+func NewGroupCache() *GroupCache {
+	return &GroupCache{
+		redis:     infra.NewRedisManager(),
+		groupRepo: repo.NewGroupRepo(),
+		keyPrefix: "group:",
+	}
+}
+
+func (c *GroupCache) Set(workspace model.Group) error {
+	b, err := json.Marshal(workspace)
+	if err != nil {
+		return err
+	}
+	err = c.redis.Set(c.keyPrefix+workspace.GetID(), string(b))
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (c *GroupCache) Get(id string) (model.Group, error) {
+	value, err := c.redis.Get(c.keyPrefix + id)
+	if err != nil {
+		return c.Refresh(id)
+	}
+	group := repo.NewGroup()
+	if err = json.Unmarshal([]byte(value), &group); err != nil {
+		return nil, err
+	}
+	return group, nil
+}
+
+func (c *GroupCache) Refresh(id string) (model.Group, error) {
+	res, err := c.groupRepo.Find(id)
+	if err != nil {
+		return nil, err
+	}
+	if err = c.Set(res); err != nil {
+		return nil, err
+	}
+	return res, nil
+}