Redis分布式锁的实现

清疚 2021-11-04 15:04 326阅读 0赞

核心：

**采用redis: setnx()命令来实现加锁，Lua脚本实现redis解锁（保证解锁的原子性）**

加锁：

SET key value NX PX 3000

注意点：1、必须给锁设置一个失效时间 （避免死锁）

2、加锁时，每个节点产生一个随机字符串（避免误删锁）

3、写入随机值与设置失效时间必须是同时的（保证加锁的原子性）

public class RedisLock implements Lock{
    	
    
        private final String KEY = "key";
    
        private ThreadLocal<String> local = new ThreadLocal<String>();
    
        //阻塞锁
    	@Override
    	public void lock() {
            if(tryLock()){
                return;
            }
    		try{
                Thread.sleep(50);
            }catch(Exception e){
    
            }
            //递归
    		lock();
    	}
    
        /**
        **非阻塞锁 如果没有获取到锁直接返回
        **/
    	@Override
    	public boolean tryLock() {
            //生成随机的value值
    		String uuid = UUID.randomUUID().toString();
            //获取jedis连接操作redis
            Jedis jedis = //;
            String res = jedis.set(KEY,uuid,"NX","PX",3000);
            if("OK".equals(res)){
                //加锁成功 设置uuid
                local.set(uuid);
                return true;
            }
    		return false;
    	}
    	
    	
        //解锁
    	@Override
    	public void unlock() {
    		String script = FileUtils.readFile("lua脚本文件路径.lua");
            Jedis jedis = //jedis连接
            //执行lua脚本 传入key和value参数 如果value相同则删除
            jedis.eval(script,Array.asList(KEY),Array.asList(local.get()));
    		
    	}
    
    	@Override
    	public void lockInterruptibly() throws InterruptedException {
    		// TODO Auto-generated method stub
    		
    	}
    
    	
    	@Override
    	public boolean tryLock(long time, TimeUnit unit) throws InterruptedException {
    		// TODO Auto-generated method stub
    		return false;
    	}
    
    	
    
    	@Override
    	public Condition newCondition() {
    		// TODO Auto-generated method stub
    		return null;
    	}
    
    }

解锁（lua脚本）：

匹配随机值，删除redis上的key数据，要保证获取数据、判断一致和删除数据三个操作是原子的。

if redis.call("get",KEYS[1]) == ARGV[1] then
        return redis.call("del",KEYS[1])
    else
        return 0
    end

其中的ARGV是从ThreadLocal中获取加锁时设置的随机数

代码使用：

public class LockTest{
    
        @Resource(name="redisLock")
        private Lock lock;
    
        private int count = 200;
    
        //并发操作
        public void test(){
        
            Thread thread1 = new Thread(new Test());
            Thread thread2 = new Thread(new Test());
            Thread thread3 = new Thread(new Test());
            thread1.start();
            thread2.start();
            thread3.start();
        
        }
    
        public class Test implements Runnable{
        
            @Override
            public void run(){
             try{
                //加锁
                lock.lock();
                count --;
             }catch(Exception e){
            
    
             }finally{
                    //解锁
                    lock.unlock();
             }
    
            }
    
    
        }
    
    
    
    
    }