Redis——构建简单限流和漏斗限流

最新推荐文章于 2024-10-08 09:05:36 发布

吴声子夜歌

最新推荐文章于 2024-10-08 09:05:36 发布

阅读量630

点赞数 2

分类专栏： Redis 文章标签： redis java redis-cell 限流

本文链接：https://blog.csdn.net/cold___play/article/details/130190672

版权

Redis 专栏收录该内容

29 篇文章 2 订阅

订阅专栏

文章介绍了如何使用Redis的zset数据结构结合时间戳来实现限流功能，限制用户在特定时间内的行为次数。此外，还探讨了漏斗限流算法，并提供了单机漏斗算法的实现。最后，提到了Redis-Cell模块，这是一个用于分布式限流的解决方案，提供原子性的限流操作。

摘要由CSDN通过智能技术生成

限流

限流算法在分布式领域是一个经常被提起的话题，当系统的处理能力有限时，如何阻止计划外的请求继续对系统施压，这是一个需要重视的问题。老钱在这里用 “断尾求生” 形容限流背后的思想，当然还有很多成语也表达了类似的意思，如弃卒保车、壮士断腕等等。

除了控制流量，限流还有一个应用目的是用于控制用户行为，避免垃圾请求。比如在 UGC 社区，用户的发帖、回复、点赞等行为都要严格受控，一般要严格限定某行为在规定时间内允许的次数，超过了次数那就是非法行为。对非法行为，业务必须规定适当的惩处策略。

如何使用 Redis 来实现简单限流策略?

系统要限定用户的某个行为在指定的时间里只能允许发生 N 次，如何使用 Redis 的数据结构来实现这个限流的功能?

# 指定用户 user_id 的某个行为 action_key 在特定的时间内 period 只允许发生一定的次数
max_count
def is_action_allowed(user_id, action_key, period, max_count):
	return True
# 调用这个接口 , 一分钟内只允许最多回复 5 个帖子
can_reply = is_action_allowed("laoqian", "reply", 60, 5) 
if can_reply:
	do_reply() 
else:
	raise ActionThresholdOverflow()

这个限流需求中存在一个滑动时间窗口，想想 zset 数据结构的 score 值，是不是可以通过 score 来圈出这个时间窗口来。而且我们只需要保留这个时间窗口，窗口之外的数据都可以砍掉。那这个 zset 的 value 填什么比较合适呢?它只需要保证唯一性即可，用 uuid 会比较浪费空间，那就改用毫秒时间戳吧。

在这里插入图片描述

如图所示，用一个 zset 结构记录用户的行为历史，每一个行为都会作为 zset 中的一个 key 保存下来。同一个用户同一种行为用一个 zset 记录。
为节省内存，我们只需要保留时间窗口内的行为记录，同时如果用户是冷用户，滑动时间窗口内的行为是空记录，那么这个 zset 就可以从内存中移除，不再占用空间。
通过统计滑动窗口内的行为数量与阈值 max_count 进行比较就可以得出当前的行为是否允许。用代码表示如下:

package com.example.demo.book;

import redis.clients.jedis.Jedis;
import redis.clients.jedis.Pipeline;
import redis.clients.jedis.Response;

/**
 * @Author: acton_zhang
 * @Date: 2023/4/16 11:56 下午
 * @Version 1.0
 * 简单限流
 */
public class SimpleRateLimiter {

    private Jedis jedis;

    public SimpleRateLimiter(Jedis jedis) {
        this.jedis = jedis;
    }

    public boolean isActionAllowed(String userId, String actionKey, int period, int maxCount) {
        String key = String.format("hist:%s:%s", userId, actionKey);
        //毫秒时间戳
        long nowTs = System.currentTimeMillis();
        Pipeline pipe = jedis.pipelined();
        pipe.multi();
        //记录行为
        pipe.zadd(key, nowTs, "" + nowTs);
        //移除时间窗口之前的行为记录，剩下的都是时间窗口内的
        pipe.zremrangeByScore(key, 0, nowTs - period * 1000);
        //获取窗口内的行为数量
        Response<Long> count = pipe.zcard(key);
        //设置zset过期时间，避免冷用户持续占用内存
        //过期时间应该等于时间窗口的长度，再多宽限1S
        pipe.expire(key, period + 1);
        //批量执行
        pipe.exec();
        pipe.close();
        //比较数量是否超标
        return count.get() <= maxCount;
    }

