Redis进阶-lua脚本

2021-08-17 11:25:14 浏览数 (1)


Pre

Redis在2.6推出了脚本功能,允许开发者使用Lua语言编写脚本传到Redis中执行。


语法

从Redis2.6.0版本开始,通过内置的Lua解释器,可以使用EVAL命令对Lua脚本进行求值。

EVAL命令的格式如下:

代码语言:javascript复制
EVAL script numkeys key [key ...] arg [arg ...]
  • script参数是一段Lua脚本程序,它会被运行在Redis服务器上下文中,这段脚本不必(也不应该)定义为一个Lua函数。
  • numkeys参数用于指定键名参数的个数
  • 键名参数 key [key …] 从EVAL的第三个参数开始算起,表示在脚本中所用到的那些Redis键(key),这些键名参数可以在 Lua中通过全局变量KEYS数组,用1为基址的形式访问( KEYS[1] , KEYS[2] ,以此类推)
  • 在命令的最后,那些不是键名参数的附加参数 arg [arg …] ,可以在Lua中通过全局变量ARGV数组访问访问的形式和KEYS变量类似( ARGV[1] 、 ARGV[2] ,诸如此类)

举例

代码语言:javascript复制
127.0.0.1:6379> EVAL "return {KEYS[1],KEYS[2],ARGV[1],ARGV[2]}" 2 key1 key2 value1 value2
1) "key1"
2) "key2"
3) "value1"
4) "value2"
127.0.0.1:6379> 

必须是大写哈,小写报错 。

代码语言:javascript复制
127.0.0.1:6379> EVAL "return {keys[1],keys[2],argv[1],argv[2]}" 2 key1 key2 value1 value2
(error) ERR Error running script (call to f_8cfd66635a33779a52579a216e55263704ce398a): @enable_strict_lua:15: user_script:1: Script attempted to access nonexistent global variable 'keys' 
127.0.0.1:6379> 

其中 “return {KEYS[1],KEYS[2],ARGV[1],ARGV[2]}” 是被求值的Lua脚本,数字2指定了键名参数的数量, key1和key2是键名参数,分别使用 KEYS[1] 和 KEYS[2] 访问,而最后的 value1和 value2则是附加参数,可以通过 ARGV[1] 和 ARGV[2] 访问它们。


jedis操作lua

在 Lua 脚本中,可以使用redis.call()函数来执行Redis命令

代码语言:javascript复制
public class JedisSingleTest {
    public static void main(String[] args) throws IOException {

        JedisPoolConfig jedisPoolConfig = new JedisPoolConfig();
        jedisPoolConfig.setMaxTotal(20);
        jedisPoolConfig.setMaxIdle(10);
        jedisPoolConfig.setMinIdle(5);

        // timeout,这里既是连接超时又是读写超时,从Jedis 2.8开始有区分connectionTimeout和soTimeout的构造函数
        JedisPool jedisPool = new JedisPool(jedisPoolConfig, "192.168.18.130", 6379, 3000, null);

        Jedis jedis = null;
        try {
            //从redis连接池里拿出一个连接执行命令
            jedis = jedisPool.getResource();

            //******* lua脚本示例 ********
            //模拟一个商品减库存的原子操作
            //lua脚本命令执行方式:redis-cli --eval /tmp/test.lua , 10
            jedis.set("product_stock_10016", "15");  //初始化商品10016的库存

            // lua脚本
            String script = " local count = redis.call('get', KEYS[1]) "  
                            " local a = tonumber(count) "  
                            " local b = tonumber(ARGV[1]) "  
                            " if a >= b then "  
                            "   redis.call('set', KEYS[1], count-b) "  
                            //模拟语法报错回滚操作"   bb == 0 "  
                            "   return 1 "  
                            " end "  
                            " return 0 ";
            // eval调用, 第一个参数为 KEYS[1]) , 第二个参数为ARGV[1]
            Object obj = jedis.eval(script, Arrays.asList("product_stock_10016"), Arrays.asList("10"));
            System.out.println(obj);

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            //注意这里不是关闭连接,在JedisPool模式下,Jedis会被归还给资源池。
            if (jedis != null)
                jedis.close();
        }
    }
}

好处

  • 减少网络开销 假设你有3条命令要执行,本来3次的网络开销,使用lua可以用一个请求完成,原先3次请求的逻辑放在redis服务器上完成。使用脚本,减少了网络往返时延。这点跟管道类似。
  • 原子操作 redis会将整个脚本作为一个整体执行,中间不会被其他命令插入。管道不是原子的,不过redis的批量操作命令(类似mset)是原子的。
  • 替代redis的事务功能 redis自带的事务功能很鸡肋,报错不支持回滚,而redis的lua脚本几乎实现了常规的事务功能,支持报错回滚操作,官方推荐如果要使用redis的事务功能可以用redis lua替代。

lua实战

高并发-【抢红包案例】之四:使用Redis Lua脚本实现抢红包并异步持久化到数据库


注意事项

注意,不要在Lua脚本中出现死循环和耗时的运算,否则redis会阻塞,将不接受其他的命令, 所以使用时要注意不能出现死循环、耗时的运算。

redis是单进程、单线程执行脚本。管道不会阻塞redis。

0 人点赞