分布式锁之Redis6+Lua脚本实现原生分布式锁

AloneDrifters

已于 2022-03-21 18:25:16 修改

阅读量323

点赞数

分类专栏：分布式、微服务、中间件文章标签： Redis 分布式锁

于 2022-03-21 18:05:03 首次发布

本文链接：https://blog.csdn.net/qq_53609683/article/details/123641556

版权

分布式、微服务、中间件专栏收录该内容

31 篇文章 3 订阅

订阅专栏

本文详细介绍了分布式锁的概念、设计要点和实现方式，以Redis为例，探讨了加锁、解锁的原子性操作以及可能出现的问题。通过使用lua脚本解决了命令非原子性的问题，同时提出了锁自动续期的挑战，并提供了官方推荐的解决方案。

摘要由CSDN通过智能技术生成

文章目录

简介

分布式锁核⼼知识介绍和注意事项

背景

就是保证同⼀时间只有⼀个客户端可以对共享资源进⾏操作

案例

优惠券领劵限制张数、商品库存超卖

核⼼

为了防⽌分布式系统中的多个进程之间相互⼲扰，我们需要⼀种分布式协调技术来对这些进程进⾏调度利⽤互斥机制来控制共享资源的访问，这就是分布式锁要解决的问题避免共享资源并发操作导致数据问题

加锁

本地锁：synchronize、lock等，锁在当前进程内，集群部署下依旧存在问题
分布式锁：redis、zookeeper等实现，虽然还是锁，但是多个进程共⽤的锁标记，可以⽤Redis、Zookeeper、Mysql等都可以

设计分布式锁应该考虑的东⻄

1.排他性 --在分布式应⽤集群中，同⼀个⽅法在同⼀时间只能被⼀台机器上的⼀个线程执⾏
2.容错性 --分布式锁⼀定能得到释放，⽐如客户端奔溃或者⽹络中断
3.满⾜可重⼊、⾼性能、⾼可⽤
4.注意分布式锁的开销、锁粒度

基于Redis实现分布式锁

实现分布式锁可以⽤ Redis、Zookeeper、Mysql数据库这⼏种 , 性能最好的是Redis

分布式锁离不开 key - value 设置
key 是锁的唯⼀标识，⼀般按业务来决定命名，⽐如想要给⼀种优惠券活动加锁，key 命名为 “coupon:id” 。value就可以使⽤固定值，⽐如设置成1

加锁 SETNX key value
解锁 del (key)
配置锁超时 expire (key，30s）

综合伪代码

1.setnx 的含义就是 SET if Not Exists，有两个参数setnx(key, value)，该⽅法是原⼦性操作
2.如果 key 不存在，则设置当前 key 成功，返回 1
3.如果当前 key 已经存在，则设置当前 key 失败，返回 0
4.得到锁的线程执⾏完任务，需要释放锁，以便其他线程可以进⼊,调⽤ del(key)
5.客户端奔溃或者⽹络中断，资源将会永远被锁住,即死锁，因此需要给key配置过期时间，以保证即使没有被显式释放，这把锁也要在⼀定时间后⾃动释放

methodA(){
 String key = "coupon_66";
 if（setnx（key，1） == 1）{
	  expire(key,30,TimeUnit.MILLISECONDS)
	  try {
		 //做对应的业务逻辑
		 //查询⽤户是否已经领券
		 //如果没有则扣减库存
		 //新增领劵记录
	  } finally {
		 del（key）
	  }
 }else{
		 //睡眠100毫秒，然后⾃旋调⽤本⽅法
		 methodA()
	 }
}

存在什么问题？

多个命令之间不是原⼦性操作，如setnx和expire之间，如果setnx成功，但是expire失败，且宕机了，则这个资源就是死锁业务超时，存在其他线程勿删，key 30秒过期，假如线程A执⾏很慢超过30秒，则key就被释放了，其他线程B就得到了锁，这个时候线程A执⾏完成，⽽B还没执⾏完成，结果就是线程A删除了线程B加的锁

问题解决

使⽤原⼦命令：设置和配置过期时间 setnx / setex
如: set key 1 ex 30 nx
java⾥⾯redisTemplate.opsForValue().setIfAbsent(“seckill_1”,“success”,30,TimeUnit.MILLISECONDS)
可以在 del 释放锁之前做⼀个判断，验证当前的锁是不是⾃⼰加的锁, 那 value 应该是存当前线程的标识或者uuid
String key = "coupon_66"String value = Thread.currentThread().getId()
进⼀步细化误删当线程A获取到正常值时，返回带代码中判断期间锁过期了，线程B刚好重新设置了新值，线程A那边有判断value是⾃⼰的标识，然后调⽤del⽅法，结果就是删除了新设置的线程B的值核⼼还是判断和删除命令不是原⼦性操作导致

总结

加锁+配置过期时间：保证原⼦性操作
解锁: 防⽌误删除、也要保证原⼦性操作

解决解锁的原子性

前⾯说了redis做分布式锁存在的问题核⼼是保证多个指令原⼦性，加锁使⽤setnx setex 可以保证原⼦性，那解锁使⽤判断和删除怎么保证原⼦性

多个命令的原⼦性：采⽤ lua脚本+redis, 由于【判断和删除】是lua脚本执⾏，所以要么全成功，要么全失败

//获取lock的值和传递的值⼀样，调⽤删除操作返回1，否则返回0
String script = "if redis.call('get',KEYS[1])== ARGV[1] then returnredis.call('del',KEYS[1]) else return 0 end";

//Arrays.asList(lockKey)是key列表，uuid是参数
Integer result = redisTemplate.execute(newDefaultRedisScript<>(script, Integer.class),Arrays.asList(lockKey), uuid);

代码实现

import net.xdclass.xdclassredis.util.JsonData;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import java.time.Duration;
import java.util.Arrays;
import java.util.Random;
import java.util.UUID;
import java.util.concurrent.TimeUnit;

@RestController
@RequestMapping("/api/v1/coupon")
public class CouponController {

    @Autowired
    private RedisTemplate<String,Object> 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);
        }
    }


}