    public static void main(String[] args) {
        Jedis jedis = new Jedis("127.0.0.1", 6379);
        jedis.auth("123456");
        SimpleRateLimiter limiter = new SimpleRateLimiter(jedis);
        for (int i = 0; i < 20; i++) {
            System.out.println(limiter.isActionAllowed("laoqian", "reply", 60, 5));
        }
    }
}

输出：

/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/bin/java "-javaagent:/Applications/IntelliJ IDEA.app/Contents/lib/idea_rt.jar=62021:/Applications/IntelliJ IDEA.app/Contents/bin" -Dfile.encoding=UTF-8 -classpath /Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/charsets.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/deploy.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/ext/cldrdata.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/ext/dnsns.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/ext/jaccess.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/ext/jfxrt.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/ext/localedata.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/ext/nashorn.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/ext/sunec.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/ext/sunjce_provider.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/ext/sunpkcs11.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/ext/zipfs.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/javaws.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/jce.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/jfr.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/jfxswt.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/jsse.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/management-agent.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/plugin.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/resources.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/jre/lib/rt.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/lib/ant-javafx.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/lib/dt.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/lib/javafx-mx.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/lib/jconsole.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/lib/packager.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/lib/sa-jdi.jar:/Library/Java/JavaVirtualMachines/jdk1.8.0_211.jdk/Contents/Home/lib/tools.jar:/Users/acton_zhang/J2EE/IdeaWrokSpace/sb_redis_demo/target/classes:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/boot/spring-boot-starter-web/2.7.4/spring-boot-starter-web-2.7.4.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/boot/spring-boot-starter/2.7.4/spring-boot-starter-2.7.4.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/boot/spring-boot/2.7.4/spring-boot-2.7.4.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/boot/spring-boot-autoconfigure/2.7.4/spring-boot-autoconfigure-2.7.4.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/boot/spring-boot-starter-logging/2.7.4/spring-boot-starter-logging-2.7.4.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/ch/qos/logback/logback-classic/1.2.11/logback-classic-1.2.11.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/ch/qos/logback/logback-core/1.2.11/logback-core-1.2.11.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/apache/logging/log4j/log4j-to-slf4j/2.17.2/log4j-to-slf4j-2.17.2.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/apache/logging/log4j/log4j-api/2.17.2/log4j-api-2.17.2.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/slf4j/jul-to-slf4j/1.7.36/jul-to-slf4j-1.7.36.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/jakarta/annotation/jakarta.annotation-api/1.3.5/jakarta.annotation-api-1.3.5.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/yaml/snakeyaml/1.30/snakeyaml-1.30.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/boot/spring-boot-starter-json/2.7.4/spring-boot-starter-json-2.7.4.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/com/fasterxml/jackson/core/jackson-databind/2.13.4/jackson-databind-2.13.4.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/com/fasterxml/jackson/core/jackson-annotations/2.13.4/jackson-annotations-2.13.4.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/com/fasterxml/jackson/core/jackson-core/2.13.4/jackson-core-2.13.4.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/com/fasterxml/jackson/datatype/jackson-datatype-jdk8/2.13.4/jackson-datatype-jdk8-2.13.4.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/com/fasterxml/jackson/datatype/jackson-datatype-jsr310/2.13.4/jackson-datatype-jsr310-2.13.4.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/com/fasterxml/jackson/module/jackson-module-parameter-names/2.13.4/jackson-module-parameter-names-2.13.4.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/boot/spring-boot-starter-tomcat/2.7.4/spring-boot-starter-tomcat-2.7.4.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/apache/tomcat/embed/tomcat-embed-core/9.0.65/tomcat-embed-core-9.0.65.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/apache/tomcat/embed/tomcat-embed-el/9.0.65/tomcat-embed-el-9.0.65.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/apache/tomcat/embed/tomcat-embed-websocket/9.0.65/tomcat-embed-websocket-9.0.65.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/spring-web/5.3.23/spring-web-5.3.23.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/spring-beans/5.3.23/spring-beans-5.3.23.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/spring-webmvc/5.3.23/spring-webmvc-5.3.23.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/spring-aop/5.3.23/spring-aop-5.3.23.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/spring-context/5.3.23/spring-context-5.3.23.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/spring-expression/5.3.23/spring-expression-5.3.23.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/hamcrest/hamcrest/2.2/hamcrest-2.2.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/spring-core/5.3.23/spring-core-5.3.23.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/spring-jcl/5.3.23/spring-jcl-5.3.23.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/boot/spring-boot-starter-data-redis/2.7.4/spring-boot-starter-data-redis-2.7.4.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/data/spring-data-redis/2.7.3/spring-data-redis-2.7.3.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/data/spring-data-keyvalue/2.7.3/spring-data-keyvalue-2.7.3.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/data/spring-data-commons/2.7.3/spring-data-commons-2.7.3.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/spring-tx/5.3.23/spring-tx-5.3.23.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/spring-oxm/5.3.23/spring-oxm-5.3.23.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/springframework/spring-context-support/5.3.23/spring-context-support-5.3.23.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/redis/clients/jedis/3.8.0/jedis-3.8.0.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/slf4j/slf4j-api/1.7.36/slf4j-api-1.7.36.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/apache/commons/commons-pool2/2.11.1/commons-pool2-2.11.1.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/com/alibaba/fastjson/1.2.66/fastjson-1.2.66.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/com/redislabs/jrebloom/1.2.0/jrebloom-1.2.0.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/com/google/code/gson/gson/2.9.1/gson-2.9.1.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/junit/junit/4.13.2/junit-4.13.2.jar:/Users/acton_zhang/software/apache-maven-3.6.3/repository/org/hamcrest/hamcrest-core/2.2/hamcrest-core-2.2.jar com.example.demo.book.SimpleRateLimiter
true
true
true
true
true
true
true
true
true
true
true
false
false
false
false
false
false
false
false
false

