使用redis实现排行榜功能

一、需求背景

        最近项目需要做排行榜功能，实现员工邀请用户注册排行榜，要求是实时更新，查询要快。员工所属支行、二级行、省行，界面要根据条件显示排名数据。效果如下图所示：

原型图展示比较随意，用excel随便写了一下，凑合着看。

二、实现思路

 1、利用数据库

       建一张统计表，字段包括：邀请人、邀请人所属支行、邀请人所属二级行、被邀请人、注册时间等关键信息，用于sql统计排名，根据条件使用group by相关字段，比较简单，这个大家应该清楚。

        数据量小，统计效率还可以。但是支行下有十几万员工，一个员工邀请10个就百万数据，如果更多，数据就更大了，统计效率不高。下面重点讨论用第二种方式实现。

2、利用redis

       我们都知道redis基于内存实现的，查询效率极高，且支持多种数据类型，其中zset是本次实现功能的关键。

• ZSet也是String类型元素的集合，且不允许重复的成员；
• 不同的是每个元素都会关联一个double类型的分数，刚好也是我们需要的邀请用户数；
• 通过分数来为集合中的成员进行排序。ZSet的成员是唯一的，但分数（score）却可以重复；

    基于上面的特性，满足我们本次的需求。好了，说了一大堆废话，下面将进入正题。

    首先，捋一下查询条件，根据前面的效果图，可以看出有以下几种情况：

1. 二级行的全部排名以及日周月榜排名
2. 支行在省行的全部排名以及日周月榜排名
3. 支行在二级行的全部排名以及日周月榜排名
4. 员工在省行的全部排名以及日周月榜排名
5. 员工在二级行的全部排名以及日周月榜排名
6. 员工在支行的全部排名以及日周月榜排名

       基于redis的Zset函数incrementScore，我们很快就能发现，其实实现各个排名，只要把key规定好即可，例如：

• 员工在省行的全部排名key，可以设置为 rank:employee:省行
• 员工在省行的日排行榜key，可以设置为 rank:emploee:省行:当天日期

下面我们来实现其中的上面的两个排行：

1.  
   String key = "rank:employee:广东省";
2.  
   redisTemplate.opsForZSet().incrementScore(key, "张三", 10);
3.  
   redisTemplate.opsForZSet().incrementScore(key, "李四", 8);
4.  
   redisTemplate.opsForZSet().incrementScore(key, "王五", 5);

执行完后，redis会保存为如下数据：

 这样的话，有了上面的数据，可以用redis的提供的函数把排行榜查出来，代码如下：

1.  
   String key = "rank:employee:广东省";
2.  
   Set<ZSetOperations.TypedTuple<String>> set = redisTemplate.opsForZSet().reverseRangeWithScores(key, 0, -1);
3.  
   JSONArray jsonArray = JSONObject.parseArray(JSONObject.toJSONString(set));
4.  
   for(int i = 0, size = jsonArray.size(); i < size; i ) {
5.  
   JSONObject o = JSONObject.parseObject(jsonArray.get(i).toString());
6.  
   System.out.println("员工：" o.getString("value") ", 邀请人数：" o.getLongValue("score"));
7.  
   }

reverseRangeWithScores方法接收三个参数，第一个是key，后面两个是分页查询，起始是从0开始，-1表示全部，如果设置为0，4，那么就是查询前5条记录，查出结果如下：

以上就实现了员工在省行的排名。类似的，员工要实现在省行的日榜，代码如下：

1.  
   String key = "rank:employee:广东省:2022-09-01";
2.  
   redisTemplate.opsForZSet().incrementScore(key, "张三", 10);
3.  
   redisTemplate.opsForZSet().incrementScore(key, "李四", 8);
4.  
   redisTemplate.opsForZSet().incrementScore(key, "王五", 5);

 执行完后，redis会保存为如下数据（这里我多设置了前一天的数据）：

一样的，用 reverseRangeWithScores方法可以把上面的结果查出来。

至于周榜、月榜，可以把每一天的数据累加起来，再做个排名，redis已经帮我们实现了这个功能，代码如下：

1.  
   Date date = DateUtil.date();
2.  
   //获取本周的第一天
3.  
   DateTime beginOfWeek = DateUtil.beginOfWeek(date);
4.  
   //到今天一共有几天
5.  
   long diffDay = DateUtil.between(date, beginOfWeek, DateUnit.DAY) 1;
6.  
    
7.  
   List<String> keys = new ArrayList<>();
8.  
   for(int i = 0; i < diffDay; i ) {
9.  
   //把需要查询的天数放一起
10.  
    keys.add("rank:employee:广东省:" DateUtil.formatDate(DateUtil.offsetDay(beginOfWeek, i)));
11.  
    }
12.  
     
13.  
    //redis使用unionAndStore做合并，将结果集放在另一个的key，也就是第三个参数
14.  
    redisTemplate.opsForZSet().unionAndStore("weekRank", keys, "employeeRankWeek");
15.  
     
16.  
    //查询结果集用employeeRankWeek这个key
17.  
    Set<ZSetOperations.TypedTuple<String>> set = redisTemplate.opsForZSet().reverseRangeWithScores("employeeRankWeek", 0, -1);
18.  
    JSONArray jsonArray = JSONObject.parseArray(JSONObject.toJSONString(set));
19.  
    for(int i = 0, size = jsonArray.size(); i < size; i ) {
20.  
    JSONObject o = JSONObject.parseObject(jsonArray.get(i).toString());
21.  
    System.out.println("员工：" o.getString("value") ", 本周邀请人数：" o.getLongValue("score"));
22.  
    }

注意代码里面说的，redis会把结果合并到另一个key，在redis上也可以看到，如下图：

查出来的结果如下图：

其实我们会发现，本周榜、本月榜无需保存每一天的数据，只要把key设置为本周或本月的第一天就可以，因为添加数据的那一刻就知道是哪一周或哪月了。

例如：key = rank:employee:广东省:2022-08-29，8月29日是本周的第一天，无论你在接下来一周内邀请多少人，都是在本周内完成的，在这个基础上累加邀请数量即可。本月榜的逻辑也是一样。

查询的时候，获取当前时间本周或本月的第一天，就可以实现本周、本月排行了。

1.  
   String key = "rank:employee:广东省";
2.  
   Date date = DateUtil.date();
3.  
    
4.  
   String week = DateUtil.formatDate(DateUtil.beginOfWeek(date));
5.  
   String month = DateUtil.formatDate(DateUtil.beginOfMonth(date));
6.  
    
7.  
   //周榜
8.  
   redisTemplate.opsForZSet().incrementScore(key ":week:" week, "张三", 17);
9.  
   //月榜
10.  
    redisTemplate.opsForZSet().incrementScore(key ":month:" month, "张三", 17);

当然了，如果要查询历史的排行，这种设计就满足不了了，还是要保存每天的数据才行。

到此就结束了，有不足的地方，还请指出，谢谢！

这篇好文章是转载于：学新通技术网