说明: Redis因为是单线程的,所以本身没有锁的概念。 所以分布式锁的实现原理是
往Redis当中写入一个key(调用方法setnx),写入成功相当于获取锁成功。写入失败也即是setnx方法返回0,获取锁失败。
注意锁的失效时间,否则容易造成死锁。
/**
* 锁在给定的等待时间内空闲,则获取锁成功 返回true, 否则返回false
* @param subKey
* @param timeout 如果timeout=0,取不到锁时,不等待,直接返回.
* @param unit
* @return
*/
public boolean tryLock(String subKey, long timeout, TimeUnit unit) {
String key = LOCK_KEY + subKey;
Jedis jedis = null;
try {
jedis = getConnection();
if (jedis == null) {
return Boolean.FALSE;
}
long nano = System.nanoTime();
do {
logger.debug("try lock key: " + key);
Long i = jedis.setnx(key, key);
if (i == 1) {
jedis.expire(key, EXPIRED_TIME);
logger.debug("get lock, key: " + key + " , expire in " + EXPIRED_TIME + " seconds.");
return Boolean.TRUE;
} else { // 存在锁
if (logger.isDebugEnabled()) {
String desc = jedis.get(key);
logger.debug("key: " + key + " locked by another business:" + desc);
}
}
if (timeout == 0) { //取不到锁时,不等待,直接返回.
break;
}
Thread.sleep(200);
} while ((System.nanoTime() - nano) < unit.toNanos(timeout));//取不到锁时等待,直到timeout
return Boolean.FALSE;
} catch (JedisConnectionException je) {
logger.error(je.getMessage(), je);
returnBorkenConnection(jedis);
} catch (Exception e) {
logger.error(e.getMessage(), e);
} finally {
returnConnection(jedis);
}
return Boolean.FALSE;
}
/**
* 如果锁空闲立即返回 获取失败 一直等待
* @param subKey
*/
public boolean lock(String subKey) {
String key = LOCK_KEY + subKey;
Jedis jedis = null;
boolean isLock = false;
try {
jedis = getConnection();
if (jedis == null) {
return isLock;
}
while (true) {
logger.debug("lock key: " + key);
Long i = jedis.setnx(key, key);
if (i == 1) {
jedis.expire(key, EXPIRED_TIME);
logger.debug("get lock, key: " + key + " , expire in " + EXPIRED_TIME + " seconds.");
isLock = true;
} else {
if (logger.isDebugEnabled()) {
String desc = jedis.get(key);
logger.debug("key: " + key + " locked by another business:" + desc);
}
isLock = false;
}
Thread.sleep(500);
}
} catch (JedisConnectionException je) {
logger.error(je.getMessage(), je);
returnBorkenConnection(jedis);
} catch (Exception e) {
logger.error(e.getMessage(), e);
} finally {
returnConnection(jedis);
}
return isLock;
}
/**
* 释放锁
* @param subKey
*/
public void unLock(String subKey) {
String key = LOCK_KEY + subKey;
Jedis jedis = null;
try {
jedis = getConnection();
if (jedis == null) {
return;
}
jedis.del(key);
logger.debug("release lock, keys :" + key);
} catch (JedisConnectionException je) {
logger.error(je.getMessage(), je);
returnBorkenConnection(jedis);
} catch (Exception e) {
logger.error(e.getMessage(), e);
} finally {
returnConnection(jedis);
}
}