58 lines
1.1 KiB
Go
58 lines
1.1 KiB
Go
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
|
|
}
|