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,是能达到我们的限速效果的。