一文让你搞懂如何手写一个redis分布式锁
作者:指北君 发布时间:2023-11-29 02:46:30
场景
既然要搞懂Redis分布式锁,那肯定要有一个需要它的场景。
高并发售票问题就是一个经典案例。
搭建环境
准备redis服务,设置redis的键值对:
set ticket 10
准备 postman、JMeter 等模拟高并发请求的工具
核心代码
@Service
public class TicketServiceImpl implements TicketService {
@Autowired
private StringRedisTemplate stringRedisTemplate;
private Logger logger = LoggerFactory.getLogger(TicketServiceImpl.class);
@Override
public String sellTicket() {
String ticketStr = stringRedisTemplate.opsForValue().get("ticket");
int ticket = 0;
if (null != ticketStr) {
ticket = Integer.parseInt(ticketStr);
}
if (ticket > 0) {
int ticketNew = ticket - 1;
stringRedisTemplate.opsForValue().set("ticket", String.valueOf(ticketNew));
logger.info("当前票的库存为:" + ticketNew);
} else {
logger.info("手速不够呀,票已经卖光了...");
}
return "抢票成功...";
}
}
分析解决问题
以上代码没有做任何的加锁操作,在高并 * 况下,票的超卖情况很严重,根本无法正常使用
分析1
既然要加分布式锁,那么我们可以使用Redis中的setnx
命令来模拟一个锁。
redis> EXISTS job # job 不存在
(integer) 0
redis> SETNX job "programmer" # job 设置成功
(integer) 1
redis> SETNX job "code-farmer" # 尝试覆盖 job ,失败
(integer) 0
当一个线程进入到当前方法中,使用 setnx
设置一个键,如果设置成功,就允许继续访问,设置失败,就不能访问该方法;
当方法运行完毕时,将这个键删除,下一次再有线程来访问时,就重新执行该操作。
public String sellTicket() {
String lock="lock";
// 如果成功设置这个值,证明目前该方法并没有 * 作,可以进行卖票操作
Boolean tag = stringRedisTemplate.opsForValue().setIfAbsent(lock, "");
if (!tag) { // 如果设置失败,证明当前方法正在被执行,不允许再次执行
// 实际开发环境应该使用队列来完成访问操作,这里主要探究分布式锁的问题,所以仅仅模拟了场景
// 这里使用自旋的方式,防止访问信息丢失
sellTicket();
return "当前访问人数过多,请稍后访问...";
}
String ticketStr = stringRedisTemplate.opsForValue().get("ticket");
int ticket = 0;
if (null != ticketStr) {
ticket = Integer.parseInt(ticketStr);
}
if (ticket > 0) {
int ticketNew = ticket - 1;
stringRedisTemplate.opsForValue().set("ticket", String.valueOf(ticketNew));
logger.info("当前票的库存为:" + ticketNew);
} else {
logger.info("手速不够呀,票已经卖光了...");
}
stringRedisTemplate.delete(lock);
return "抢票成功...";
}
分析2
上述的代码在程序正常运行下不会出现票超卖的问题,但是我们需要考虑:
1.如果程序运行中系统出现了异常,导致无法删除lock
,就会造成死锁的问题。也许有人马上就会想到,使用 try{} finally {}
,在finally中进行删除锁的操作。
但是,如果是分布式架构,第一个服务器接收到请求,加了锁,此时第二个服务器也接收到请求,setnx
命令失败,需要执行return操作,根据finally的特性,执行return之前,需要先执行finally里的代码,于是,第二个服务器把锁给删除了,程序中锁失效了,肯定会出现票超卖等一系列问题。
2.如果程序在运行中直接彻底死了(比如,程序员闲着没事儿,来了个 kill -9;或者断电),就算加了finally,finally也不能执行,还是会出现死锁问题
解决方法:
给锁加一个标识符,只允许自己来操作锁,其他访问程序不能操作锁
还要给锁加一个过期时间,这样就算程序死了,当时间过期后,还是能够继续执行
public String sellTicket() {
String lock="lock"; // 锁的键
String lockId = UUID.randomUUID().toString(); // 锁的值:唯一标识
try{
// 如果成功设置这个值,证明目前该方法并没有 * 作,可以进行卖票操作
// 添加一个过期时间,暂定为 30秒,这里的操作具有原子性,如果过期时间设置失败,键也会设置失败
Boolean tag = stringRedisTemplate.opsForValue().setIfAbsent(lock, lockId, 30, TimeUnit.SECONDS);
if (!tag) { // 如果设置失败,证明当前方法正在被执行,不允许再次执行
// 实际开发环境应该使用队列来完成访问操作,这里主要探究分布式锁的问题,所以仅仅模拟了场景
// 不设置回调的话,访问信息会丢失
sellTicket();
return "当前访问人数过多,请稍后访问...";
}
String ticketStr = stringRedisTemplate.opsForValue().get("ticket");
int ticket = 0;
if (null != ticketStr) {
ticket = Integer.parseInt(ticketStr);
}
if (ticket > 0) {
int ticketNew = ticket - 1;
stringRedisTemplate.opsForValue().set("ticket", String.valueOf(ticketNew));
logger.info("当前票的库存为:" + ticketNew);
} else {
logger.info("手速不够呀,票已经卖光了...");
}
} finally {
// 如果redis中的值,和当前的值一致,才允许删除锁。
if (lockId.equals(stringRedisTemplate.opsForValue().get(lock))) {
stringRedisTemplate.delete(lock);
}
}
return "抢票成功...";
}
分析3
写到这里已经可以解决大部分问题了,但是还需要考虑一个问题:
如果程序运行的极慢(硬件处理慢或者进行了GC),导致30秒已经到了,锁已经失效了,程序还没有运行完成,这时候,就会有另一个线程总想钻个空子,导致票的超卖问题。
这里我们可以使用 sleep 模拟一下
......
if (ticket > 0) {
try {
// 为了测试方便,过期时间和线程暂停时间都改成了3秒
Thread.sleep(3000);
} catch (InterruptedException e) {
e.printStackTrace();
}
int ticketNew = ticket - 1;
stringRedisTemplate.opsForValue().set("ticket", String.valueOf(ticketNew));
......
这样运行就会出现极其严重的超卖问题
那么该如何设置这个过期时间呢?继续加大?这显然是不合适的,因为无论多么大,总有可能出现问题。
解决方法
我们可以使用守护线程,来保证这个时间永不过期
public String sellTicket() {
String lock="lock"; // 锁的键
String lockId = UUID.randomUUID().toString(); // 锁的值:唯一标识
MyThread myThread = null; // 锁的守护线程
try{
// 如果成功设置这个值,证明目前该方法并没有 * 作,可以进行卖票操作
// 添加一个过期时间,暂定为 3 秒,这里的操作具有原子性,如果过期时间设置失败,键也会设置失败
Boolean tag = stringRedisTemplate.opsForValue().setIfAbsent(lock, lockId, 3, TimeUnit.SECONDS);
if (!tag) { // 如果设置失败,证明当前方法正在被执行,不允许再次执行
// 实际开发环境应该使用队列来完成访问操作,这里主要探究分布式锁的问题,所以仅仅模拟了场景
// 不设置回调的话,访问信息会丢失
sellTicket();
return "当前访问人数过多,请稍后访问...";
}
// 开启守护线程, 每隔三分之一的时间,给锁续命
myThread = new MyThread(lock);
myThread.setDaemon(true);
myThread.start();
String ticketStr = stringRedisTemplate.opsForValue().get("ticket");
int ticket = 0;
if (null != ticketStr) {
ticket = Integer.parseInt(ticketStr);
}
if (ticket > 0) {
try {
Thread.sleep(3000);
} catch (InterruptedException e) {
e.printStackTrace();
}
int ticketNew = ticket - 1;
stringRedisTemplate.opsForValue().set("ticket", String.valueOf(ticketNew));
logger.info("当前票的库存为:" + ticketNew);
} else {
logger.info("手速不够呀,票已经卖光了...");
}
} finally {
// 如果redis中的值,和当前的值一致,才允许删除锁。
if (lockId.equals(stringRedisTemplate.opsForValue().get(lock))) {
// 程序运行结束,需要关闭守护线程
myThread.stop();
stringRedisTemplate.delete(lock);
logger.info("释放锁成功...");
}
}
return "抢票成功...";
}
/** 使用后台线程进行续命
* 守护线程
* 在主线程下 如果有一个守护线程 这个守护线程的生命周期 跟主线程是同生死的
*/
class MyThread extends Thread{
String lock;
MyThread (String lock) {
this.lock = lock;
}
@Override
public void run() {
while (true) {
try {
// 三分之一的时间
Thread.sleep(1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
// 假设线程还活着,就要给锁续命
logger.info("线程续命ing...");
stringRedisTemplate.expire(lock, 3, TimeUnit.SECONDS);
}
}
}
来源:https://mp.weixin.qq.com/s/Qw9icDnE28wnIX6qv44D0g
猜你喜欢
- 假设下面是你的视频网站链接列表,如果别人想爬取你的数据十分轻松,看规则就知道数据库是序列自增的http://www.xxxx.com/vid
- 1 自定义类加载器自定义类加载器的代码很简单,只需要继承ClassLoader类,覆写findClass方法即可其默认实现是会抛出一个异常:
- [LeetCode] 131.Palindrome Partitioning 拆分回文串Given a string s, par
- 一个是新浪微博,腾讯微博的分享按钮,一个是他们的绑定情况(其实就是是否授权)。点击微博分享中新浪或腾讯按钮,就进行相应的授权(若没授权),显
- 初次安装Android Studio,遇到了不少问题,这是其中的一个,分享如下,同时求各位dalao关注一下啦((*^__^*) )使用不同
- Logback 背景Logback是由log4j创始人设计的另一个开源日志组件,官方网站:http://logback.qos.ch。它当前
- JDK8中有双冒号的用法,就是把方法当做参数传到stream内部,使stream的每个元素都传入到该方法里面执行一下。代码其实很简单:以前的
- 引言设计: 嗯? 这个图片点击跳转进详情再返回图片怎么变白闪一下呢?产品: 是啊是啊! 一定是个bug开发: 囧囧囧在开发过程中, 也许你也
- 介绍备忘录模式(Memento Pattern)是一种行为型设计模式,它允许在不破坏封装性的前提下,捕获并保存一个对象的内部状态,并在之后可
- 接触FFmpeg有一段时间了,它是音视频开发的开源库,几乎其他所有播放器、直播平台都基于FFmpeg进行二次开发。本篇文章来总结下采用FFm
- public static String getCharset(File file) { &n
- 1、首先导入solrj需要的的架包2、需要注意的是低版本是solr是使用SolrServer进行URL实例的,5.0之后已经使用SolrCl
- 提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档前言这两天在项目中使用到Java的导入导出功能,以前对这块有一定了解,但是没
- Android 集成FlutterFlutter 作为 Google 开源的新一代跨平台、高性能 UI 框架,旨在帮助开发者高效地构建出跨平
- 本文实例讲述了java中Object类用法。分享给大家供大家参考。具体如下:1、Object类是所有java类的基类如果在类的声明中未使用e
- import java.io.BufferedReader;import java.io.File;import java.io.FileI
- 本章将具体讲解下拉菜单的交互。使用下拉菜单(Dropdown)插件,您可以向任何组件(比如导航栏、标签页、胶囊式导航菜单、按钮等)添加下拉菜
- 该工具包含是封装了jedis,包含redis.properties和jedisPool,序列化使用的是protostuff,map类型操作使
- 一、文件的编码package com.study.io;/*** 测试文件编码*/public class EncodeDemo {/***
- 前言在我们平时使用图形化界面的时候,会发现来建立一个文件夹或者一个文档的时候很简单,只需要在桌面单击鼠标右键就可以了。但是,在我们写项目的时