Java DelayQueue怎么实现延时任务-java教程-PHP中文网

Java DelayQueue怎么实现延时任务

WBOY

发布： 2023-05-13 16:40:06

转载

1735 人浏览过

一、DelayQueue的应用原理

DelayQueue是一个无界的BlockingQueue的实现类，用于放置实现了Delayed接口的对象，其中的对象只能在其到期时才能从队列中取走。

BlockingQueue即阻塞队列，java提供的面向多线程安全的队列数据结构，当队列内元素数量为0的时候，试图从队列内获取元素的线程将被阻塞或者抛出异常。

这里的“无界”队列，是指队列的元素数量不存在上限，队列的容量会随着元素数量的增加而扩容。

Java DelayQueue怎么实现延时任务

DelayQueue实现了BlockingQueue接口，所以具有无界、阻塞的特点，除此之外它自己的核心特点就是：

「放入该队列的延时任务对象，只要到达延时时间之后才能被取到」。

DelayQueue 不接收null元素

「DelayQueue 只接受那些实现了java.util.concurrent.Delayed接口的对象」

二、订单延时任务的实现

了解了DelayQueue的特点之后，我们就可以利用它来实现延时任务了，实现java.util.concurrent.Delayed接口。

import org.jetbrains.annotations.NotNull;

import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.concurrent.Delayed;
import java.util.concurrent.TimeUnit;

/**
 * 延时订单任务
 */
public class OrderDelayObject implements Delayed {
  private String name;
  private long delayTime;   //延时时间
  //实际业务中这里传订单信息对象，我这里只做demo，所以使用字符串了
  private String order;

  public OrderDelayObject(String name, long delayTime, String order) {
    this.name = name;
    //延时时间加上当前时间
    this.delayTime = System.currentTimeMillis() + delayTime;
    this.order = order;
  }

  //获取延时任务的倒计时时间
  @Override
  public long getDelay(TimeUnit unit) {
    long diff = delayTime - System.currentTimeMillis();
    return unit.convert(diff, TimeUnit.MILLISECONDS);
  }

  //延时任务队列，按照延时时间元素排序，实现Comparable接口
  @Override
  public int compareTo(@NotNull Delayed obj) {
    return Long.compare(this.delayTime, ((OrderDelayObject) obj).delayTime);
  }

  @Override
  public String toString() {
    Date date = new Date(delayTime);
    SimpleDateFormat sd = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");

    return "\nOrderDelayObject:{"
            + "name=" + name
            + ", time=" + sd.format(date)
            + ", order=" + order
            + "}";
  }
}

登录后复制

上文类中的order为订单信息对象，在实际的业务开发过程中应该是传递订单信息，用于取消订单业务的实现（订单30分钟不付款自动取消）。

Delayed接口继承自 Comparable接口，所以需要实现compareTo方法，用于延时任务在队列中按照“延时时间”进行排序。

getDelay方法是Delayed接口方法，实现该方法提供获取延时任务的倒计时时间

三、订单处理

首先我们需要一个容器，永久保存延时任务队列，如果是Spring开发环境我们可以这样做。

@Bean("orderDelayQueue")
public DelayQueue<OrderDelayObject> orderDelayQueue(){
    return new DelayQueue<OrderDelayObject>();
}

登录后复制

当用户下单的时候，将订单下单任务放入延时队列

@Resource
private DelayQueue<OrderDelayObject> orderDelayQueue;

//发起订单下单的时候将订单演示对象放入orderDelayQueue
orderDelayQueue.add(
        new OrderDelayObject(
                "订单延时取消任务",
                30 * 60 * 1000,    //延时30分钟
                "延时任务订单对象信息"
        )
);

登录后复制

系统内开启一个线程，不断的从队列中获取消息，获取到之后对延时消息进行处理。DelayQueue的take方法从队列中获取延时任务对象，如果队列元素数量为0，或者没有到达“延时时间的任务”，该线程会被阻塞。

@Component
public class DelayObjectConsumer  implements InitializingBean {

  @Resource
  private DelayQueue<OrderDelayObject> orderDelayQueue;

  @Override
  public void afterPropertiesSet() throws Exception {
    while (true) {
      OrderDelayObject task = orderDelayQueue.take();
      System.out.println(task.toString());
      System.out.println(task.getOrder());
      //根据order订单信息，去查询该订单的支付信息
      //如果用户没有进行支付，将订单从数据库中关闭
      //如果订单并发量比较大，这里可以考虑异步或线程池的方式进行处理
    }
  }
}

登录后复制

需要说明的是，这里的while-true循环的延时任务处理是顺序执行的，在订单并发量比较大的时候，需要考虑异步处理的方式完成订单的关闭操作。我之前写过一个SpringBoot的可观测、易配置的线程池开源项目，可能会对你有帮助，源代码地址

经过我的测试，放入orderDelayQueue的延时任务，在半小时之后得到正确的执行处理。说明我们的实现是正确的。