在springboot中使用redis的stringRedisTemplate想实现对某个键值+n并且指定过期的功能,为了实现原子性,采用了lua脚本的方式。但是发现,键值对都需要string类型才可以不报错。
public Long increment(String key, long value) {
String luaScript = "local value = redis.call('incr', KEYS[1], ARGV[1])\n" +
"redis.call('expire', KEYS[1], ARGV[2])\n" +
"return tostring(value)";
// 创建 RedisScript 对象,指定脚本内容和返回值类型
DefaultRedisScript<String> script = new DefaultRedisScript<>(luaScript, String.class);
// 调用 StringRedisTemplate 的执行方法,传入脚本、键名、参数列表.后面两个参数需要string
String res = stringRedisTemplate.execute(script, Collections.singletonList(key), String.valueOf(value), String.valueOf(DEFAULT_MINUTE * 60));
return Long.valueOf(res);
}
debug进去到代码会执行keysAndArgs(argsSerializer, keys, args)序列化,对于stringRedisTemplate中,默认的序列化类是StringRedisSerializer
@SuppressWarnings({ "unchecked", "rawtypes" })
protected byte[][] keysAndArgs(RedisSerializer argsSerializer, List<K> keys, Object[] args) {
final int keySize = keys != null ? keys.size() : 0;
byte[][] keysAndArgs = new byte[args.length + keySize][];
int i = 0;
if (keys != null) {
for (K key : keys) {
if (keySerializer() == null && key instanceof byte[]) {
keysAndArgs[i++] = (byte[]) key;
} else {
keysAndArgs[i++] = keySerializer().serialize(key);
}
}
}
for (Object arg : args) {
if (argsSerializer == null && arg instanceof byte[]) {
keysAndArgs[i++] = (byte[]) arg;
} else {
// 这里执行序列化
keysAndArgs[i++] = argsSerializer.serialize(arg);
}
}
return keysAndArgs;
}
查看StringRedisSerializer的serialize方法发现,只支持string的键值对。
@Override
public byte[] serialize(@Nullable String string) {
return (string == null ? null : string.getBytes(charset));
}
得出结果,在stringRedisTemplate执行lua脚本时候,需要对KEY和ARGV参数进行string的转换才可以。