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

本文介绍了如何利用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,是能达到我们的限速效果的。



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值