redis中使用lua语言

港控/mmm° 2022-05-25 04:43 300阅读 0赞

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当中设置数据

![70][]

编写一段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

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

执行结果如下图：

![70 1][]

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 );
    
    
        }
    
    }

[70]: /images/20220525/de492b94d6954e63b6033d369e8d2e9e.png
[70 1]: /images/20220525/ef4dd76944704e439ffad0aeaa3dad1b.png