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

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