微信抢红包过期失效实战案例
前言
微信红包业务,发红包之后如果24小时之内没有被领取完就自动过期失效。
架构设计
业务流程
-
老板发红包,此时缓存初始化红包个数,红包金额(单位分),并异步入库。
-
红包数据入延迟队列,唯一标识+失效时间
-
红包数据出延迟队列,根据唯一标识清空红包缓存数据、异步更新数据库、异步退回红包金额
代码案例
这里我们使用Java
内置的DelayQueue
来实现,DelayQueue
是一个无界的BlockingQueue
,用于放置实现了Delayed
接口的对象,其中的对象只能在其到期时才能从队列中取走。这种队列是有序的,即队头对象的延迟到期时间最长。
老板发了10个红包一共200人民币,假装只有9个人抢红包。
发红包,缓存数据进入延迟队列:
/**
* 有人没抢 红包发多了
* 红包进入延迟队列
* 实现过期失效
* @param redPacketId
* @return
*/
@ApiOperation(value="抢红包三",nickname="爪哇笔记")
@PostMapping("/startThree")
public Result startThree(long redPacketId){
int skillNum = 9;
final CountDownLatch latch = new CountDownLatch(skillNum);//N个抢红包
/**
* 初始化红包数据,抢红包拦截
*/
redisUtil.cacheValue(redPacketId+"-num",10);
/**
* 初始化红包金额,单位为分
*/
redisUtil.cacheValue(redPacketId+"-money",20000);
/**
* 加入延迟队列 24s秒过期
*/
RedPacketMessage message = new RedPacketMessage(redPacketId,24);
RedPacketQueue.getQueue().produce(message);
/**
* 模拟 9个用户抢10个红包
*/
for(int i=1;i<=skillNum;i++){
int userId = i;
Runnable task = () -> {
/**
* 抢红包 判断剩余金额
*/
Integer money = (Integer) redisUtil.getValue(redPacketId+"-money");
if(money>0){
Result result = redPacketService.startTwoSeckil(redPacketId,userId);
if(result.get("code").toString().equals("500")){
LOGGER.info("用户{}手慢了,红包派完了",userId);
}else{
Double amount = DoubleUtil.divide(Double.parseDouble(result.get("msg").toString()), (double) 100);
LOGGER.info("用户{}抢红包成功,金额:{}", userId,amount);
}
}
latch.countDown();
};
executor.execute(task);
}
try {
latch.await();
Integer restMoney = Integer.parseInt(redisUtil.getValue(redPacketId+"-money").toString());
LOGGER.info("剩余金额:{}",restMoney);
} catch (InterruptedException e) {
e.printStackTrace();
}
return Result.ok();
}
红包队列消息:
/**
* 红包队列消息
*/
public class RedPacketMessage implements Delayed {
private static final DateTimeFormatter F = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
/**
* 默认延迟3秒
*/
private static final long DELAY_MS = 1000L * 3;
/**
* 红包 ID
*/
private final long redPacketId;
/**
* 创建时间戳
*/
private final long timestamp;
/**
* 过期时间
*/
private final long expire;
/**
* 描述信息
*/
private final String description;
public RedPacketMessage(long redPacketId, long expireSeconds) {
this.redPacketId = redPacketId;
this.timestamp = System.currentTimeMillis();
this.expire = this.timestamp + expireSeconds * 1000L;
this.description = String.format("红包[%s]-创建时间为:%s,超时时间为:%s", redPacketId,
LocalDateTime.ofInstant(Instant.ofEpochMilli(timestamp), ZoneId.systemDefault()).format(F),
LocalDateTime.ofInstant(Instant.ofEpochMilli(expire), ZoneId.systemDefault()).format(F));
}
public RedPacketMessage(long redPacketId) {
this.redPacketId = redPacketId;
this.timestamp = System.currentTimeMillis();
this.expire = this.timestamp + DELAY_MS;
this.description = String.format("红包[%s]-创建时间为:%s,超时时间为:%s", redPacketId,
LocalDateTime.ofInstant(Instant.ofEpochMilli(timestamp), ZoneId.systemDefault()).format(F),
LocalDateTime.ofInstant(Instant.ofEpochMilli(expire), ZoneId.systemDefault()).format(F));
}
public long getRedPacketId() {
return redPacketId;
}
public long getTimestamp() {
return timestamp;
}
public long getExpire() {
return expire;
}
public String getDescription() {
return description;
}
@Override
public long getDelay(TimeUnit unit) {
return unit.convert(this.expire - System.currentTimeMillis(), TimeUnit.MILLISECONDS);
}
@Override
public int compareTo(Delayed o) {
return (int) (this.getDelay(TimeUnit.MILLISECONDS) - o.getDelay(TimeUnit.MILLISECONDS));
}
}
红包延迟队列:
/**
* 红包延迟队列
*/
public class RedPacketQueue {
/** 用于多线程间下单的队列 */
private static DelayQueue<RedPacketMessage> queue = new DelayQueue<>();
/**
* 私有的默认构造子,保证外界无法直接实例化
*/
private RedPacketQueue(){}
/**
* 类级的内部类,也就是静态的成员式内部类,该内部类的实例与外部类的实例
* 没有绑定关系,而且只有被调用到才会装载,从而实现了延迟加载
*/
private static class SingletonHolder{
/**
* 静态初始化器,由JVM来保证线程安全
*/
private static RedPacketQueue queue = new RedPacketQueue();
}
//单例队列
public static RedPacketQueue getQueue(){
return SingletonHolder.queue;
}
/**
* 生产入队
* 1、执行加锁操作
* 2、把元素添加到优先级队列中
* 3、查看元素是否为队首
* 4、如果是队首的话,设置leader为空,唤醒所有等待的队列
* 5、释放锁
*/
public Boolean produce(RedPacketMessage message){
return queue.add(message);
}
/**
* 消费出队
* 1、执行加锁操作
* 2、取出优先级队列元素q的队首
* 3、如果元素q的队首/队列为空,阻塞请求
* 4、如果元素q的队首(first)不为空,获得这个元素的delay时间值
* 5、如果first的延迟delay时间值为0的话,说明该元素已经到了可以使用的时间,调用poll方法弹出该元素,跳出方法
* 6、如果first的延迟delay时间值不为0的话,释放元素first的引用,避免内存泄露
* 7、判断leader元素是否为空,不为空的话阻塞当前线程
* 8、如果leader元素为空的话,把当前线程赋值给leader元素,然后阻塞delay的时间,即等待队首到达可以出队的时间,在finally块中释放leader元素的引用
* 9、循环执行从1~8的步骤
* 10、如果leader为空并且优先级队列不为空的情况下(判断还有没有其他后续节点),调用signal通知其他的线程
* 11、执行解锁操作
*/
public RedPacketMessage consume() throws InterruptedException {
return queue.take();
}
}
红包延迟队列过期消费,监听任务:
/**
* 红包延迟队列过期消费
*/
@Component("redPacket")
public class TaskRunner implements ApplicationRunner {
private final static Logger LOGGER = LoggerFactory.getLogger(TaskRunner.class);
@Autowired
private RedisUtil redisUtil;
ExecutorService executorService = Executors.newSingleThreadExecutor(r -> {
Thread thread = new Thread(r);
thread.setName("RedPacketDelayWorker");
thread.setDaemon(true);
return thread;
});
@Override
public void run(ApplicationArguments var){
executorService.execute(() -> {
while (true) {
try {
RedPacketMessage message = RedPacketQueue.getQueue().consume();
if(message!=null){
long redPacketId = message.getRedPacketId();
LOGGER.info("红包{}过期了",redPacketId);
/**
* 获取剩余红包个数以及金额
*/
int num = (int) redisUtil.getValue(redPacketId+"-num");
int restMoney = (int) redisUtil.getValue(redPacketId+"-money");
LOGGER.info("剩余红包个数{},剩余红包金额{}",num,restMoney);
/**
* 清空红包数据
*/
redisUtil.removeValue(redPacketId+"-num");
redisUtil.removeValue(redPacketId+"-money");
/**
* 异步更新数据库、异步退回红包金额
*/
}
} catch (InterruptedException e) {
e.printStackTrace();
}
}
});
}
}
适用场景
淘宝订单到期,下单成功后60s之后给用户发送短信通知,限时支付、缓存系统等等。
演示
在 Application
中有接口演示说明,你可以在抢红包 Red Packet Controller
接口中输入任何参数进行测试,也可以配合数据库稍加修改即可作为生产环境的抢红包功能模块。
源码
作者: 小柒
出处: https://blog.52itstyle.vip
分享是快乐的,也见证了个人成长历程,文章大多都是工作经验总结以及平时学习积累,基于自身认知不足之处在所难免,也请大家指正,共同进步。
本文版权归作者所有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出, 如有问题, 可邮件(345849402@qq.com)咨询。