简单限流

image.png

  2.     /**
  3.      *  判断用户行为是否达到阈值
  4.      *
  5.      * @param userId    用户id
  6.      * @param action    用户行为
  7.      * @param maxCount  次数阈值
  8.      * @param period    限制时间窗口大小
  9.      * @return
  10.      */
  11.     private boolean is_action_allowed(String userId, String action,  long maxCount, int period) {
  12.         String key = userId + ":" + action;
  13.         // nowTs ~ targetTs 规定时间窗口
  14.         long nowTs = System.currentTimeMillis();
  15.         long edgeTs = nowTs - period * 1000;
  18.         // 设置新值 zadd key value score
  19.         stringRedisTemplate.opsForZSet().add(key, String.valueOf(nowTs), nowTs);
  21.         // 去除时间窗口外的记录 zremrangebyscore(key, min, max) 删除 score 为 [min, max]  的数据
  22.         // 时间窗口就是 [beforeTs, nowTs]
  23.         stringRedisTemplate.opsForZSet().removeRangeByScore(key, 0, edgeTs);
  25.         // 获取时间窗口内的数量 zcard key
  26.         Long count = stringRedisTemplate.opsForZSet().zCard(key);
  28.         // 设置过期时间，便于后期剔除冷用户
  29.         stringRedisTemplate.expire(key, period, TimeUnit.SECONDS);
  31.         if (Objects.isNull(count)) {
  32.             return true;
  33.         } else {
  34.             // 如果时间窗口内的数量小于阈值, 说明可以操作
  35.             return count <= maxCount;
  36.         }
  37.     }
  1.    @Test
  2.     public void testLimit() {
  3.         String userId = "2020";
  4.         String action = "write";
  5.         boolean ret;
  6.         for (int i=0; i<16; i++) {
  7.             ret = is_action_allowed(userId, action, 20, 10);
  8.             if (!ret) {
  9.                 System.out.println("限流! 当前次数: " + i);
  10.             }
  11.         }
  12.     }