自定义分布式锁

最新推荐文章于 2024-04-04 05:46:18 发布

hq091117

最新推荐文章于 2024-04-04 05:46:18 发布

阅读量220

点赞数

分类专栏： redis 文章标签： java redis

本文链接：https://blog.csdn.net/hq091117/article/details/117250493

版权

redis 专栏收录该内容

13 篇文章 0 订阅

订阅专栏

自定义分布式锁

 /**
     * 获取分布式锁
     * @param lockKey 锁
     * @param requestId 请求标识
     * @param expireTime 单位秒
     * @param waitTimeout 单位毫秒
     * @return 是否获取成功
     */
    public boolean tryLock(String lockKey, String requestId, int expireTime,long waitTimeout) {

        lockKey = wrapperPrefix(lockKey);
        // 当前时间
        long nanoTime = System.nanoTime();
        try{
            String script = "if redis.call('setNx',KEYS[1],ARGV[1]) then if redis.call('get',KEYS[1])==ARGV[1] then return redis.call('expire',KEYS[1],ARGV[2]) else return 0 end end";

            logger.info("开始获取分布式锁-key[{}]",lockKey);
            int count = 0;
            do{
                RedisScript<Long> redisScript = new DefaultRedisScript<>(script, Long.class);

                logger.debug("尝试获取分布式锁-key[{}]requestId[{}]count[{}]",lockKey,requestId,count);
                Object result = redisTemplate.execute(redisScript, Collections.singletonList(lockKey),requestId,expireTime);

                if(SUCCESS.equals(result)) {
                    logger.debug("尝试获取分布式锁-key[{}]成功",lockKey);
                    return true;
                }
                //休眠500毫秒
                Thread.sleep(500L);
                count++;
            }while ((System.nanoTime() - nanoTime) < TimeUnit.MILLISECONDS.toNanos(waitTimeout));

        }catch(Exception e){
            logger.error("尝试获取分布式锁-key["+lockKey+"]异常",e);
        }

        logger.warn("获取分布式锁-key[{}]失败",lockKey);

        return false;
    }

    /**
     * 释放锁
     * @param lockKey 锁
     * @param requestId 请求标识
     * @return 是否释放成功
     */
    public boolean releaseLock(String lockKey, String requestId) {

        lockKey = wrapperPrefix(lockKey);
        String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";

        RedisScript<Long> redisScript = new DefaultRedisScript<>(script, Long.class);

        Object result = redisTemplate.execute(redisScript, Collections.singletonList(lockKey), requestId);
        if (SUCCESS.equals(result)) {
            logger.info("释放分布式锁-key[{}]成功",lockKey);
            return true;
        }

        logger.error("释放分布式锁-key[{}]失败",lockKey);
        return false;

    }

    /**
     * 包装前缀kshop
     * @param key
     * @return
     */
    private String wrapperPrefix(String key){
        return PREFIX + key;
    }