Redis分布式锁Java代码实现

最新推荐文章于 2024-01-13 16:07:43 发布

Als1521

最新推荐文章于 2024-01-13 16:07:43 发布

阅读量124

点赞数

分类专栏： redis 文章标签： java redis 数据库

本文链接：https://blog.csdn.net/Als1521/article/details/130272221

版权

redis 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

该代码展示了如何使用Java实现基于Redis的分布式锁，包括可重入特性。通过StringRedisTemplate操作Redis，利用Lua脚本保证原子性，实现锁的获取、释放以及自动续期功能。

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

#Redis分布式锁Java代码实现

/**
 * redis分布式锁 可重入
 * */
public class RedisDistributedLock implements Lock {

    private StringRedisTemplate stringRedisTemplate;

    private String lockName; //KEYS[1]
    private String uuidValue; //ARGV[1]
    private long expireTime; //ARGV[2]

    public RedisDistributedLock(StringRedisTemplate stringRedisTemplate, String lockName) {
        this.stringRedisTemplate = stringRedisTemplate;
        this.lockName = lockName;
        this.uuidValue = IdUtil.simpleUUID() + ":" + Thread.currentThread().getId();
        this.expireTime = 50L;
    }

    @Override
    public void lock() {
        tryLock();
    }


    @Override
    public boolean tryLock() {
        try {
            tryLock(-1, TimeUnit.SECONDS);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        return false;
    }

    @Override
    public boolean tryLock(long time, @NotNull TimeUnit unit) throws InterruptedException {
        if (time == -1L){
            String script = "if redis.call('exists', KEYS[1]) == 0 or redis.call('hexists', KEYS[1], ARGV[1]) == 1 then " +
                                    "redis.call('hincrby', KEYS[1], ARGV[1], 1)  " +
                                    "redis.call('expire', KEYS[1], ARGV[2])   " +
                            "else   " +
                                    " return 0  " +
                            "end";
            while(!stringRedisTemplate.execute(new DefaultRedisScript<>(script, Boolean.class), Arrays.asList(lockName), uuidValue, expireTime)) {
                // 暂停60毫秒
                try { TimeUnit.MILLISECONDS.sleep(60);} catch (InterruptedException e) { e.printStackTrace();};
            }
            renewExpire();
            return true;
        }
        return false;
    }

    // 自动续期  watch dog的思想
    private void renewExpire() {

        String script = "if redis.call('HEXISTS', KEYS[1], ARGV[1]) == 1 then  " +
                                "return redis.call('expire', KEYS[1], ARGV[2])  " +
                        "else  " +
                                "return 0 " +
                        "end";

        new Timer().schedule(new TimerTask() {
            @Override
            public void run() {
                if (stringRedisTemplate.execute(new DefaultRedisScript<>(script, Boolean.class), Arrays.asList(lockName), uuidValue, expireTime)){
                    renewExpire();
                }
            }
        }, (this.expireTime * 1000) / 3);
    }

    @Override
    public void unlock() {
        String script = "if redis.call('HEXISTS', KEYS[1], ARGV[1]) == 0 then  " +
                                "return nil  " +
                        "elseif redis.call('HINCRBY', KEYS[1] ,ARGV[1], -1) == 0 then  " +
                                "return redis.call('del', KEYS[1])   " +
                        "else  " +
                                "return 0 " +
                        "end";
        Long flag = stringRedisTemplate.execute(new DefaultRedisScript<>(script, Long.class), Arrays.asList(lockName), uuidValue, expireTime);
        if (flag == null){
            throw new RuntimeException("this lock doesn't not exists!!!!!");
        }
    }

    @Override
    public void lockInterruptibly() throws InterruptedException {

    }

    @NotNull
    @Override
    public Condition newCondition() {
        return null;
    }
}