Redis分散ロック実装原理java版


説明:
Redisは単一スレッドであるため,それ自体にロックの概念はない.
だから分布式ロックの実現原理は
Redisにキー(呼び出しメソッド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);
  }
 }