chenqb 发表于 2018-11-4 08:07:51

redis 分布式锁

  分布式锁实现,redis实现。
  setnx命令:setnx key value。
  结果:如果redis里这个key,则返回0,如果redis里没这个key,则返回value是1。
  getset命令:
  如果redis里有这个key,则返回旧数据的value。如果redis里没有这个key就直接插入进去了。
  /**
  * 获取redis对应的爬虫key的锁
  * @param lockKey
  * @return
  */
  public boolean obtainLock(String lockKey) {
  // 1. 通过SETNX试图获取一个lock
  boolean obtainLockSuccess = false;
  boolean lock = false;
  //默认为当前时间加6000毫秒,一分钟,根据需求更改
  long value= System.currentTimeMillis() + 6000 + 1;
  //分布式锁开始
  lock = setNX(lockKey, String.valueOf(value));
  if (lock) {//如果等于true,说明setNX成功,不懂的话可以自己登陆redis客户端试一试setnx命令
  obtainLockSuccess = true;
  }else {
  long oldValue = Long.valueOf(getLock(lockKey));
  //超时
  if (oldValue < System.currentTimeMillis()) {
  String getValue = getSet(lockKey, String.valueOf(value));
  // 获取锁成功,什么意思?是因为需要判断最开始设值进去的value和getset得道的value是否是一样的,不一样啊,说明这期间
  //有人getset了,你得到的是别得值,说明获取这个锁失败,就是并发的有人比你快一步,自己这块需要拿笔在本上写一写。
  if (Long.valueOf(getValue) == oldValue) {
  obtainLockSuccess = true;
  }else { //获取锁失败
  obtainLockSuccess = false;
  }
  }else { //未超时,则直接返回失败,现在还要人用到这个key的锁
  obtainLockSuccess = false;
  }
  }
  return obtainLockSuccess;
  }
  /**
  * 释放锁,也就是删除锁
  * @param lockKey
  */

  public void>  long current = System.currentTimeMillis();
  // 避免删除非自己获取得到的锁
  if (current < Long.valueOf(valueOperations.get(lockKey)))
  redisTemplate.delete(lockKey);
  }
  /**
  * 获取这个Key的value,把这个value当入key,再放入redis队列里,String类型
  * @param key
  * @param value
  * @return
  */
  public boolean setNX(String key, String value) {
  return valueOperations.setIfAbsent(key, value);
  }
  /**
  * 获取setNx放入里的Key
  * @param key
  * @return
  */
  public String getLock(String key) {
  return valueOperations.get(key);
  }
  /**
  * 这个getSet是redis的特性,这个是先设置新的值覆盖,返回给原来老的值
  * @param key
  * @param value
  * @return
  */
  public String getSet(String key, String value) {
  return valueOperations.getAndSet(key, value);
  }

页: [1]
查看完整版本: redis 分布式锁