catface_backend_go/app/model_redis/selectedAnimal4Prefer.go

77 lines
1.9 KiB
Go

package model_redis
import (
"catface/app/global/variable"
"catface/app/utils/redis_factory"
"encoding/json"
)
// INFO 辅助 animal - list - prefer 模式下的查询特化。
type SelectedAnimal4Prefer struct {
EncounteredCatsId []int64 `json:"encountered_cats_id"` // #1 对应第一阶段:近期路遇关联
NewCatsId []int64 `json:"new_cats_id"` // #2 对应第二阶段:近期新增
key int64 `json:"-"` // redis 的 key 值
notPassNewCat bool `json:"-"`
}
func (s *SelectedAnimal4Prefer) NotPassNew() bool {
return s.notPassNewCat
}
func (s *SelectedAnimal4Prefer) Length() int {
return len(s.NewCatsId) + len(s.EncounteredCatsId)
}
func (s *SelectedAnimal4Prefer) NumEnc() int {
return len(s.EncounteredCatsId)
}
func (s *SelectedAnimal4Prefer) GetAllIds() []int64 {
return append(s.EncounteredCatsId, s.NewCatsId...)
}
func (s *SelectedAnimal4Prefer) AppendEncIds(ids []int64) {
for _, id := range ids {
s.EncounteredCatsId = append(s.EncounteredCatsId, int64(id))
}
}
// BASE CURD
func (s *SelectedAnimal4Prefer) GenerateKey() int64 { // TODO 之后迁移到 model_redis 的基类去。
s.key = variable.SnowFlake.GetId()
s.notPassNewCat = true
return s.key
}
func (s *SelectedAnimal4Prefer) GetKey() int64 { // TODO 同上
return s.key
}
func (s *SelectedAnimal4Prefer) GetDataByKey(key int64) (bool, error) {
s.key = key
redisClient := redis_factory.GetOneRedisClient()
defer redisClient.ReleaseOneRedisClient()
if res, err := redisClient.String(redisClient.Execute("get", key)); err == nil {
json.Unmarshal([]byte(res), s)
return true, nil
} else {
return false, err
}
}
func (s *SelectedAnimal4Prefer) SetDataByKey() (bool, error) {
redisClient := redis_factory.GetOneRedisClient()
defer redisClient.ReleaseOneRedisClient()
data, _ := json.Marshal(s)
if _, err := redisClient.Execute("set", s.key, data); err == nil {
return true, nil
} else {
return false, err
}
}