今天翻看去年写的代码,发现了有一块用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 = 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 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);
}