这段代码还是略显复杂，需要读者花一定的时间好好啃。它的整体思路就是:每一个行为到来时，都维护一次时间窗口。将时间窗口外的记录全部清理掉，只保留窗口内的记录。

zset 集合中只有 score 值非常重要，value 值没有特别的意义，只需要保证它是唯一的就可以了。
因为这几个连续的 Redis 操作都是针对同一个 key 的，使用 pipeline 可以显著提升 Redis 存取效率。但这种方案也有缺点，因为它要记录时间窗口内所有的行为记录，如果这个量很大，比如限定 60s 内操作不得超过 100w 次这样的参数，它是不适合做这样的限流的，因为会消耗大量的存储空间。

漏斗限流

漏斗限流是最常用的限流方法之一，顾名思义，这个算法的灵感源于漏斗(funnel)的结
构。

漏洞的容量是有限的，如果将漏嘴堵住，然后一直往里面灌水，它就会变满，直至再也装不进去。如果将漏嘴放开，水就会往下流，流走一部分之后，就又可以继续往里面灌水。如果漏嘴流水的速率大于灌水的速率，那么漏斗永远都装不满。如果漏嘴流水速率小于灌水的速率，那么一旦漏斗满了，灌水就需要暂停并等待漏斗腾空。

所以，漏斗的剩余空间就代表着当前行为可以持续进行的数量，漏嘴的流水速率代表着系统允许该行为的最大频率。下面我们使用代码来描述单机漏斗算法。

package com.example.demo.book;

import java.util.HashMap;
import java.util.Map;

/**
 * @Author: acton_zhang
 * @Date: 2023/4/17 12:53 上午
 * @Version 1.0
 * 漏斗限流
 */
public class FunnelRateLimiter {
    static class Funnel {
        int capacity; //漏斗容量
        float leakingRate; //漏斗流水速率
        int leftQuota; //漏斗剩余空间
        long leakingTs; //上一次漏水时间
        
        public Funnel(int capacity, float leakingRate) {
            this.capacity = capacity;
            this.leakingRate = leakingRate;
            this.leftQuota = capacity;
            this.leakingTs = System.currentTimeMillis();
        }
        
