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
|
|||
|
|
}
|