Redis分布式锁解决抢购问题

时间:2023-06-28 22:43:53
Redis分布式锁解决抢购问题

转:https://segmentfault.com/a/1190000011421467

废话不多说,首先分享一个业务场景-抢购。一个典型的高并发问题,所需的最关键字段就是库存,在高并发的情况下每次都去数据库查询显然是不合适的,因此把库存信息存入Redis中,利用redis的锁机制来控制并发访问,是一个不错的解决方案。

首先是一段业务代码:

@Transactional
public void orderProductMockDiffUser(String productId){
//1.查库存
int stockNum = stock.get(productId);
if(stocknum == 0){
throw new SellException(ProductStatusEnum.STOCK_EMPTY);
//这里抛出的异常要是运行时异常,否则无法进行数据回滚,这也是spring中比较基础的
}else{
//2.下单
orders.put(KeyUtil.genUniqueKey(),productId);//生成随机用户id模拟高并发
sotckNum = stockNum-1;
try{
Thread.sleep(100);
} catch (InterruptedExcption e){
e.printStackTrace();
}
stock.put(productId,stockNum);
}
}

这里有一种比较简单的解决方案,就是synchronized关键字。

public synchronized void orderProductMockDiffUser(String productId)

这就是java自带的一种锁机制,简单的对函数加锁和释放锁。但问题是这个实在是太慢了,感兴趣的可以可以写个接口用apache ab压测一下。

ab -n 500 -c 100 http://localhost:8080/xxxxxxx

下面就是redis分布式锁的解决方法。首先要了解两个redis指令
SETNX 和 GETSET,可以在redis中文网上找到详细的介绍。
SETNX就是set if not exist的缩写,如果不存在就返回保存value并返回1,如果存在就返回0。
GETSET其实就是两个指令GET和SET,首先会GET到当前key的值并返回,然后在设置当前Key为要设置Value。

首先我们先新建一个RedisLock类:

/***
* 加锁
* @param key
* @param value 当前时间+超时时间
* @return 锁住返回true
*/
public boolean lock(String key,String value){
if(stringRedisTemplate.opsForValue().setIfAbsent(key,value)){//setNX 返回boolean
return true;
}
//如果锁超时 ***
String currentValue = stringRedisTemplate.opsForValue().get(key);
if(!StringUtils.isEmpty(currentValue) &amp;&amp; Long.parseLong(currentValue)<System.currentTimeMillis()){
//获取上一个锁的时间
String oldvalue = stringRedisTemplate.opsForValue().getAndSet(key,value);
if(!StringUtils.isEmpty(oldvalue)&amp;&amp;oldvalue.equals(currentValue)){
return true;
}
}
return false;
}
/***
* 解锁
* @param key
* @param value
* @return
*/
public void unlock(String key,String value){
try {
String currentValue = stringRedisTemplate.opsForValue().get(key);
if(!StringUtils.isEmpty(currentValue)&amp;&amp;currentValue.equals(value)){
stringRedisTemplate.opsForValue().getOperations().delete(key);
}
} catch (Exception e) {
log.error(&quot;解锁异常&quot;);
}
}

}" title="" data-original-title="复制">

@Slf4j

@Component

