redis优化秒杀
- 1. 异步秒杀思路
- 1.1 在redis存入库存和订单信息
- 1.2 具体流程图
- 2. 实现
- 2.1 总结
- 3. Redis的消息队列
- 3.1 基于list实现消息队列
- 3.2 基于PubSub实现消息队列
- 3.3 基于stream实现消息队列
- 3.3.1 stream的单消费模式
- 3.3.2 stream的消费者组模式
- 3.4 基于stream消息队列实现异步秒杀
本文为学习redis时做的笔记,学习内容来自
黑马程序员Redis入门到实战教程,该教程是循序渐进的,所以不是一上来就讲完最后的解决方案了,请耐心看完
1. 异步秒杀思路
这是我们原本的秒杀思路,其中的流程都要经过mysql数据库,而mysql数据库的并发性能不是很好,而且为了避免线程安全问题,还加入了分布式锁,所以整个流程的性能不好,现在我们要去优化它。
我们可以把这整个流程比作一个餐馆点菜的过程,前台点菜并将菜品写在小票上,给顾客一份,后厨一份,后厨根据小票的内容依次做菜
根据这个例子,我们的流程也可以分为两个部分:
- 第一部分是判断秒杀资格(判断秒杀库存和校验一人一单)
- 第二部分是减库存创建订单流程
两个部分各自为一个线程,主线程判断秒杀资格,如果用户有资格,就开启一个独立线程完成耗时较久的第二部分
同时,我们也要优化判断秒杀资格的性能,将库存和订单存入redis中,如果判断用户有资格,先将优惠券id,用户id、订单id保存在阻塞队列中,并将订单id返回给用户,用户可以通过这个订单id完成支付操作,虽然此时还没有创建订单,但是在队列中迟早会创建,之后开启独立线程读取队列中的信息,完成下单。
现在的业务流程变成了在redis中判断秒杀资格,保存信息在队列中并返回订单id,性能和吞吐量大大提高
1.1 在redis存入库存和订单信息
现在讨论这两个东西需要什么样的结构去存储
因为库存只是一个数值,我们使用redis中的string类型去存储,key是优惠券的id,value是库存
到时只需判断库存是否大于0,如果用户有资格,库存要减一,相当于在redis中预减库存
因为需要一个优惠券id(key)能存很多用户id,而且用户id不能重复,所以订单信息我们使用set结构存储
到时只需看value中是否有该用户id来判断该用户是否下过单
1.2 具体流程图
为了保证过程的原子性,需要用到lua脚本,通过执行lua脚本后的结果来返回异常信息或者订单id,这样创建订单的时效性就没有那么强了,完全可以照着数据库能承受的范围去执行写的操作,用户只需要订单id就能完成支付操作
2. 实现
需求:
- 新增秒杀优惠券的同时,将优惠券信息保存在redis中
修改添加秒杀优惠券的方法,将优惠券信息保存在redis中
@Override
@Transactional
public void addSeckillVoucher(Voucher voucher) {
// 保存优惠券
save(voucher);
// 保存秒杀信息
SeckillVoucher seckillVoucher = new SeckillVoucher();
seckillVoucher.setVoucherId(voucher.getId());
seckillVoucher.setStock(voucher.getStock());
seckillVoucher.setBeginTime(voucher.getBeginTime());
seckillVoucher.setEndTime(voucher.getEndTime());
seckillVoucherService.save(seckillVoucher);
// 保存优惠券信息到redis
stringRedisTemplate.opsForValue().set(SECKILL_STOCK_KEY + voucher.getId(), voucher.getStock().toString());
}
添加一个秒杀优惠券
{
"title": "120元代金券",
"beginTime": "2023-11-01T01:11:11",
"actualValue": 12000,
"shopId": 1,
"subTitle": "周一至周五均可使用",
"payValue": 10000,
"stock": 100,
"endTime": "2024-11-01T01:11:11",
"type": 1,
"rules": "全场通用\\n无需预约\\n可无限叠加\\不兑现、不找零\\n仅限堂食"
}
- 基于lua脚本,判断秒杀库存、一人一单、决定用户是都抢购成功
写lua脚本
- 先写参数列表,再组合成key,最后写逻辑
-- 1. 参数列表
-- 1.1 优惠券id
local voucherId = ARGV[1];
-- 1.2 用户id
local userId = ARGV[2];
-- 2. key
-- 2.1 库存key
local stockKey = "seckill:stock" .. voucherId;
-- 2.2 订单key
local orderKey = "seckill:order" .. voucherId;
-- 3. 脚本业务
-- 判断库存是否充足
if (tonumber(redis.call('get',stockKey)) <= 0) then
-- 库存不足
return 1
end
-- 判断用户是否下单
if (redis.call('sismember',orderKey,userId) == 1) then
-- 用户下过单
return 2
end
-- 扣减库存
redis.call('incrby',stockKey,-1);
-- 下单,保存用户id到set集合
redis.call('sadd',orderKey,userId);
重写秒杀逻辑
- 调用lua脚本,根据返回的数字判断,并返回订单号
private static final DefaultRedisScript<Long> SECKILL_SCRIPT;
//初始化脚本
static {
SECKILL_SCRIPT = new DefaultRedisScript();
//读取文件位置,classpath就是resource
SECKILL_SCRIPT.setLocation(new ClassPathResource("seckill.lua"));
SECKILL_SCRIPT.setResultType(Long.class);
}
@Override
public Result seckillVoucher(Long voucherId) {
// 获取用户id
Long userId = UserHolder.getUser().getId();
// 1. 调用lua脚本
//不需要传key,所以传个空集合
Long result = stringRedisTemplate.execute(
SECKILL_SCRIPT,
Collections.emptyList(),
voucherId.toString(),userId.toString()
);
int intValue = result.intValue();
//2. 判断结果是否为0
if (intValue != 0) {
return Result.fail(intValue == 1 ? "库存不足" : "不要重复下单");
}
//3. 将优惠券id,用户id、订单id保存在阻塞队列中
//TODO 将优惠券id,用户id、订单id保存在阻塞队列中
long orderId = redisIdWorker.nextId("order");
//4. 返回订单id
return Result.ok(orderId);
}
- 如果抢购成功,将优惠券id和用户id封装后存入阻塞队列
//阻塞队列
private BlockingQueue<VoucherOrder> orderTasks = new ArrayBlockingQueue<>(1024 * 1024);
阻塞队列BlockingQueue,当一个线程尝试从一个阻塞队列中获取元素,如果队列中没有元素,这个线程就会被阻塞,当队列中有元素时就会被唤醒并获取元素
seckillVoucher中添加:
//3.2 放入阻塞队列中
orderTasks.add(order);
- 开启线程任务,不断从阻塞队列中获取信息,实现异步下单功能
- 建立线程池,当这个类初始化完线程就去执行VoucherOrderHandler类中的run方法,不断获取队列中的订单信息,去创建订单
@PostConstruct 注解效果是当前类初始化完就去执行
//线程池
private static final ExecutorService SECKILL_ORDER_EXECUTOR = Executors.newSingleThreadExecutor();
//类加载完就执行
@PostConstruct
private void init() {
SECKILL_ORDER_EXECUTOR.submit(new VoucherOrderHandler());
}
class VoucherOrderHandler implements Runnable {
@Override
public void run() {
while (true) {
try {
//1.获取队列中的订单信息
VoucherOrder voucherOrder = orderTasks.take();
//2.创建订单
handleVoucherOrder(voucherOrder);
} catch (Exception e) {
log.error("订单异常:" + e);
}
}
}
}
- 执行handleVoucherOrder方法,获取锁,最后执行创建订单方法
注1:在这个方法中不能使用threadlocal去获取用户信息,因为是异步下单,这是一个子线程,不是主线程,没有用户的信息,所以从订单中获取用户id
注2:与上同理,在这个子线程中也无法获取到代理对象,将代理对象设置为成员变量,再从主线程中获取到代理对象
private void handleVoucherOrder(VoucherOrder voucherOrder) {
Long userId = voucherOrder.getUserId();
//创建锁对象
RLock redisLock = redissonClient.getLock("lock:order:" + userId);
//获取锁方法
boolean lockFlag = redisLock.tryLock();
//判断是否获取成功
if (!lockFlag) {
log.error("不要重复下单");
return;
}
//事务方法执行起来可能会出现异常,但最后都要释放锁,所以try-catch起来
try {
proxy.createVoucherOrder(voucherOrder);
} finally {
redisLock.unlock();
}
}
- 创建订单逻辑
@Transactional
public void createVoucherOrder(VoucherOrder voucherOrder) {
//5.一人一单
Long userId = UserHolder.getUser().getId();
//查询
Integer count = query().eq("user_id", userId).eq("voucher_id", voucherOrder.getVoucherId()).count();
//判断订单是否存在
if (count > 0) {
log.error("用户已经购买过一次");
return;
}
//6.扣减库存
boolean success = seckillVoucherService.update().setSql("stock = stock-1")
.eq("voucher_id", voucherOrder.getVoucherId())
.gt("stock", 0)
.update();
if (!success) {
log.error("不要重复下单");
return;
}
save(voucherOrder);
}
2.1 总结
整个流程是:
- 主线程:
发送请求进入seckillVoucher方法,先判断用户是否有秒杀的资格(通过lua脚本),然后创建订单(将用户id、优惠券id、订单id放进订单里),将订单放入阻塞队列
- 子线程:
在类初始化的时候去执行线程池,线程池的任务是是不断地从队列中获取订单信息,然后去创建订单。
创建订单先获取锁,再判断一人一单,减库存,最后执行添加订单方法
完整代码:
/**
* <p>
* 服务实现类
* </p>
*
* @author 虎哥
* @since 2021-12-22
*/
@Slf4j
@Service
public class VoucherOrderServiceImpl extends ServiceImpl<VoucherOrderMapper, VoucherOrder> implements IVoucherOrderService {
@Autowired
private ISeckillVoucherService seckillVoucherService;
@Autowired
private RedisIdWorker redisIdWorker;
@Autowired
private StringRedisTemplate stringRedisTemplate;
@Autowired
private RedissonClient redissonClient;
private IVoucherOrderService proxy;
private static final DefaultRedisScript<Long> SECKILL_SCRIPT;
//初始化脚本
static {
SECKILL_SCRIPT = new DefaultRedisScript();
//读取文件位置,classpath就是resource
SECKILL_SCRIPT.setLocation(new ClassPathResource("seckill.lua"));
SECKILL_SCRIPT.setResultType(Long.class);
}
//阻塞队列
private BlockingQueue<VoucherOrder> orderTasks = new ArrayBlockingQueue<>(1024 * 1024);
//线程池
private static final ExecutorService SECKILL_ORDER_EXECUTOR = Executors.newSingleThreadExecutor();
//类加载完就执行
@PostConstruct
private void init() {
SECKILL_ORDER_EXECUTOR.submit(new VoucherOrderHandler());
}
class VoucherOrderHandler implements Runnable {
@Override
public void run() {
while (true) {
try {
//1.获取队列中的订单信息
VoucherOrder voucherOrder = orderTasks.take();
//2.创建订单
handleVoucherOrder(voucherOrder);
} catch (Exception e) {
log.error("订单异常:" + e);
}
}
}
}
private void handleVoucherOrder(VoucherOrder voucherOrder) {
Long userId = voucherOrder.getUserId();
//创建锁对象
RLock redisLock = redissonClient.getLock("lock:order:" + userId);
//获取锁方法
boolean lockFlag = redisLock.tryLock();
//判断是否获取成功
if (!lockFlag) {
log.error("不要重复下单");
return;
}
//事务方法执行起来可能会出现异常,但最后都要释放锁,所以try-catch起来
try {
proxy.createVoucherOrder(voucherOrder);
} finally {
redisLock.unlock();
}
}
@Override
public Result seckillVoucher(Long voucherId) {
// 获取用户id
Long userId = UserHolder.getUser().getId();
// 1. 调用lua脚本
//不需要传key,所以传个空集合
Long result = stringRedisTemplate.execute(
SECKILL_SCRIPT,
Collections.emptyList(),
voucherId.toString(), userId.toString()
);
int intValue = result.intValue();
//2. 判断结果是否为0
if (intValue != 0) {
return Result.fail(intValue == 1 ? "库存不足" : "不要重复下单");
}
//3. 将优惠券id,用户id、订单id保存在阻塞队列中
//3.1.创建订单
VoucherOrder order = new VoucherOrder();
//3.1.1 订单id
long orderId = redisIdWorker.nextId("order");
order.setId(orderId);
//3.1.2 用户id
order.setUserId(userId);
//3.1.3 优惠券id
order.setVoucherId(voucherId);
//3.2 放入阻塞队列中
orderTasks.add(order);
//3.3 获取当前的代理对象(事物)
proxy = (IVoucherOrderService) AopContext.currentProxy();
//4. 返回订单id
return Result.ok(orderId);
}
@Transactional
public void createVoucherOrder(VoucherOrder voucherOrder) {
//5.一人一单
Long userId = voucherOrder.getUserId();
//查询
Integer count = query().eq("user_id", userId).eq("voucher_id", voucherOrder.getVoucherId()).count();
//判断订单是否存在
if (count > 0) {
log.error("用户已经购买过一次");
return;
}
//6.扣减库存
boolean success = seckillVoucherService.update().setSql("stock = stock-1")
.eq("voucher_id", voucherOrder.getVoucherId())
.gt("stock", 0)
.update();
if (!success) {
log.error("不要重复下单");
return;
}
save(voucherOrder);
}
}
测试:
现在我们的阻塞队列使用的是jvm的内存,将来有无数的请求进来,内存可能会满,而且如果服务重启或者宕机,订单信息就消失了,可能会出现一些问题,接下来学习一下redis的消息队列
3. Redis的消息队列
消息队列,字面意思就是存放消息的队列,最简单的消息队列包含三个角色
- 生产者:发送消息到消息队列
- 消息队列:存储和管理消息,也称为消息代理
- 消费者:从消息队列获取消息并处理消息
市面上有很多的消息队列的产品,但是搭建他们也是需要成本的,既然我们已经搭建起了redis集群,为了减少成本,可以使用redis提供的三种不同的方式
- list结构:基于list结构模拟消息队列
- PubSub:订阅发布,基本的点对点消息模型
- Stream:比较完善的消息队列模型
3.1 基于list实现消息队列
Redis的list数据结构是一个双向链表,很容易模拟出队列效果。
队列是入口和出口不在一边,因此我们可以利用: LPUSH 结合 RPOP、或者 RPUSH 结合LPOP来实现。
不过要注意的是,当队列中没有消息时RPOP或LPOP操作会返回null,并不像VM的阻塞队列那样会阻塞并等待消息。因此这里应该使用BRPOP或者BLPOP来实现阻塞效果。
优点:
- 利用Redis存储,不受限于JVM内存上限
- 基于Redis的持久化机制,数据安全性有保证
- 可以满足消息有序性
缺点:
- 无法避免消息丢失
- 只支持单消费者
3.2 基于PubSub实现消息队列
Pubsub(发布订阅)是Redis2.0版本引入的消息传递模型。顾名思义,消费者可以订阅一个或多个channel,生产者向对应channel发送消息后,所有订阅者都能收到相关消息
- SUBSCRIBE channel[channell : 订阅一个或多个频道
- PUBLISH channel msg : 向一个频道发送消息
- PSUBSCRIBE pattern[pattern]: 订与pattern格式匹配的所有频道
优点:
- 采用发布订阅模型,支持多生产、多消费
缺点:
- 不支持数据持久化
- 无法避免消息丢失
- 消息堆积有上限,超出时数据丢失
3.3 基于stream实现消息队列
3.3.1 stream的单消费模式
Stream是Redis5.0引入的一种新数据类型,可以实现一个功能非常完善的消息队列
- 发送消息的命令:
示例:
xadd users * name zhuyi love lvhan
返回id
- 查看队列中消息数量的命令
xlen key
示例:
- 读取队列中消息的命令
- [COUNT count] 是每次读取消息的最大数量
- [BLOCK milliseconds] 当没消息时,是否阻塞及阻塞时长
- STREAMS key 要从哪个队列开始读,key就是队列名
- ID 起始id,只返回大于该id的消息。0:代表从第一个消息开始,$:代表从最新的消息开始
示例:
从第一个消息开始读
因为消息已经读过,没有最新的消息,所以读不出来
特点:
- 消息可回溯
- 一个消息可以被多个消费者读取
- 可以阻塞读取
- 有消息漏读的风险。使用$,在读一条消息的时候,有超过一条以上的消息进入队列,只会读取最后一条
3.3.2 stream的消费者组模式
消费者组,将多个消费者划分到一个组中,监听同一个队列,有以下好处:
- 消息分流:队列中的消息会分流给组内的不同消费者,而不是重复消费,从而加快消息处理的速度。如果想让一个消息被多个消费者消费,可以多加几个组
- 消息标识:消费者组会维护一个标识,记录最后一个被处理的消息,哪怕消费者宕机重启,还会从标识之后读取消息,确保每一个消息都会被消费
- 消息确认:消费者获取消息后,消息处于pending(待处理)状态,并存入一个pending-list。当处理完成后需要通过XACK来确认消息,标记消息为已处理,才会从pending-list移除,保证所有的消息只要被获取到了,就能至少被消费一次
- 创建消费者组:
- key:队列名称
- groupName:消费者组名称
- ID:起始ID标识,$代表队列中最新的消息,0则代表队列中第一个消息
- MKSTREAM:队列不存在时自动创建队列
其他常见命令:
示例:
- 从消费者组读取消息
- group:消费组名称
- consumer:消费者名称,如果消费者不存在,会自动创建一个消费者
- count:本次查询的最大数量
- BLOCK millisecond:当没有消息时最长等待时间
- NOACK:无需手动ACK,获取到消息后自动确认
- STREAMS key:指定队列名称
- ID:获取消息的起始id:1. “>”:从下一个未消费的消息开始。2. 其他:根据指定id从pending-list中获取已消费但未确认的消息,例如0,是从pending-list中的第一个消息开始
示例:
- 确认消息
我们获取到消息消费后,一定要确认它,把他从pending-list中移除
- key:队列名称
- group:消费组名称
- ID:获取消息的起始id:1. “>”:从下一个未消费的消息开始。2. 其他:根据指定id从pending-list中获取已消费但未确认的消息,例如0,是从pending-list中的第一个消息开始
示例:
消费者监听消息的基本思路:
特点:
- 消息可回溯
- 可以多消费者争抢消息,加快消费速度
- 可以阻塞读取
- 没有消息漏读的风险
- 有消息确认机制,保证消息至少被消费一次
如果你的公司业务比较庞大,对消息队列要求比较严格,还是建议使用更专业的消息队列,如rabbitmq等,但如果是中小型公司,对消息队列需要没那么大,redis的stream就已经能满足需求了
3.4 基于stream消息队列实现异步秒杀
- 创建一个Stream类型的消息队列,定义为stream.orders,这里直接在客户端完成了
XGROUP CREATE stream.orders g1 0 mkstream
- 修改之前的秒杀下单lua脚本,在认定有抢购资格后,直接向stream.orders中添加消息,内容包括voucherId、userId、orderId
修改lua脚本:
主要添加了一个订单id的参数,在业务的最后向队列发送消息
-- 1. 参数列表
-- 1.1 优惠券id
local voucherId = ARGV[1]
-- 1.2 用户id
local userId = ARGV[2]
-- 1.3 订单id
local orderId = ARGV[3]
-- 2. key
-- 2.1 库存key
local stockKey = 'seckill:stock:' .. voucherId
-- 2.2 订单key
local orderKey = 'seckill:order:' .. voucherId
-- 3. 脚本业务
-- 判断库存是否充足
if (tonumber(redis.call('get',stockKey)) <= 0) then
-- 库存不足
return 1
end
-- 判断用户是否下单
if (redis.call('sismember',orderKey,userId) == 1) then
-- 用户下过单
return 2
end
-- 扣减库存
redis.call('incrby',stockKey,-1);
-- 下单,保存用户id到set集合
redis.call('sadd',orderKey,userId);
-- 发送消息
redis.call('xadd','stream.orders','*','voucherId',voucherId,'userId',userId,'id',orderId)
return 0
修改一下调用lua脚本的逻辑:
新添加一个订单id的参数
// 1. 调用lua脚本
//不需要传key,所以传个空集合
Long result = stringRedisTemplate.execute(
SECKILL_SCRIPT,
Collections.emptyList(),
voucherId.toString(), userId.toString(),String.valueOf(orderId)
);
int intValue = result.intValue();
- 项目启动时,开启一个线程任务,尝试获取stream.orders中的消息,完成下单
在原有的线程任务逻辑上修改,从消息队列中获取订单信息,判断一下订单信息是否为空,如果为空,说明没有消息,继续下一次循环,如果有,去解析数据,拿到订单,通过以前写过的createVoucherOrder()方法来创建订单,最后一定要确认消息,将消息从pending-list中移除。
如果在执行时出现了错误或者服务宕机,通过handlePendingList()方法处理pending-list中已消费但未确认的订单,这里如果出现异常,就不用再调用这个方法了
private class VoucherOrderHandler implements Runnable {
@Override
public void run() {
while (true) {
try {
// 1.获取消息队列中的订单信息 XREADGROUP GROUP g1 c1 COUNT 1 BLOCK 2000 STREAMS s1 >
List<MapRecord<String, Object, Object>> list = stringRedisTemplate.opsForStream().read(
Consumer.from("g1", "c1"),
StreamReadOptions.empty().count(1).block(Duration.ofSeconds(2)),
StreamOffset.create("stream.orders", ReadOffset.lastConsumed())
);
// 2.判断订单信息是否为空
if (list == null || list.isEmpty()) {
// 如果为null,说明没有消息,继续下一次循环
continue;
}
// 解析数据
MapRecord<String, Object, Object> record = list.get(0);
Map<Object, Object> value = record.getValue();
VoucherOrder voucherOrder = BeanUtil.fillBeanWithMap(value, new VoucherOrder(), true);
// 3.创建订单
handleVoucherOrder(voucherOrder);
// 4.确认消息 XACK
stringRedisTemplate.opsForStream().acknowledge("stream.orders", "g1", record.getId());
} catch (Exception e) {
log.error("处理订单异常", e);
handlePendingList();
}
}
}
private void handlePendingList() {
while (true) {
try {
// 1.获取pending-list中的订单信息 XREADGROUP GROUP g1 c1 COUNT 1 BLOCK 2000 STREAMS s1 0
List<MapRecord<String, Object, Object>> list = stringRedisTemplate.opsForStream().read(
Consumer.from("g1", "c1"),
StreamReadOptions.empty().count(1),
StreamOffset.create("stream.orders", ReadOffset.from("0"))
);
// 2.判断订单信息是否为空
if (list == null || list.isEmpty()) {
// 如果为null,说明没有异常消息,结束循环
break;
}
// 解析数据
MapRecord<String, Object, Object> record = list.get(0);
Map<Object, Object> value = record.getValue();
VoucherOrder voucherOrder = BeanUtil.fillBeanWithMap(value, new VoucherOrder(), true);
// 3.创建订单
handleVoucherOrder(voucherOrder);
// 4.确认消息 XACK
stringRedisTemplate.opsForStream().acknowledge("stream.orders", "g1", record.getId());
} catch (Exception e) {
log.error("处理订单异常", e);
}
}
}
}
XACK确认消息:stringRedisTemplate.opsForStream().acknowledge("s1", "g1", record.getId());