如何在Redis的ZRANK中获得相同分数的相同排名?

Ank*_*jra 5 redis spring-data-redis

如果我有5位成员的得分如下

a - 1
b - 2
c - 3
d - 3
e - 5
Run Code Online (Sandbox Code Playgroud)

c的ZRANK返回2,d的ZRANK返回3

对于相同的分数,是否可以获得相同的排名?
示例:ZRANK c = 2,d = 2,e = 3
如果是,那么如何在spring-data-redis中实现呢?

Ita*_*ber 4

任何真正的解决方案都需要满足原始问题中缺少的要求。我的第一个答案假设了一个小数据集,但这种方法无法扩展,因为至少在 O(N) 中完成了密集排名(例如通过 Lua)。

所以,假设有很多分数的用户,for_stack建议的方向更好,即多种数据结构的组合。我相信这就是他最后一句话的要点。

要存储用户的分数,您可以使用哈希。虽然从概念上讲,您可以使用单个键来存储所有用户分数的哈希,但实际上您需要对哈希进行哈希处理,以便它可以扩展。为了使这个示例简单,我将忽略哈希缩放。

这是在 Lua 中添加(更新)用户分数的方法:

local hscores_key = KEYS[1]
local user = ARGV[1]
local increment = ARGV[2]
local new_score = redis.call('HINCRBY', hscores_key, user, increment)
Run Code Online (Sandbox Code Playgroud)

接下来,我们想要跟踪每个离散评分值的当前用户数量,因此我们为此保留另一个散列:

local old_score = new_score - increment
local hcounts_key = KEYS[2]
local old_count = redis.call('HINCRBY', hcounts_key, old_score, -1)
local new_count = redis.call('HINCRBY', hcounts_key, new_score, 1)
Run Code Online (Sandbox Code Playgroud)

现在,我们需要维护的最后一件事是每个分数排名,以及排序集。每个新分数都会作为成员添加到 zset 中,并且删除没有更多用户的分数:

local zdranks_key = KEYS[3]
if new_count == 1 then
  redis.call('ZADD', zdranks_key, new_score, new_score)
end
if old_count == 0 then
  redis.call('ZREM', zdranks_key, old_score)
end
Run Code Online (Sandbox Code Playgroud)

由于使用了排序集,此 3 件脚本的复杂度为 O(logN),但请注意,N 是离散分数值的数量,而不是系统中的用户数量。获取用户的密集排名是通过另一个更短、更简单的脚本完成的:

local hscores_key = KEYS[1]
local zdranks_key = KEYS[2]
local user = ARGV[1]

local score = redis.call('HGET', hscores_key, user)
return redis.call('ZRANK', zdranks_key, score)
Run Code Online (Sandbox Code Playgroud)