手把手教你彻底掌握分布式锁lua脚本+redis原生代码编写
简介：手把手教你彻底掌握分布式锁+原生代码编写
前面说了redis做分布式锁存在的问题
核心是保证多个指令原子性，加锁使用setnx setex 可以保证原子性，那解锁使用 判断和删除怎么保证原子性
多个命令的原子性：采用 lua脚本+redis, 由于【判断和删除】是lua脚本执行，所以要么全成功，要么全失败
//获取lock的值和传递的值一样，调用删除操作返回1，否则返回0
String script = "if redis.call('get',KEYS[1]) == ARGV[1] then return redis.call('del',KEYS[1]) else return 0 end";
//Arrays.asList(lockKey)是key列表，uuid是参数
Integer result = redisTemplate.execute(new DefaultRedisScript<>(script, Integer.class), Arrays.asList(lockKey), uuid);
全部代码
/**
* 原生分布式锁 开始
* 1、原子加锁 设置过期时间，防止宕机死锁
* 2、原子解锁：需要判断是不是自己的锁
*/
@RestController
@RequestMapping("/api/v1/coupon")
public class CouponController {
@Autowired
private StringRedisTemplate redisTemplate;
@GetMapping("add")
public JsonData saveCoupon(@RequestParam(value = "coupon_id",required = true) int couponId){
//防止其他线程误删
String uuid = UUID.randomUUID().toString();
String lockKey = "lock:coupon:"+couponId;
lock(couponId,uuid,lockKey);
return JsonData.buildSuccess();
}
private void lock(int couponId,String uuid,String lockKey){
//lua脚本
String script = "if redis.call('get',KEYS[1]) == ARGV[1] then return redis.call('del',KEYS[1]) else return 0 end";
Boolean nativeLock = redisTemplate.opsForValue().setIfAbsent(lockKey,uuid,Duration.ofSeconds(30));
System.out.println(uuid+"加锁状态:"+nativeLock);
if(nativeLock){
//加锁成功
try{
//TODO 做相关业务逻辑
TimeUnit.SECONDS.sleep(10L);
} catch (InterruptedException e) {
} finally {
//解锁
Long result = redisTemplate.execute( new DefaultRedisScript<>(script,Long.class),Arrays.asList(lockKey),uuid);
System.out.println("解锁状态:"+result);
}
}else {
//自旋操作
try {
System.out.println("加锁失败，睡眠5秒 进行自旋");
TimeUnit.MILLISECONDS.sleep(5000);
} catch (InterruptedException e) { }
//睡眠一会再尝试获取锁
lock(couponId,uuid,lockKey);
}
}
}
遗留一个问题，锁的过期时间，如何实现锁的自动续期 或者 避免业务执行时间过长，锁过期了？
原生方式的话，一般把锁的过期时间设置久一点，比如10分钟时间
原生代码+redis实现分布式锁使用比较复杂，且有些锁续期问题更难处理
使用特别简单: 看看公司做的工业级项目代码
《小滴课堂-Redis6学习笔记》
