com.github.dennisit.vplus.data.lock.redis.RedisDistributeLock Maven / Gradle / Ivy
/*--------------------------------------------------------------------------
* Copyright (c) 2010-2020, Elon.su All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
* Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
* Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* Neither the name of the elon developer nor the names of its
* contributors may be used to endorse or promote products derived from
* this software without specific prior written permission.
* Author: Elon.su, you can also mail [email protected]
*--------------------------------------------------------------------------
*/
package com.github.dennisit.vplus.data.lock.redis;
import com.github.dennisit.vplus.data.lock.DistributeLock;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.data.redis.core.StringRedisTemplate;
import java.util.concurrent.TimeUnit;
/**
* Created by Elon.su on 17/11/6.
*/
public class RedisDistributeLock implements DistributeLock {
private static final Logger LOG = LoggerFactory.getLogger(RedisDistributeLock.class);
/**
* redis客户端
*/
private StringRedisTemplate redisTemplate;
/**
* 分布式锁Redis
* @param redisTemplate
*/
private RedisDistributeLock(StringRedisTemplate redisTemplate){
this.redisTemplate = redisTemplate;
}
@Override
public boolean tryLock(String key, int tryMilliSeconds) {
try {
long start = System.currentTimeMillis();
while (true) {
if (redisTemplate.opsForValue().setIfAbsent(key, "")) {
redisTemplate.expire(key, tryMilliSeconds, TimeUnit.MILLISECONDS);
return true;
}
long end = System.currentTimeMillis();
if (tryMilliSeconds > 0 && tryMilliSeconds < (end - start)) {
return false;
}
}
} catch (Exception e) {
LOG.error("get RedisDistributeLock {} error", key, e);
}
return false;
}
@Override
public void unlock(String key) {
redisTemplate.delete(key);
}
}