79 lines
1.9 KiB
Go
79 lines
1.9 KiB
Go
package dao
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
|
||
"go-common/library/cache/redis"
|
||
"go-common/library/log"
|
||
)
|
||
|
||
func sortedKey(categoryID int64, field int) string {
|
||
return fmt.Sprintf("art_sort_%d_%d", categoryID, field)
|
||
}
|
||
|
||
// ExpireSortCache expire sort cache
|
||
func (d *Dao) ExpireSortCache(c context.Context, categoryID int64, field int) (ok bool, err error) {
|
||
key := sortedKey(categoryID, field)
|
||
conn := d.artRedis.Get(c)
|
||
defer conn.Close()
|
||
var ttl int64
|
||
if ttl, err = redis.Int64(conn.Do("TTL", key)); err != nil {
|
||
PromError("redis:排序缓存ttl")
|
||
log.Error("conn.Do(TTL, %s) error(%+v)", key, err)
|
||
}
|
||
if ttl > (d.redisSortTTL - d.redisSortExpire) {
|
||
ok = true
|
||
}
|
||
return
|
||
}
|
||
|
||
// AddSortCaches add sort articles cache
|
||
func (d *Dao) AddSortCaches(c context.Context, categoryID int64, field int, arts [][2]int64, maxLength int64) (err error) {
|
||
var (
|
||
id, score int64
|
||
key = sortedKey(categoryID, field)
|
||
conn = d.artRedis.Get(c)
|
||
count int
|
||
)
|
||
defer conn.Close()
|
||
if len(arts) == 0 {
|
||
return
|
||
}
|
||
if err = conn.Send("DEL", key); err != nil {
|
||
PromError("redis:删除排序缓存")
|
||
log.Error("conn.Send(DEL, %s) error(%+v)", key, err)
|
||
return
|
||
}
|
||
count++
|
||
for _, art := range arts {
|
||
id = art[0]
|
||
score = art[1]
|
||
if err = conn.Send("ZADD", key, "CH", score, id); err != nil {
|
||
PromError("redis:增加排序缓存")
|
||
log.Error("conn.Send(ZADD, %s, %d, %v) error(%+v)", key, score, id, err)
|
||
return
|
||
}
|
||
count++
|
||
}
|
||
if err = conn.Send("EXPIRE", key, d.redisSortTTL); err != nil {
|
||
PromError("redis:排序缓存设定过期")
|
||
log.Error("conn.Send(EXPIRE, %s, %d) error(%+v)", key, d.redisSortTTL, err)
|
||
return
|
||
}
|
||
count++
|
||
if err = conn.Flush(); err != nil {
|
||
PromError("redis:增加排序缓存flush")
|
||
log.Error("conn.Flush error(%+v)", err)
|
||
return
|
||
}
|
||
for i := 0; i < count; i++ {
|
||
if _, err = conn.Receive(); err != nil {
|
||
PromError("redis:增加排序缓存receive")
|
||
log.Error("conn.Receive error(%+v)", err)
|
||
return
|
||
}
|
||
}
|
||
return
|
||
}
|