简单限流接口

限流接口 - 结合redis对接口进行限流

应用场景：对接口进行限制，比如说一分钟只能发送5条短信，一个接口一分钟之内只能调用2次。

/**
 * @Description: 限流接口 - 结合redis对接口使用限流
 * @Author: yufengpeng
 * @CreateDate: 2019/6/26 11:08
 */
public class RateLimitUtil {

    @Autowired
    private RedisTemplate<String, Object> redisTemplate;

    /**
     * 对接口进行限制：比如说一分钟只能发送5条短信，一个接口一分钟之内只能调用2次等
     * Redis zSet数据结构  zCard 查询缓存中在时间窗口内的请求数量不能超过 maxCount
     *
     * @param address  请求地址IP + 接口URI = address
     * @param period   时间间隔
     * @param maxCount 最大请求次数
     * @return boolean  是否允许访问
     */
    public boolean isActionAllowed(String address, int period, int maxCount) {

        //取执行操作的第四个值
        int index = 3;

        List<Object> result = redisTemplate.executePipelined(new SessionCallback<Object>() {
            @Override
            public <K, V> Object execute(RedisOperations<K, V> redisOperations) throws DataAccessException {
                //获取当前时间
                long nowTs = System.currentTimeMillis();
                String key = CacheConstant.RATE_LIMIT + address;

                //定义redis的Zset操作
                ZSetOperations operations = redisOperations.opsForZSet();
                //1、加入到缓存
                operations.add(key, UUID.randomUUID().toString(), nowTs);
                //2、设置有效时间
                redisTemplate.expire(key, period + 1, TimeUnit.SECONDS);
                //3、超时将其从域中移除
                operations.removeRangeByScore(key, 0, nowTs - period * 1000);
                //4、获取有序集的大小
                operations.zCard(key);

                return null;
            }
        });
        //统计当前时间窗口的数据
        int count = Integer.parseInt(String.valueOf(result.get(index)));
        return count <= maxCount;
    }
}

对一分钟只能发送5条短信，也可以参考这篇文章：结合redis限制短信发送次数