You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
68 lines
1.9 KiB
68 lines
1.9 KiB
package dao |
|
|
|
import ( |
|
"context" |
|
"fmt" |
|
"go-common/library/cache/redis" |
|
"go-common/library/log" |
|
) |
|
|
|
const ( |
|
// r_<实验组名>_<oid>_<type> |
|
// 用redis ZSet存储热门评论列表，score为热评分数，member为rpID |
|
_replyZSetFormat = "r_%s_%d_%d" |
|
) |
|
|
|
func keyReplyZSet(name string, oid int64, tp int) string { |
|
return fmt.Sprintf(_replyZSetFormat, name, oid, tp) |
|
} |
|
|
|
// PingRedis redis health check. |
|
func (d *Dao) PingRedis(ctx context.Context) (err error) { |
|
conn := d.redis.Get(ctx) |
|
defer conn.Close() |
|
_, err = conn.Do("SET", "ping", "pong") |
|
return |
|
} |
|
|
|
// ExpireReplyZSetRds expire reply list. |
|
func (d *Dao) ExpireReplyZSetRds(ctx context.Context, name string, oid int64, tp int) (ok bool, err error) { |
|
conn := d.redis.Get(ctx) |
|
defer conn.Close() |
|
key := keyReplyZSet(name, oid, tp) |
|
if ok, err = redis.Bool(conn.Do("EXPIRE", key, d.redisReplyZSetExpire)); err != nil { |
|
if err == redis.ErrNil { |
|
err = nil |
|
return |
|
} |
|
log.Error("redis EXPIRE key(%s) error(%v)", key, err) |
|
} |
|
return |
|
} |
|
|
|
// ReplyZSetRds get reply list from redis sorted set. |
|
func (d *Dao) ReplyZSetRds(ctx context.Context, name string, oid int64, tp, start, end int) (rpIDs []int64, err error) { |
|
conn := d.redis.Get(ctx) |
|
defer conn.Close() |
|
key := keyReplyZSet(name, oid, tp) |
|
values, err := redis.Values(conn.Do("ZREVRANGE", key, start, end)) |
|
if err != nil { |
|
log.Error("redis ZREVRANGE(%s, %d, %d) error(%v)", key, start, end, err) |
|
return |
|
} |
|
if err = redis.ScanSlice(values, &rpIDs); err != nil { |
|
log.Error("redis ScanSlice(%v) error(%v)", values, err) |
|
} |
|
return |
|
} |
|
|
|
// CountReplyZSetRds count reply num. |
|
func (d *Dao) CountReplyZSetRds(ctx context.Context, name string, oid int64, tp int) (count int, err error) { |
|
conn := d.redis.Get(ctx) |
|
defer conn.Close() |
|
key := keyReplyZSet(name, oid, tp) |
|
if count, err = redis.Int(conn.Do("ZCARD", key)); err != nil { |
|
log.Error("conn.Do(ZCARD, %s) error(%v)", key, err) |
|
} |
|
return |
|
}
|
|
|