65 lines
1.2 KiB
Go
65 lines
1.2 KiB
Go
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
|
|
}
|