分布式锁一般有三种实现方式:1. 数据库乐观锁;2. 基于Redis的分布式锁;3. 基于ZooKeeper的分布式锁。本篇博客将介绍第二种方式,基于Redis实现分布式锁。
首先,为了确保分布式锁可用,我们至少要确保锁的实现同时满足以下四个条件:
- 互斥性。在任意时刻,只有一个客户端能持有锁。
- 不会发生死锁。即使有一个客户端在持有锁的期间崩溃而没有主动解锁,也能保证后续其他客户端能加锁。
- 具有容错性。只要大部分的Redis节点正常运行,客户端就可以加锁和解锁。
- 解铃还须系铃人。加锁和解锁必须是同一个客户端,客户端自己不能把别人加的锁给解了。
import org.apache.commons.lang3.StringUtils;
import org.springframework.data.redis.core.StringRedisTemplate;
/**
* @描述:Redis 做分布式锁,利用 redis的setnx(SET if Not eXists)命令 和 getset 命令,即下面代码中的方法 setIfAbsent 和 getAndSet
*
*/
public class RedisLock {
private StringRedisTemplate redisTemplate;
/**
* 加锁
* @param key 被秒杀商品的id
* @param value 当前线程操作时的 System.currentTimeMillis() + 2000,2000是超时时间,这个地方不需要去设置redis的expire,
* 也不需要超时后手动去删除该key,因为会存在并发的线程都会去删除,造成上一个锁失效,结果都获得锁去执行,并发操作失败了就。
* @return
*/
public boolean lock(String key, String value) {
//如果key值不存在,则返回 true,且设置 value
if (redisTemplate.opsForValue().setIfAbsent(key, value)) {
return true;
}
//获取key的值,判断是是否超时
String curVal = redisTemplate.opsForValue().get(key);
if (StringUtils.isNotEmpty(curVal) && Long.parseLong(curVal) < System.currentTimeMillis()) {
//获得之前的key值,同时设置当前的传入的value。这个地方可能几个线程同时过来,但是redis本身天然是单线程的,所以getAndSet方法还是会安全执行,
//首先执行的线程,此时curVal当然和oldVal值相等,因为就是同一个值,之后该线程set了自己的value,后面的线程就取不到锁了
String oldVal = redisTemplate.opsForValue().getAndSet(key, value);
if(StringUtils.isNotEmpty(oldVal) && oldVal.equals(curVal)) {
return true;
}
}
return false;
}
/**
* 解锁
* @param key
* @param value
*/
public void unlock(String key, String value) {
try {
String curVal = redisTemplate.opsForValue().get(key);
if (StringUtils.isNotEmpty(curVal) && curVal.equals(value)) {
redisTemplate.opsForValue().getOperations().delete(key);
}
} catch (Exception e) {
e.printStackTrace();
}
}
}