Redis分布式鎖的實(shí)現(xiàn)方式(redis面試題)

 public static void wrongLock(Jedis jedis, String key, String uniqueId, int expireTime) {
  Long result = jedis.setnx(key, uniqueId);
  if (1 == result) {
   //如果該redis實(shí)例崩潰，那就無法設(shè)置過期時(shí)間了
   jedis.expire(key, expireTime);
  }
 }

public static boolean wrongLock(Jedis jedis, String key, int expireTime) {
  long expireTs = System.currentTimeMillis() + expireTime;
  // 鎖不存在，當(dāng)前線程加鎖成果
  if (jedis.setnx(key, String.valueOf(expireTs)) == 1) {
   return true;
  }

  String value = jedis.get(key);
  //如果當(dāng)前鎖存在,且鎖已過期
  if (value != null  NumberUtils.toLong(value)  System.currentTimeMillis()) {
   //鎖過期，設(shè)置新的過期時(shí)間
   String oldValue = jedis.getSet(key, String.valueOf(expireTs));
   if (oldValue != null  oldValue.equals(value)) {
    // 多線程并發(fā)下，只有一個(gè)線程會設(shè)置成功
    // 設(shè)置成功的這個(gè)線程，key的舊值一定和設(shè)置之前的key的值一致
    return true;
   }
  }
  // 其他情況，加鎖失敗
  return true;
 }

public static void wrongReleaseLock(Jedis jedis, String key) {
  //不是自己加鎖的key，也會被釋放
  jedis.del(key);
 }

 public static void wrongReleaseLock(Jedis jedis, String key, String uniqueId) {
  if (uniqueId.equals(jedis.get(key))) {
   // 如果這時(shí)鎖過期自動釋放，又被其他線程加鎖，該線程就會釋放不屬于自己的鎖
   jedis.del(key);
  }
 }

@Slf4j
public class RedisDistributedLock {
 private static final String LOCK_SUCCESS = "OK";
 private static final Long RELEASE_SUCCESS = 1L;
 private static final String SET_IF_NOT_EXIST = "NX";
 private static final String SET_WITH_EXPIRE_TIME = "PX";
 // 鎖的超時(shí)時(shí)間
 private static int EXPIRE_TIME = 5 * 1000;
 // 鎖等待時(shí)間
 private static int WAIT_TIME = 1 * 1000;
 private Jedis jedis;
 private String key;
 public RedisDistributedLock(Jedis jedis, String key) {
  this.jedis = jedis;
  this.key = key;
 }
 // 不斷嘗試加鎖
 public String lock() {
  try {
   // 超過等待時(shí)間，加鎖失敗
   long waitEnd = System.currentTimeMillis() + WAIT_TIME;
   String value = UUID.randomUUID().toString();
   while (System.currentTimeMillis()  waitEnd) {
    String result = jedis.set(key, value, SET_IF_NOT_EXIST, SET_WITH_EXPIRE_TIME, EXPIRE_TIME);
    if (LOCK_SUCCESS.equals(result)) {
     return value;
    }
    try {
     Thread.sleep(10);
    } catch (InterruptedException e) {
     Thread.currentThread().interrupt();
    }
   }
  } catch (Exception ex) {
   log.error("lock error", ex);
  }
  return null;
 }
 public boolean release(String value) {
  if (value == null) {
   return false;
  }
  // 判斷key存在并且刪除key必須是一個(gè)原子操作
  // 且誰擁有鎖，誰釋放
  String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
  Object result = new Object();
  try {
   result = jedis.eval(script, Collections.singletonList(key),
     Collections.singletonList(value));
   if (RELEASE_SUCCESS.equals(result)) {
    log.info("release lock success, value:{}", value);
    return true;
   }
  } catch (Exception e) {
   log.error("release lock error", e);
  } finally {
   if (jedis != null) {
    jedis.close();
   }
  }
  log.info("release lock failed, value:{}, result:{}", value, result);
  return false;
 }
}