首页 > 代码库 > redis 分布式锁 获取redis的key

redis 分布式锁 获取redis的key


今天翻看去年写的代码,发现了有一块用redis写了分布式锁,业务场景就是,再搭建第一版爬虫的时候,用来 定时任务来获取redis里的key值,保证同一时间只有一台机器拿到这个key,用到这样的办法来做,可以后来发现,机器多的话,反而花费时间在争抢锁上,毫无意义,也就放弃了这版爬虫框架。

下面是代码,自己可以拿去试试,调调,因为用的时间短,可能会有bug。


/**

* 获取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 = http://www.mamicode.com/Long.valueOf(getLock(lockKey));

       //超时

       if (oldValue < System.currentTimeMillis()) {

        String getValue = http://www.mamicode.com/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 releaseLock(String lockKey) {

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);

}


本文出自 “10093778” 博客,请务必保留此出处http://10103778.blog.51cto.com/10093778/1914852

redis 分布式锁 获取redis的key