延迟队列DelayQueue

最新推荐文章于 2024-10-03 10:27:34 发布

默剧y

最新推荐文章于 2024-10-03 10:27:34 发布

阅读量356

点赞数 8

文章标签： java 前端算法

本文链接：https://blog.csdn.net/falao11/article/details/135550551

版权

本文介绍了如何在Java中使用Slf4j日志框架配合DelayQueue实现一个简单的定时任务队列，通过`DelayedTask`类封装任务并设置到期时间，`DelayQueue`用于按到期顺序执行任务。

摘要由CSDN通过智能技术生成

示例代码

@Slf4j
public class DelayQueueTest {

    public static  class DelayedTask implements Delayed {

        /**
         * 消息体
         * */
        private Object data;

        /**
         * 到期时间，毫秒
         * */
        private Long deadline;

        public DelayedTask(Object data, int expireSeconds){
            this.data = data;
            this.deadline = System.currentTimeMillis() + expireSeconds * 1000;
        }

        /**
         * 获取元素的剩余的有效期
         * */
        @Override
        public long getDelay(TimeUnit unit) {
            long remain = deadline - System.currentTimeMillis();
            if(remain <= 0){
                remain = 0;
            }
            return unit.convert(remain, TimeUnit.MILLISECONDS);
        }

        /**
         * 比较元素的剩余，看谁先到期
         * */
        @Override
        public int compareTo(Delayed o) {
            long result = this.getDelay(TimeUnit.MILLISECONDS) - o.getDelay(TimeUnit.MILLISECONDS);
            if(result > 0){
                return 1;
            }else if(result < 0){
                return -1;
            }else{
                return 0;
            }
        }
    }

    public static void main(String[] args) throws InterruptedException {
        DelayQueue<DelayedTask> delayQueue = new DelayQueue<>();
        DelayedTask task1 = new DelayedTask("data 5 second", 5);
        DelayedTask task2 = new DelayedTask("data 3 second", 3);
        DelayedTask task3 = new DelayedTask("data 1 second", 1);
        // 向延迟队列放任务
        delayQueue.put(task1);
        delayQueue.put(task2);
        delayQueue.put(task3);
        log.info("消息发送完成");
        // 如果队列中没有到期的元素，take会阻塞等待
        while(true){
            DelayedTask task = delayQueue.take();
            log.info("获取延迟任务：task：{}", task.data);
        }
    }
}