        void makeSpace() {
            long nowTs = System.currentTimeMillis();
            //距离上一次漏水过去了多久
            long deltaTs = nowTs - leakingTs;
            //上一次漏水后腾出的空间
            int deltaQuota = (int)(deltaTs * leakingRate);
            //间隔时间太长，整数数字过大溢出，整个漏斗空了
            if (deltaQuota < 0) {
                this.leftQuota = capacity;
                this.leakingTs = nowTs;
                return;
            }
            //腾出空间太小，最小单位是1
            if (deltaQuota < 1) {
                return;
            }
            //增加空间
            this.leftQuota += deltaQuota;
            //记录漏水时间
            this.leakingTs = nowTs;
            //剩余空间不得高于容量
            if (this.leftQuota > this.capacity) {
                this.leftQuota = this.capacity;
            }
        }
        
        boolean watering(int quota) {
            makeSpace();
            //判断剩余空间是否足够
            if (this.leftQuota >= quota) {
                this.leftQuota -= quota;
                return true;
            }
            return false;
        }
    }
    
    private Map<String, Funnel> funnels = new HashMap<>();//所有的漏斗

    /**
     * 
     * @param userId 
     * @param actionKey
     * @param capacity 漏斗容量
     * @param leakingRate 漏嘴流水速率 quota/s
     * @return
     */
    public boolean isActonAllowed(String userId, String actionKey, int capacity, float leakingRate) {
        String key = String.format("%s:%s", userId, actionKey);
        Funnel funnel = funnels.get(key);
        if (funnel == null) {
            funnel = new Funnel(capacity, leakingRate);
            funnels.put(key, funnel);
        }
        //需要一个quota
        return funnel.watering(1);
    }
}

Funnel 对象的 make_space 方法是漏斗算法的核心，其在每次灌水前都会被调用以触发漏水，给漏斗腾出空间来。能腾出多少空间取决于过去了多久以及流水的速率。Funnel 对象占据的空间大小不再和行为的频率成正比，它的空间占用是一个常量。

问题来了，分布式的漏斗算法该如何实现?能不能使用 Redis 的基础数据结构来搞定?
我们观察 Funnel 对象的几个字段，我们发现可以将 Funnel 对象的内容按字段存储到一个 hash 结构中，灌水的时候将 hash 结构的字段取出来进行逻辑运算后，再将新值回填到 hash 结构中就完成了一次行为频度的检测。

但是有个问题，我们无法保证整个过程的原子性。从 hash 结构中取值，然后在内存里运算，再回填到 hash 结构，这三个过程无法原子化，意味着需要进行适当的加锁控制。而一旦加锁，就意味着会有加锁失败，加锁失败就需要选择重试或者放弃。
如果重试的话，就会导致性能下降。如果放弃的话，就会影响用户体验。同时，代码的复杂度也跟着升高很多。这真是个艰难的选择，我们该如何解决这个问题呢?Redis-Cell 救星来了!

Redis-Cell

Redis 4.0 提供了一个限流 Redis 模块，它叫 redis-cell。该模块也使用了漏斗算法，并提供了原子的限流指令。有了这个模块，限流问题就非常简单了。

安装：
linux下载地址：https://github.com/brandur/redis-cell/releases/download/v0.2.5/redis-cell-v0.2.5-x86_64-unknown-linux-gnu.tar.gz

mac下载地址：https://github.com/brandur/redis-cell/releases/download/v0.2.5/redis-cell-v0.2.5-x86_64-apple-darwin.tar.gz

解压后获取libredis_cell.so或libredis_cell.dylib文件。

添加redis.conf配置，启动redis即可

 loadmodule ../etc/libredis_cell.dylib

在这里插入图片描述

基本使用：
该模块只有 1 条指令 cl.throttle，它的参数和返回值都略显复杂，接下来让我们来看看这个指令具体该如何使用。

#语法
cl.throttle key 令牌桶容量-1 令牌产生个数 令牌产生时间 本次取走的令牌数（默认1，负数为放入令牌）

#返回
0成功，1失败
令牌桶容量
当前桶内剩余的令牌数
成功时该值为-1，失败时表示还需要等待多少秒可以有足够的令牌
表示预计多少秒后令牌桶会满

>c6.throttle user:reply 15 30 60 1
1) (integer) 0 # 0 表示允许，1 表示拒绝
2) (integer) 16 # 漏斗容量 capacity
3) (integer) 15 # 漏斗剩余空间 left_quota
4) (integer) -1 # 如果拒绝了，需要多长时间后再试(漏斗有空间了，单位秒)
5) (integer) 2 # 多长时间后，漏斗完全空出来(left_quota==capacity，单位秒)
 
