Redis使用lua脚本 php [转]

版本:自2.6.0起可用。 时间复杂度:取决于执行的脚本。 使用Lua脚本的好处:

    减少网络开销。可以将多个请求通过脚本的形式一次发送,减少网络时延。 原子操作。redis会将整个脚本作为一个整体执行,中间不会被其他命令插入。因此在编写脚本的过程中无需担心会出现竞态条件,无需使用事务。 复用。客户端发送的脚步会永久存在redis中,这样,其他客户端可以复用这一脚本而不需要使用代码完成相同的逻辑。

如何使用

基本使用

命令格式: EVAL script numkeys key [key ...] arg [arg ...] 说明:

    script是第一个参数,为Lua 5.1脚本。该脚本不需要定义Lua函数(也不应该)。 第二个参数numkeys指定后续参数有几个key。 key [key ...],是要操作的键,可以指定多个,在lua脚本中通过KEYS[1], KEYS[2]获取 arg [arg ...],参数,在lua脚本中通过ARGV[1], ARGV[2]获取。

简单实例:

[plain]

  1. 127.0.0.1:6379> eval "return ARGV[1]" 0 100
  2. "100"
  3. 127.0.0.1:6379> eval "return {ARGV[1],ARGV[2]}" 0 100 101
  4. 1) "100"
  5. 2) "101"
  6. 127.0.0.1:6379> eval "return {KEYS[1],KEYS[2],ARGV[1],ARGV[2]}" 2 key1 key2 first second
  7. 1) "key1"
  8. 2) "key2"
  9. 3) "first"
  10. 4) "second"
  11. 127.0.0.1:6379> eval "redis.call(SET, KEYS[1], ARGV[1]);redis.call(EXPIRE, KEYS[1], ARGV[2]); return 1;" 1 test 10 60
  12. (integer) 1
  13. 127.0.0.1:6379> ttl test
  14. (integer) 59
  15. 127.0.0.1:6379> get test
  16. "10"

注:

    {}在lua里是指数据类型table,类似数组。 redis.call()可以调用redis命令。

命令行里使用

如果直接使用redis-cli命令,格式会有点不一样: redis-cli --eval lua_file key1 key2 , arg1 arg2 arg3 注意的地方:

    eval 后面参数是lua脚本文件,.lua后缀 不用写numkeys,而是使用,隔开。注意,前后有空格。

示例: incrbymul.lua

[plain]

  1. local num = redis.call(GET, KEYS[1]);
  2. if not num then
  3. return 0;
  4. else
  5. local res = num * ARGV[1];
  6. redis.call(SET,KEYS[1], res);
  7. return res;
  8. end

命令行运行:

[plain]

  1. $ redis-cli --eval incrbymul.lua lua:incrbymul , 8
  2. (integer) 0
  3. $ redis-cli incr lua:incrbymul
  4. (integer) 1
  5. $ redis-cli --eval incrbymul.lua lua:incrbymul , 8
  6. (integer) 8
  7. $ redis-cli --eval incrbymul.lua lua:incrbymul , 8
  8. (integer) 64
  9. $ redis-cli --eval incrbymul.lua lua:incrbymul , 2
  10. (integer) 128

由于redis没有提供命令可以实现将一个数原子性的乘以N倍,这里我们就用Lua脚本实现了,运行过程中确保不会被其它客户端打断。

phpredis里使用

接着上面的例子: incrbymul.php

[plain]

  1. <?php
  2. $lua = <<<EOF
  3. local num = redis.call(GET, KEYS[1]);
  4. if not num then
  5. return 0;
  6. else
  7. local res = num * ARGV[1];
  8. redis.call(SET,KEYS[1], res);
  9. return res;
  10. end
  11. EOF;
  12. $redis = new Redis();
  13. $redis->connect(127.0.0.1, 6379);
  14. $ret = $redis->eval($lua, array("lua:incrbymul", 2), 1);
  15. echo $ret;

运行:

[plain]

  1. $ redis-cli set lua:incrbymul 0
  2. OK
  3. $ redis-cli incr lua:incrbymul
  4. (integer) 1
  5. $ php incrbymul.php
  6. 2
  7. $ php incrbymul.php
  8. 4

eval原型: Redis::eval(string script, [array keys, long num_keys]) eval函数的第3个参数为KEYS个数,phpredis依据此值将KEYS和ARGV做区分。

经验分享 程序员 微信小程序 职场和发展