redis中使用lua脚本处理业务逻辑

本文介绍: 介绍在redis中使用 lua 脚本方法，并且介绍 redisTempla t e和jedis中使用 lua 脚本

lua 脚本语言小巧轻便，运行性能优异，被很多语言作为嵌入式脚本语言使用。redis对lua 脚本语言有很好的支持，对于开发者，如果要处理的数据涉及到只是简单的计算逻辑，就不必将redis中的数据拿到应用程序中执行了，直接写一段lua 脚本程序就可以在redis 服务端完成逻辑处理。
有关lua 语言的语法内容，这里就不做过多的介绍，这里只是介绍在redis中如何运行lua 脚本。

在redis 命令行运行lua脚本非常简单，只需要定义好lua 代码的字符串，在red i s 客户端使用 eval 命令运行脚本，命令格式如下：

127.0.0.1:6379&gt; eval script numkeys key [key ...] arg [arg ...]

下面演示一下如何使用上面的命令：

127.0.0.1:6379&gt; eval "return 'hello,world!';" 0 
"hello,world!"
127.0.0.1:6379&gt; 
127.0.0.1:6379&gt; eval "return redis.call('set', KEYS[1], ARGV[1]);" 1 lua_key lua_val
OK
127.0.0.1:6379&gt; 
127.0.0.1:6379&gt; eval "return redis.call('get', KEYS[1]);" 1 lua_key
"lua_val"
127.0.0.1:6379&gt; 
127.0.0.1:6379&gt; eval "if redis.call('exists', KEYS[1]) > 0 then return redis.call('get', KEYS[1]); else return ''; end;" 1 kkkk 
""
127.0.0.1:6379> 
127.0.0.1:6379> eval "if redis.call('exists', KEYS[1]) > 0 then return redis.call('get', KEYS[1]); else return ''; end;" 1 lua_key 
"lua_val"
127.0.0.1:6379>

在lua脚本中调用 r edis命令使用 redis.call() 方法，方法的第一个参数是命令的名称，后面跟参数，所有的键放入KEYS中，值放入ARGV中，在脚本中参数下标从1开始。
除了上面的调用方式，我们也可以将lua脚本写入到系统文件中，然后通过命令行调用本地文件执行：
我们在系统新建一个lua脚本文件，命名为lua_sc ript.lua，内容如下：

local exist = redis.call('exists', KEYS[1]);
if exist > 0 then
  local rs = redis.call('incrby', KEYS[1], ARGV[1]);
  return '1_' .. rs;   // 这里是lua中的字符串拼接方法：str1 .. str2 会将两个字符串拼接成一个字符串
else 
  local rs = redis.call('set', KEYS[1], ARGV[2]);
  if rs then
    local rs = redis.call('incrby', KEYS[1], ARGV[1]);
    return '2_' .. rs;
  else
    return nil;
  end;
end;

脚本的执行流程是：先判断键是否存在，如果存在直接执行 inc rby指令进行加数操作；如果不存在，先初始化键的值，然后再执行 inc rby指令进行加数操作。

lua脚本文件执行调用命令是：

redis-cli --eval myscript.lua key1 key2 , arg1 arg2 arg3

[root@localhost src]# ./redis-cli --eval /root/lua_scripts/lua_script.lua incr:key:0001 , -2 200
"2_198"
[root@localhost src]# 
[root@localhost src]# ./redis-cli --eval /root/lua_scripts/lua_script.lua incr:key:0001 , -2 200
"1_196"
[root@localhost src]#

String lua = "return redis.call('set', KEYS[1], ARGV[1]);";
DefaultRedisScript<String> script = new DefaultRedisScript<>(lua, String.class);

List<String> keys = Collections.singletonList("java:lua:key");
String rs = redisTemplate.execute(script, keys, "java_lua_value");

DefaultRedisScript<String> script = new DefaultRedisScript<>();
script.setResultType(String.class);
script.setScriptSource(new ResourceScriptSource(new FileSystemResource("E:\lua_script.lua")));
List<String> keys = Collections.singletonList("ncr:key:0001");
String rs = redisTemplate.execute(script, keys, "-2", "200");

String script = "return redis.call('get', KEYS[1]);";
Object rs = jedis.eval(script, 1, "incr:key:0001");
System.out.println(rs);

127.0.0.1:6379> script load "return redis.call('get', KEYS[1]);"
"796941151549c416aa77522fb347487236c05e46"
127.0.0.1:6379>

Object rs = jedis.evalsha("796941151549c416aa77522fb347487236c05e46", 1, "incr:key:0001");
System.out.println(rs);

Object rs = redisTemplate.execute(new RedisCallback<Object>() {
    @Override
    public Object doInRedis(RedisConnection connection) throws DataAccessException {
        byte[] rs = connection.evalSha("796941151549c416aa77522fb347487236c05e46", ReturnType.VALUE, 1, "incr:key:0001".getBytes(StandardCharsets.UTF_8));
        return new String(rs);
    }
});
System.out.println(rs);