7.4-手把手教你彻底掌握分布式锁lua脚本+redis原生代码

7.4-手把手教你彻底掌握分布式锁lua脚本+redis原生代码编写

手把手教你彻底掌握分布式锁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学习笔记》