public class RedisService {

@Autowired

private StringRedisTemplate stringRedisTemplate;
<span class="hljs-comment">/***
* 加锁
* <span class="hljs-doctag">@param</span> key
* <span class="hljs-doctag">@param</span> value 当前时间+超时时间
* <span class="hljs-doctag">@return</span> 锁住返回true
*/</span>
<span class="hljs-keyword">public</span> boolean lock(String key,String value){
<span class="hljs-keyword">if</span>(stringRedisTemplate.opsForValue().setIfAbsent(key,value)){<span class="hljs-comment">//setNX 返回boolean</span>
<span class="hljs-keyword">return</span> <span class="hljs-literal">true</span>;
}
<span class="hljs-comment">//如果锁超时 ***</span>
String currentValue = stringRedisTemplate.opsForValue().<span class="hljs-keyword">get</span>(key);
<span class="hljs-keyword">if</span>(!StringUtils.isEmpty(currentValue) &amp;&amp; <span class="hljs-built_in">Long</span>.parseLong(currentValue)&lt;System.currentTimeMillis()){
<span class="hljs-comment">//获取上一个锁的时间</span>
String oldvalue = stringRedisTemplate.opsForValue().getAndSet(key,value);
<span class="hljs-keyword">if</span>(!StringUtils.isEmpty(oldvalue)&amp;&amp;oldvalue.equals(currentValue)){
<span class="hljs-keyword">return</span> <span class="hljs-literal">true</span>;
}
}
<span class="hljs-keyword">return</span> <span class="hljs-literal">false</span>;
}
<span class="hljs-comment">/***
* 解锁
* <span class="hljs-doctag">@param</span> key
* <span class="hljs-doctag">@param</span> value
* <span class="hljs-doctag">@return</span>
*/</span>
<span class="hljs-keyword">public</span> void unlock(String key,String value){
<span class="hljs-keyword">try</span> {
String currentValue = stringRedisTemplate.opsForValue().<span class="hljs-keyword">get</span>(key);
<span class="hljs-keyword">if</span>(!StringUtils.isEmpty(currentValue)&amp;&amp;currentValue.equals(value)){
stringRedisTemplate.opsForValue().getOperations().delete(key);
}
} <span class="hljs-keyword">catch</span> (Exception e) {
log.error(<span class="hljs-string">"解锁异常"</span>);
}
}

}

这个项目是springboot的项目。首先要加入redis的pom依赖,该类只有两个功能,加锁和解锁,解锁比较简单,就是删除当前key的键值对。我们主要来说一说加锁这个功能。
首先,锁的value值是当前时间加上过期时间的时间戳,Long类型。首先看到用setiFAbsent方法也就是对应的SETNX,在没有线程获得锁的情况下可以直接拿到锁,并返回true也就是加锁,最后没有获得锁的线程会返回false。 最重要的是中间对于锁超时的处理,如果没有这段代码,当秒杀方法发生异常的时候,后续的线程都无法得到锁,也就陷入了一个死锁的情况。我们可以假设CurrentValue为A,并且在执行过程中抛出了异常,这时进入了两个value为B的线程来争夺这个锁,也就是走到了注释*的地方。currentValue==A,这时某一个线程执行到了getAndSet(key,value)函数(某一时刻一定只有一个线程执行这个方法,其他要等待)。这时oldvalue也就是之前的value等于A,在方法执行过后,oldvalue会被设置为当前的value也就是B。这时继续执行,由于oldValue==currentValue所以该线程获取到锁。而另一个线程获取的oldvalue是B,而currentValue是A,所以他就获取不到锁啦。多线程还是有些乱的,需要好好想一想。
接下来就是在业务代码中加锁啦:首要要@Autowired注入刚刚RedisLock类,不要忘记对这个类加一个@Component注解否则无法注入

private static final int TIMEOUT= 10*1000;
@Transactional
public void orderProductMockDiffUser(String productId){
long time = System.currentTimeMillions()+TIMEOUT;
if(!redislock.lock(productId,String.valueOf(time)){
throw new SellException(101,"换个姿势再试试")
}
//1.查库存
int stockNum = stock.get(productId);
if(stocknum == 0){
throw new SellException(ProductStatusEnum.STOCK_EMPTY);
//这里抛出的异常要是运行时异常,否则无法进行数据回滚,这也是spring中比较基础的
}else{
//2.下单
orders.put(KeyUtil.genUniqueKey(),productId);//生成随机用户id模拟高并发
sotckNum = stockNum-1;
try{
Thread.sleep(100);
} catch (InterruptedExcption e){
e.printStackTrace();
}
stock.put(productId,stockNum);
}
redisLock.unlock(productId,String.valueOf(time));
}

大功告成了!比synchronized快了不知道多少倍,再也不会被老板骂了!

            </div>
posted on
2019-09-20 08:43 
duende99 
阅读(...) 
评论(...) 
编辑 
收藏