利用redis的表达式实现速率请求控制器

最新推荐文章于 2025-01-10 17:03:21 发布

罗小粥

最新推荐文章于 2025-01-10 17:03:21 发布

阅读量875

点赞数

CC 4.0 BY-SA版权

分类专栏： java 文章标签： redis JAVA

本文链接：https://blog.youkuaiyun.com/lz1204518/article/details/76431463

java 专栏收录该内容

4 篇文章

订阅专栏

本文介绍了如何利用Redis 2.6.0版本后的Lua解释器，通过EVAL命令来编写脚本实现速率请求控制器。示例中展示了具体的JAVA Redis工具类和限速脚本的使用，确保了在第三次请求时达到限速效果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Redis 2.6.0 版本开始，通过内置的 Lua 解释器，可以使用 EVAL命令对 Lua 脚本进行求值,废话不多说，直接上源码。

1.redis工具类：

   /**redis 脚本扩展器
     * @param script
     * @param keys
     * @param args
     * @return
     */
    public Object eval(final String script,final List<String> keys,final List<String> args) {
    	
    	return new Executor<Object>(jedisSentinelPool) {

			@Override
			Object execute() {
			 return jedis.eval(script, keys, args);
			
			}
		}.getResult();
    }

上面我用一个内部类主要是用来方便统一释放资源，主要是用到

jedis.eval(script, keys, args)

参数script 是你需要传入的lua脚本，可以是脚本的路径，也可以是脚本代码，本次以传入代码为例。（传入脚本路径没试过，请自行测试。）

keys 是你需要传入的key参数的集合。

args 是你脚本需要传入的参数集合。

2.编写控制速率的脚本并且调用

	String lua="	local times = redis.call('incr',KEYS[1]) "
    			+ "		if times == 1 then "
    			+ "			redis.call('expire',KEYS[1], ARGV[1]);"
    			+ "		end"
    			+ "		if times > tonumber(ARGV[2]) then "
    			+ "			 return 0;"
    			+ "		end  "
    			+ "		     return 1;";
    	List<String> keys=new ArrayList<String>();
    	keys.add("kinglaw");
    	List<String> params=new ArrayList<String>();
    	params.add("60");
    	params.add("2");
    	System.out.println(redisCache.eval(lua, keys, params));

lua里面脚本的意思就是，对传参进来的key，进行自增长操作，如果是第一次操作，就给key设置过期时间，时间为传进来的第一个参数，如果自增数字大于传进来的第二个参数，那么就返回0，否则返回1，我这边设置的是速率是60秒内两次，方便演示效果。

演示打印效果：

这是我请求三次的效果，第三次的时候脚本返回了0，是能达到我们的限速效果的。