redis中使用lua语言

lua的安装：

curl -R -O http://www.lua.org/ftp/lua-5.3.0.tar.gz
tar zxf lua-5.3.0.tar.gz
cd lua-5.3.0
make linux test
make install

在redis当中使用 lua解释器执行脚本

    eval "return redis.call('set',KEYS[1],ARGV[1])" 1 hello world

执行后可以在redis当中设置数据

编写一段lua脚本如下,该脚本的目的是用于对某个ip的频率进行限制，限制其1分钟访问1次

----限流，对某个ip的频率进行限制，1分钟访问10次
----传入参数  KEYS[1] 要操作的目标键   ARGS[1] 访问超时时间    ARGS[2]  访问次数
local num = redis.call('incr',KEYS[1])
----如果是第一次传递过来的值，则设置一个有效期时间是访问超时时间
if tonumber(num)==1 then
        redis.call('expire',KEYS[1],ARGS[1])
        return 1
----如果 当前的计数大于传递过来的次数
elseif tonumber(num)>tonumber(ARGV[2]) then
        return 0
else
        return 1
end

然后再在redis中执行下面的命令，执行超过10次，最后输出结果0

./redis-cli --eval "../lua/limit_ip.lua" ip:limit:192.168.159.136 , 6000 10    

执行这条命令的时候一定要注意参数与参数之间一定要带上空格，否则会执行失败

执行结果如下图：

redis在这里调用了10次lua脚本之后返回了状态0，这个状态可以用于是否进行用户是否频繁访问的判断

注意一点 lua脚本是一个原子性的操作，只有当lua脚本执行完毕之后，redis的其他操作才能正常执行。

接下来在 java中调用  lua脚本来操作 redis，在项目中导入下面的 pom文件

<dependency>
  <groupId>redis.clients</groupId>
  <artifactId>jedis</artifactId>
  <version>2.9.0</version>
</dependency>

编写一个 LuaDemo的java文件.

package com.gupao.vip.mic.dubbo.redis;

import redis.clients.jedis.Jedis;

import java.util.ArrayList;
import java.util.List;

public class LuaDemo {

    public static void main(String[]args) throws Exception {
        //RedisManager中定义了连接redis的方法
        Jedis jedis = RedisManager.getJedis();
        //这里使用字符串用来记录lua的脚本代码
        String lua =
                "local num = redis.call('incr',KEYS[1])\n" +
                "if tonumber(num)==1 then\n" +
                "        redis.call('expire',KEYS[1],ARGS[1])\n" +
                "        return 1\n" +
                "elseif tonumber(num)>tonumber(ARGV[2]) then\n" +
                "        return 0\n" +
                "else\n" +
                "        return 1\n" +
                "end\n";

        String shaLoad = jedis.scriptLoad(  lua );
        System.out.println(shaLoad);

        List<String> keys = new ArrayList<String>();
        keys.add("ip:limit:127.0.0.1");

        List<String> arggs = new ArrayList<String>();
        arggs.add("6000");
        arggs.add("10");

        //
        Object obj = jedis.evalsha( jedis.scriptLoad(  lua ),keys, arggs );
        System.out.println( obj );


    }

}