上面这个指令的意思是允许「用户」的频率为每 60s 最多 30 次(漏水速 率)，漏斗的初始容量为 15，也就是说一开始可以连续回复 15 个帖子，然后才开始受漏水 速率的影响。

也可以理解为：

桶容量为16，每60秒产生30个令牌。本次取走一个，还剩15个令牌，2秒后令牌将桶装满(0.5s产生一个令牌)。

在执行限流指令时，如果被拒绝了，就需要丢弃或重试。cl.throttle 指令考虑的非常周到，连重试时间都帮你算好了，直接取返回结果数组的第四个值进行 sleep 即可，如果不想阻塞线程，也可以异步定时任务来重试

Java实现：

package com.example.demo.book;

import redis.clients.jedis.HostAndPort;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisCluster;
import redis.clients.jedis.JedisPoolConfig;
import redis.clients.jedis.commands.JedisCommands;

import java.io.IOException;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

/**
 * @Author: acton_zhang
 * @Date: 2023/4/17 2:27 上午
 * @Version 1.0
 */
public class RedisCellDistributedRateLimiter {

    //jedis命令客户端
    protected JedisCommands jedis;
    //自定义限流key
    protected String key;
    //时间窗口 单位秒
    protected int period;
    //最大令牌容量
    protected int maxCapacity;

    public RedisCellDistributedRateLimiter(JedisCommands jedis, String key, int period, int maxCapacity) {
        this.jedis = jedis;
        this.key = key;
        this.period = period;
        this.maxCapacity = maxCapacity;
    }

    public boolean tryAcquire(int quote) throws IOException {
        List<String> keys = new ArrayList<>();
        keys.add(key);
        List<String> argvs = new ArrayList<>();
        argvs.add(String.valueOf(maxCapacity));
        argvs.add(String.valueOf(maxCapacity));
        argvs.add(String.valueOf(period));
        argvs.add(String.valueOf(quote));
        Object result = null;
        if (jedis instanceof Jedis) {
            result = ((Jedis) this.jedis).eval(LUA_SCRIPT, keys, argvs);
        } else if (jedis instanceof JedisCluster) {
            result = ((JedisCluster) this.jedis).eval(LUA_SCRIPT, keys, argvs);
        } else {
            throw new RuntimeException("redis instance is error") ;
        }
        return ((List<Long>)result).get(0) == 0;
    }

    //lua脚本
    public static final String LUA_SCRIPT = "local key = KEYS[1]\n"+
            "local init_burst = tonumber(ARGV[1])\n"+
            "local max_burst = tonumber(ARGV[2])\n"+
            "local period = tonumber(ARGV[3])\n"+
            "local quota = ARGV[4]\n"+
            "return redis.call('CL.THROTTLE',key,init_burst,max_burst,period,quota)";


    public static void main(String[] args) throws Exception{
        //构造Jedis集群
        // Set<HostAndPort> nodeList  = new HashSet<>();
        // nodeList.add( new HostAndPort("127.0.0.1", 6237));
        // JedisCluster jedisCluster =
        //         new JedisCluster(nodeList,10000,10000,10,"****", new JedisPoolConfig());

        Jedis jedis = new Jedis("127.0.0.1", 6379);
        jedis.auth("123456");
        String key = "redis-cell-test";
        int period = 5;
        int maxCapacity = 1;
        //构造限流器
        RedisCellDistributedRateLimiter limiter =
                new RedisCellDistributedRateLimiter(jedis, key, period, maxCapacity);
        //计数器
        int pass = 0;
        int nopass = 0;
        for (;;){
            Thread.sleep(1000);
            if (limiter.tryAcquire(1)){
                pass++;
                System.out.println("第" + pass + "次通过限流");
            }else {
                nopass++;
                System.out.println("第" + nopass + "次无法通过");
            }
        }
    }
}

第1次通过限流
第2次通过限流
第1次无法通过
第2次无法通过
第3次无法通过
第3次通过限流
第4次无法通过
第5次无法通过
第6次无法通过
第7次无法通过
第4次通过限流
第8次无法通过
第9次无法通过
第10次无法通过
第11次无法通过
第5次通过限流
第12次无法通过
第13次无法通过
第14次无法通过

平均4秒通过一次。