Redis(十二) lua脚本

// 同时设置俩个String键值对，并且key的值通过附带参数传入(不需要返回值)
127.0.0.1:6379>EVAL "redis.call('set',KEYS[1],ARGV[1]);redis.call('set',KEYS[2],ARGV[2])" 2 "name1" "name2" "疾风剑豪" "快乐风男"
(nil)

// 设置一个String键值对，并对另一个键值对递增，然后返回递增后的值
127.0.0.1:6379>EVAL "redis.call('set',KEYS[1],ARGV[1]);return redis.call('incr',KEYS[2])" 2 "name1" "clickCount" "疾风剑豪"
(integer) 6

// 同时修改某个hash结构的俩个属性(name、age)
127.0.0.1:6379>EVAL "redis.call('hset',KEYS[1], ARGV[1], ARGV[2]);redis.call('hset',KEYS[1], ARGV[3], ARGV[4])" 1 "user" "name" "张三" "age" "18"
(nil)

// 对某个值为数字的字符串，值+3
127.0.0.1:6379>EVAL "redis.call('set',KEYS[1], redis.call('get', KEYS[1]) + ARGV[1])" 1 "quantity" "3"
(nil)

// 对某个值为数字的字符串，值+3，并返回增加后的值
127.0.0.1:6379>EVAL "redis.call('set',KEYS[1], redis.call('get', KEYS[1]) + ARGV[1]);return redis.call('get', KEYS[1])" 1 "quantity" "3"
"7"

// 设置一个字符串，并对key2的值递增，如果key2的值并非数字，执行报错，但第一条命令仍然执行成功
127.0.0.1:6379>EVAL "redis.call('set',KEYS[1],ARGV[1]);redis.call('incr',KEYS[2])" 2 "key1" "key2" "二桃杀三士"
(nil)

// 上述命令后面再加一个正常命令，执行仍然报错，但是第一条命令生效，第三条不生效，可以判定遇到异常命令直接终止脚本执行
127.0.0.1:6379>EVAL "redis.call('set',KEYS[1],ARGV[1]);redis.call('incr',KEYS[2]);redis.call('set',KEYS[3],ARGV[1])" 3 "key1" "key2" "key3" "二桃杀三士"
(nil)

-- 返回结果集合
local list = {};

-- 元素数量参数转化为lua数据类型
local num = tonumber(ARGV[1]);

-- 循环
while (num > 0)
do
	-- 弹出第一个元素(如果为空返回nil)
	local item = redis.call('LPOP', KEYS[1]);

	-- 如果为空，终止循环(虽然命令返回nil，但逻辑判断要用布尔)
	if item == false then
		break;
	end;

	-- 将元素添加到返回结果集
	table.insert(list, item);

	-- 元素数量递减1
	num = num -1;
end;

return list;

-- ip地址对应value递增，并返回给局部变量cnt
local cnt = redis.call('INCR', KEYS[1])

-- 如果递增后的值大于访问限制数,返回1
if cnt > tonumber(ARGV[1])
	then
		return 1
	end

-- 如果递增后的值等于1，说明key之前不存在。要么之前没有访问，要么正好到下一个时间窗，设置key的过期时间(参数n秒)
if cnt == 1
	then
		redis.call('PEXPIRE', KEYS[1], ARGV[2])
	end
return 0;

// lua脚本
String lua = "redis.call('set',KEYS[1],ARGV[1]);redis.call('set',KEYS[2],ARGV[2])";

// keys数组
List<String> keys = Arrays.asList("key1", "key2");

// args数组
List<String> args = Arrays.asList("aaaaa", "bbbbb");

// 通过connection执行EVAL命令
Object evalResult = redisTemplate.execute(new RedisCallback<Object>() {
	@Override
	public Object doInRedis(RedisConnection redisConnection) throws DataAccessException {

		// 获取connection
		Object nativeConnection = redisConnection.getNativeConnection();

		// 单机模式(jedis)
		if (nativeConnection instanceof Jedis) {
			return (Object) ((Jedis) nativeConnection).eval(lua, keys, args);
		}

		// 集群模式(jedis)
		if (nativeConnection instanceof JedisCluster) {
			return (Object) ((JedisCluster) nativeConnection).eval(lua, keys, args);
		}

		return null;
	}
});

// lua脚本
String lua = "redis.call('set',KEYS[1],ARGV[1]);redis.call('set',KEYS[2],ARGV[2])";

// keys、args字节数组的数组(前2个字节数组为keys，后俩个为args)
byte[][] keysAndArgsBytesArray = new byte[][]{"key1".getBytes(), "key2".getBytes(), "aaaaa".getBytes(), "bbbbb".getBytes()};

// 通过connection执行EVAL命令
Object evalResult = redisTemplate.execute(new RedisCallback<Object>() {
	@Override
	public Object doInRedis(RedisConnection redisConnection) throws DataAccessException {

		// 获取connection
		Object nativeConnection = redisConnection.getNativeConnection();

		// 单机模式(lettuce)
		if (nativeConnection instanceof LettuceConnection) {
			RedisScriptingCommands commands = ((LettuceConnection) nativeConnection).scriptingCommands();
			return (Object) commands.eval(lua.getBytes(), ReturnType.BOOLEAN, 2, keysAndArgsBytesArray);
		}

		// 集群模式(lettuce)
		if (nativeConnection instanceof LettuceClusterConnection) {
			RedisScriptingCommands commands = ((LettuceConnection) nativeConnection).scriptingCommands();
			return (Object) commands.eval(lua.getBytes(), ReturnType.BOOLEAN, 2, keysAndArgsBytesArray);
		}

		return null;
	}
});

// lua脚本
String lua = "redis.call('set',KEYS[1],ARGV[1]);redis.call('set',KEYS[2],ARGV[2])";

// 执行EVAL命令，READ_ONLY优先发送到从节点，READ_WRITE发送到主节点
Object evalResult = redissonClient.getScript().eval(RScript.Mode.READ_ONLY, lua, RScript.ReturnType.INTEGER, keys, args);

// 将Lua脚本内容上传到Redis服务器
String sha1 = redissonClient.getScript().scriptLoad(lua);

// 执行sha1脚本
Object evalShaResult = redissonClient.getScript().evalSha(RScript.Mode.READ_ONLY, sha1, RScript.ReturnType.INTEGER, keys, args);

// 判断一或多个sha1是否存在
String sha1Test = "asdasdasdasdasd";
List<Boolean> exist = redissonClient.getScript().scriptExists(sha1, sha1Test);