redistemplate 使用lua脚本
redistemplate 使用lua脚本
依赖
org.springframework.boot
spring-boot-starter-web
org.springframework.boot
spring-boot-starter-data-redis
com.alibaba
fastjson
1.2.75
注意事项
springboot使用RedisTemplate操作lua脚本的使用需要配置lettuce客户端,jedis是不支持的,现在springboot也默认是lettuce
spring:
redis:
cluster:
nodes: 192.168.101.98:2001,192.168.101.98:2002,192.168.101.98:2003,192.168.101.98:2004,192.168.101.98:2005,192.168.101.98:2006
lettuce:
pool:
min-idle: 8
max-idle: 16
max-active: 16
max-wait: 10000
案例一
如果不是用复杂的数据类型,用StringRedisTemplate就可以了,里面默认定义了序列化的方式,如果使用集群的reids,
需要保证key在同一个slot,可以用{xx}强制hash
@Autowired
private StringRedisTemplate stringRedisTemplate;
/**
* 抢红包
* 用户id 红包id
* @param userId
* @param redEnvelopeId
* @return
*/
@Override
public String grabRedEnvelope(Long userId, Long redEnvelopeId) {
DefaultRedisScript<String> redisScript = new DefaultRedisScript<>();
redisScript.setResultType(String.class);
redisScript.setScriptText(LuaScript.redLua);
List<String> keyList = new ArrayList<>();
/**
* 产生的小红包key
*/
keyList.add("{envelope}:redEnvelopeId:" + redEnvelopeId);
/**
* 红包领取记录key
*/
keyList.add("{envelope}:record:" + redEnvelopeId);
keyList.add("{envelope}" + userId);
//keyList.add(String.valueOf(userId));
/**
* -1 已经抢到红包 -2 红包已经完了 ，其余是抢到红包并返回红包余额
*/
return stringRedisTemplate.execute(redisScript, keyList,userId+"");
lua脚本
/**
* -1 已经抢到红包 -2 红包被抢光 re 红包金额 ，keys[1]、keys[2]、keys[3]分别为存储小红包的key、红包领取记录key、用户id
*/
public static String redLua = "if redis.call('hexists',KEYS[2],ARGV[1]) ~=0 then \n" +
" return '-1';\n" +
" else \n" +
"local re=redis.call('rpop',KEYS[1]);\n" +
"if re then\n" +
"redis.call('hset',KEYS[2],ARGV[1],1);\n" +
"return re;\n" +
"else\n" +
"return '-2';\n" +
"end\n" +
"end";
案例二
如果value是复杂数据类型,这个时候stringRedisTemplate就不了,需要自己定制RedisTemplate
@Configuration
public class MyRedisConf {
@Bean(name = "myRedisTemplate")
public RedisTemplate<String, Object> redisTemplate(RedisConnectionFactory redisConnectionFactory) {
RedisTemplate<String, Object> template = new RedisTemplate<>();
template.setConnectionFactory(redisConnectionFactory);
FastJsonRedisSerializer serializer = new FastJsonRedisSerializer(Object.class);
// value值的序列化采用fastJsonRedisSerializer
template.setValueSerializer(serializer);
template.setHashValueSerializer(serializer);
// key的序列化采用StringRedisSerializer
template.setKeySerializer(new StringRedisSerializer());
template.setHashKeySerializer(new StringRedisSerializer());
template.setConnectionFactory(redisConnectionFactory);
template.afterPropertiesSet();
return template;
}
}
这里还是要指定返回参数的序列化类型,不然会报错
@Resource(name = "myRedisTemplate")
private RedisTemplate<String,Object> redisTemplate;
@RequestMapping("/test02")
public void test02(){
String script = "local receive_arg_json = cjson.decode(ARGV[1]);\n" +
"local v1 = receive_arg_json.v1;\n" +
"local v2 = receive_arg_json.v2;\n" +
"\n" +
"redis.call('set','wxy',KEYS[1]);" +
"return 'success';";
DefaultRedisScript<String> defaultRedisScript = new DefaultRedisScript<>();
defaultRedisScript.setResultType(String.class);
defaultRedisScript.setScriptText(script);
List<String> keyList = new ArrayList<>();
keyList.add("{wxy}:a");
keyList.add("{wxy}:b");
Map<String,String> argvMap = new HashMap<>();
argvMap.put("v1","v1");
argvMap.put("v2","v2");
String execute = redisTemplate.execute(defaultRedisScript,redisTemplate.getValueSerializer(), RedisSerializer.string(), keyList, argvMap);
System.out.println("--------------------------"+execute);
}
案例三
@RequestMapping("/test03")
public void test03() {
String script = "--返回的变量\n" +
"local result = {}\n" +
"--获取KEY\n" +
"--开抢时间\n" +
"local key1 = KEYS[1]\n" +
"--过期时间\n" +
"local key2 = KEYS[2]\n" +
"--红包金额分配key\n" +
"local key3 = KEYS[3]\n" +
"--红包领取记录key\n" +
"local key4 = KEYS[4]\n" +
"--红包实体信息\n" +
"--local key5 = KEYS[5]\n" +
"\n" +
"--获取value ,占时是uid\n" +
"--local value1 = ARGV[1]\n" +
"--1.校验是否过期\n" +
"if( redis.call('ttl',key2) == -2)\n" +
"then\n" +
"result[1] = '1'\n" +
"result[2] = cjson.encode('红包已过期')\n" +
"return result;\n" +
"end\n" +
"\n" +
"--2.校验是否触发\n" +
"local trigger_time = redis.call('ttl',key1)\n" +
"if(trigger_time ~= -2)\n" +
"then\n" +
"result[1] = '2'\n" +
"result[2] = trigger_time\n" +
"return result;\n" +
"end\n" +
"\n" +
"--3.校验是否已经抢过\n" +
"if (redis.call('hexists',key4,ARGV[1]) ~=0) \n" +
"then\n" +
"result[1] = '3'\n" +
"result[2] = cjson.encode('已抢到红包')\n" +
"return result;\n" +
"else\n" +
"local re = redis.call('rpop',key3)\n" +
"if re\n" +
"then\n" +
"redis.call('hset',key4,ARGV[1],1)\n" +
"result[1] = '4'\n" +
"result[2] = re\n" +
"return result;\n" +
"else\n" +
"result[1] = '5'\n" +
"result[2] = cjson.encode('来慢了')\n" +
"return result;\n" +
"end\n" +
"end\n" +
"\n" +
" \n" +
"\n" +
"\n" +
"\n" +
"\n" +
"\n" +
"\n" +
"\n";
DefaultRedisScript<List> defaultRedisScript = new DefaultRedisScript<>();
defaultRedisScript.setResultType(List.class);
defaultRedisScript.setScriptText(script);
List<String> keyList = new ArrayList<>();
keyList.add(RED_ENVELOPE_LOCK_KEY);
keyList.add(RED_ENVELOPE_TTL_KEY);
keyList.add(RED_ENVELOPE_AMOUNT_KEY);
keyList.add(RED_ENVELOPE_RECORD);
List execute = redisTemplate.execute(defaultRedisScript, keyList, 123);
System.out.println("--------------------------" + execute);
}
参考
https://blog.csdn.net/clfvita/article/details/73955953
https://blog.csdn.net/fsw4848438/article/details/81540495?utm_medium=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromBaidu-1.control&depth_1-utm_source=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromBaidu-1.control