Redis+lua实现分布式锁(非重入锁)

流程图:

 代码

public class LockRedis {

    private static final Logger logger = LoggerFactory.getLogger(LockRedis.class);

    @Autowired
    private RedisTemplate<String, Object> redisTemplate;

    private ThreadLocal<String> lockFlag = new ThreadLocal<>();

    /**
     * 释放锁的原子操作lua脚本
     */
    public static final String UNLOCK_LUA;
    /**
     * 默认的锁超时时间:15秒
     */
    private Long default_expire = 15000L;
    /**
     * 默认的重试次数 100次
     */
    private Integer default_retryTimes = 100;
    /**
     * 默认的重试间隔时间:0.3秒
     */
    private Long default_sleepMillis = 300L;

    static {
        StringBuilder sb = new StringBuilder();
        sb.append("if redis.call(\"get\",KEYS[1]) == ARGV[1] ");
        sb.append("then ");
        sb.append("    return redis.call(\"del\",KEYS[1]) ");
        sb.append("else ");
        sb.append("    return 0 ");
        sb.append("end ");
        UNLOCK_LUA = sb.toString();
    }

    /**
     * 加锁并自定义过期时间
     * @param key
     * @param default_expire
     * @return
     */
    public boolean lock(String key, long default_expire) {
        return lock(key, default_expire, null, null);
    }

    /**
     * 加锁
     * @param key
     * @return
     */
    public boolean lock(String key) {
        return lock(key, null, null, null);
    }

    /**
     * @param key         锁住的key
     * @param expire      锁存在时间
     * @param retryTimes  重试次数
     * @param sleepMillis 重试间隔
     * @return
     */
    public boolean lock(String key, Long expire, Integer retryTimes, Long sleepMillis) {
        expire = expire == null || expire < 0 ? default_expire : expire;
        retryTimes = retryTimes == null || retryTimes <= 0 ? default_retryTimes : retryTimes;
        sleepMillis = sleepMillis == null || sleepMillis <= 0 ? default_sleepMillis : sleepMillis;
        boolean result = false;
        do {
            result = setRedis(key, expire);
            if (!result) {
                try {
                    Thread.sleep(sleepMillis);
                } catch (InterruptedException e) {
                    logger.error("休眠线程异常", e);
                    return false;
                }
            }
        } while ((!result) && retryTimes-- > 0);
        return result;
    }

    private boolean setRedis(final String key, final long expire) {
        try {
            final String uuid = UUID.randomUUID().toString();
            lockFlag.set(uuid);
            final Boolean result = redisTemplate.execute((RedisCallback<Boolean>) redisConnection ->
                    redisConnection.set(key.getBytes(), uuid.getBytes(), Expiration.from(expire, TimeUnit.MILLISECONDS),
                            RedisStringCommands.SetOption.SET_IF_ABSENT)
            );
            return result;
        } catch (Exception e) {
            logger.error("设置redis锁异常", e);
        }
        return false;
    }

    /**
     * 释放锁
     * @param key
     * @return
     */
    public boolean unLock(final String key) {
        try {
            final Boolean result = redisTemplate.execute((RedisCallback<Boolean>) redisConnection ->
                    redisConnection.eval(UNLOCK_LUA.getBytes(), ReturnType.BOOLEAN, 1, key.getBytes(), lockFlag.get().getBytes()));
            return result;
        } catch (Exception e) {
            logger.error("释放redis锁异常", e);
        } finally {
            lockFlag.remove();
        }
        return false;
    }
}

 

 

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值