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

来源:互联网 发布:语文名师讲课软件 编辑:程序博客网 时间:2024/06/06 08:25

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



原创粉丝点击