95 lines
2.8 KiB
Go
95 lines
2.8 KiB
Go
package rankings
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/redis/go-redis/v9"
|
|
"strconv"
|
|
)
|
|
|
|
// Ranking 排行榜结构体
|
|
type Ranking struct {
|
|
c *redis.Client
|
|
}
|
|
|
|
const EcpmRankingsListPrefix = "ecpm:rankings:"
|
|
|
|
// NewRanking 创建一个新的排行榜实例
|
|
func NewRanking(c *redis.Client) *Ranking {
|
|
return &Ranking{
|
|
c: c,
|
|
}
|
|
}
|
|
|
|
func GetRankingsCacheKey(appId uint32, t uint32) string {
|
|
return fmt.Sprintf("%sappId:%d:type:%d", EcpmRankingsListPrefix, appId, t)
|
|
}
|
|
|
|
// SetList 向排行榜中添加成员及其分数
|
|
func (r *Ranking) SetList(ctx context.Context, key string, data ...redis.Z) {
|
|
r.c.ZAdd(ctx, key, data...)
|
|
}
|
|
|
|
// GetList 获取排行榜，按照分数从高到低排序
|
|
func (r *Ranking) GetList(ctx context.Context, key string, start, stop int64) (data []redis.Z, err error) {
|
|
return r.c.ZRevRangeWithScores(ctx, key, start, stop).Result()
|
|
}
|
|
|
|
// GetRank 获取指定成员在排行榜中的排名（排名从 0 开始，分数越高排名越靠前）
|
|
func (r *Ranking) GetRank(ctx context.Context, key, member string) (rank int64, err error) {
|
|
return r.c.ZRevRank(ctx, key, member).Result()
|
|
}
|
|
|
|
// GetRankAndScore 获取指定成员在排行榜中的排名（排名从 0 开始，分数越高排名越靠前）
|
|
func (r *Ranking) GetRankAndScore(ctx context.Context, key, member string) (rank int64, score float64, err error) {
|
|
rankScore := r.c.ZRevRankWithScore(ctx, key, member).Val()
|
|
return rankScore.Rank, rankScore.Score, nil
|
|
}
|
|
|
|
// GetScore 获取指定成员在排行榜中的分数
|
|
func (r *Ranking) GetScore(ctx context.Context, key, member string) (score float64, err error) {
|
|
return r.c.ZScore(ctx, key, member).Result()
|
|
}
|
|
|
|
// IncrScore 增加分数
|
|
func (r *Ranking) IncrScore(ctx context.Context, key, member string, increment float64) (score float64, err error) {
|
|
return r.c.ZIncrBy(ctx, key, increment, member).Result()
|
|
}
|
|
|
|
// Remove 删除排行榜
|
|
func (r *Ranking) Remove(ctx context.Context, key string) (err error) {
|
|
return r.c.Del(ctx, key).Err()
|
|
}
|
|
|
|
// 使用 Lua 保证原子性操作
|
|
var getHigherByScoreScript = redis.NewScript(`
|
|
local members = redis.call('ZRANGEBYSCORE', KEYS[1], ARGV[1],'+inf','LIMIT', 0, 1, 'WITHSCORES')
|
|
if #members == 0 then return nil end
|
|
return members
|
|
`)
|
|
|
|
// AtomicGetHigherUser 获取排行榜的前一名
|
|
func (r *Ranking) AtomicGetHigherUser(ctx context.Context, key string, score float64) (userId uint64, scoreRes uint32, err error) {
|
|
s, err := getHigherByScoreScript.Run(ctx, r.c, []string{key}, score).Slice()
|
|
if err != nil {
|
|
return
|
|
}
|
|
if s == nil {
|
|
return
|
|
}
|
|
|
|
if len(s) != 2 {
|
|
return 0, 0, fmt.Errorf("redis type error len(s) !=2")
|
|
}
|
|
|
|
userId, err = strconv.ParseUint(s[0].(string), 10, 64)
|
|
if err != nil {
|
|
return
|
|
}
|
|
tmp, err := strconv.ParseUint(s[1].(string), 10, 64)
|
|
if err != nil {
|
|
return
|
|
}
|
|
return userId, uint32(tmp >> 32), nil
|
|
}
|