DelayQueue实现延时消息队列(简易版MQ)

1、需求案例

只要涉及到电商支付相关的订单未支付超时后自动取消订单的操作。我们在开发过程中实现该功能也有很多,例如 消息中间件、定时任务等,每种方法都有各自的优点。这里我使用java DelayQueue容器来实现,优点是本地存储,系统资源消耗低,缺点是单机模式。这种案列还有很多,比如延迟发送消息 考试系统:学生参与考试,不能提前30分钟交卷或者开考前30分钟不能交卷等等。

2、代码实现

1、编写DelayedOrderVo实现类

/**
 * @author Wu_启龙
 * @version 1.0
 * @date 2020/9/4 10:45
 */
@Data
@Accessors(chain = true)
@NoArgsConstructor
public class DelayedOrderVo implements Delayed {

    //设备序列号
    private String orderId;

    //设备下线的过期时间
    private long expire;

    //设备过期时间间隔定义(毫秒),这里方便测试暂时设置为10000毫秒(也就是10秒)
    public static final long expireTime = TimeUnit.MILLISECONDS.toMillis(10000);

    public DelayedOrderVo(String orderId, LocalDateTime serialTime) {
        this.orderId= orderId;
        //转成毫秒
        long creatTime = serialTime.toInstant(ZoneOffset.of("+8")).toEpochMilli();
        this.expire = expireTime + creatTime;
    }


    /**
     * 获取剩余时间
     *
     * @param unit
     * @return
     */
    @Override
    public long getDelay(TimeUnit unit) {
        return unit.convert(expire - System.currentTimeMillis(), TimeUnit.MILLISECONDS);
    }

    @Override
    public int compareTo(Delayed other) {
        return (int) (this.getDelay(TimeUnit.MILLISECONDS) - other.getDelay(TimeUnit.MILLISECONDS));
    }
}

2、编写DelayQueue业务类

/**
 * @author Wu_启龙
 * @version 1.0
 * @date 2020/9/4 11:19
 */

@Service
@Slf4j
public class OrderCanelService {
    @Resource
    private OrderService orderService;  //订单增删改查业务逻辑

    //用于存放需要未支付计时订单
    private final static DelayQueue<DelayedOrderVo> delayQueue = new DelayQueue<>();


    //订单取消,数据库改变订单状态,DelayQueue容器移除该订单记录
    public void cancelOrder(String orderNo) {
        //数据库改变订单状态
        orderService.cancelOrder(orderNo);
        //容器遍历找到对应的订单记录,并从容器中移除
        Iterator val = delayQueue.iterator();
        while (val.hasNext()) {
            OrderAutoEntity orderAutoEntity = (OrderAutoEntity) val.next();
            if(orderAutoEntity.getOrderNo().equals(orderNo)){
                delayQueue.remove(orderAutoEntity);
            }
        }
    }

    //往队列中新增订单记录
    public void add(OrderAutoEntity orderAutoEntity) {
        delayQueue.put(orderAutoEntity);
    }

    /**
     * 服务器启动时,自动加载待支付订单
     */
    @PostConstruct
    public void initOrderStatus() {
        log.info(">>>>>>>>>>> 系统启动时,加载所有待支付订单到延时队列 >>>>>>>>>>>>.");
        //未支付订单查询
        QueryWrapper<Order> wrapper = new QueryWrapper();
        wrapper.select("order_no", "create_time").eq("status", "0");
        //获取所有未支付订单,这里用了mybatisplus操作数据库
        List<Map<String, Object>> orders = orderService.listMaps(wrapper);

        //逐个构造Delay的实现类,添加进容器
        for (Map<String, Object> order : orders) {
            Timestamp createTime = (Timestamp) order.get("create_time");
            OrderAutoEntity orderAutoEntity = new OrderAutoEntity((String) order.get("order_no"), createTime.toLocalDateTime());
            delayQueue.add(orderAutoEntity);
        }

        //启动一个线程持续健康订单超时
        Executors.newSingleThreadExecutor().execute(() -> {
            try {
                while (true) {
                    if (delayQueue.size() > 0) {
                        //容器中超时的订单记录会被取出
                        OrderAutoEntity order = delayQueue.take();
                        //修改数据库,容器中移除数据
                        cancelOrder(order.getOrderNo());
                    }
                }
            } catch (InterruptedException e) {
                log.error("InterruptedException error:", e);
            }
        });
    }
}

3、编写订单业务逻辑

需要注意的是:我们前台每次新增订单时,也需要再给容器中添加一条记录。(这里提供一些伪代码,具体的根据自己的业务场景去进行编写)

@RestController
@Slf4j
@RequestMapping("/api")
public class OrderController {
    @Resource
    private OrderCanelService orderCanelService ;
    ...
    //前端页面调用的订单新增接口
    @PostMapping("/order")
    @Transactional(rollbackFor = Exception.class)
    public ResponseEntity addOrder(@RequestBody Map<String, Object> params) {
        ...
            //构造订单延时类(OrderAutoEntity),这里是伪代码
            OrderAutoEntity orderAutoEntity = new OrderAutoEntity(orderNo, createTime);
            orderAutoService.add(orderAutoEntity);
        ...
    }
    ...
}

总结说明

我这种操作只支持单机情况,一般还可以进一步优化,利用redis:在新增订单时,除了存到数据库,再保留一份到redis中,这样我们在OrderCanelService 的初始化函数initOrderStatus中就不需要查数据库,每次启动就直接读redis中的数据。这里我就没做这么复杂,我们具体的项目中也没用DelayQueue啦,而是用的mq,如果项目中没有使用mq的小伙伴,而正好需要做这个功能的时候呢,则可以参考我这样的写法。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 202,905评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,140评论 2 379
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 149,791评论 0 335
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,483评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,476评论 5 364
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,516评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,905评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,560评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,778评论 1 296
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,557评论 2 319
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,635评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,338评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,925评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,898评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,142评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,818评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,347评论 2 342