快捷導(dǎo)航

java使用DelayQueue實(shí)現(xiàn)延時(shí)任務(wù)

更新時(shí)間：2023年10月18日 16:54:28 作者：weixin_44203609

項(xiàng)目中經(jīng)常會(huì)用到類似一些需要延遲執(zhí)行的功能,比如緩存,java提供了DelayQueue來很輕松的實(shí)現(xiàn)這種功能,下面小編就來和大家介紹一下如何使用DelayQueue實(shí)現(xiàn)延時(shí)任務(wù)吧

1、背景

項(xiàng)目中經(jīng)常會(huì)用到類似一些需要延遲執(zhí)行的功能，比如緩存。java提供了DelayQueue來很輕松的實(shí)現(xiàn)這種功能。Delayed接口中的getDelay方法返回值小于等于0的時(shí)候，表示時(shí)間到達(dá)，可以從DelayQueue中通過take()方法取的到期的對(duì)象。到期對(duì)象是實(shí)現(xiàn)了Delayed的類。

2、demo

2.1 依賴配置

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <!--非必須-->
        <dependency>
            <groupId>cn.hutool</groupId>
            <artifactId>hutool-all</artifactId>
            <version>5.8.0.M1</version>
        </dependency>
        <!--非必須-->
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <version>1.18.22</version>
        </dependency>
 
        <!--非必須-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
        </dependency>

2.2 整體架構(gòu)

** 工具類:**

執(zhí)行任務(wù)所需的基礎(chǔ)參數(shù)

import lombok.Data;
@Data
public class TaskBase {
    //任務(wù)參數(shù)，根據(jù)業(yè)務(wù)需求多少都行
    private String identifier;
    public TaskBase(String identifier) {
        this.identifier = identifier;
    }
}

執(zhí)行的任務(wù)和時(shí)間

import cn.hutool.core.date.DateUtil;
 
import java.util.Date;
import java.util.concurrent.Delayed;
import java.util.concurrent.TimeUnit;
/**
 * 延時(shí)任務(wù)
 */
public class DelayTask implements Delayed {
    //任務(wù)參數(shù)
    final private TaskBase data;
    //任務(wù)的延時(shí)時(shí)間，單位毫秒
    final private long expire;
 
    /**
     * 構(gòu)造延時(shí)任務(wù)
     *
     * @param data   業(yè)務(wù)數(shù)據(jù)
     * @param expire 任務(wù)延時(shí)時(shí)間（ms）
     */
    public DelayTask(TaskBase data, long expire) {
        super();
        this.data = data;
        this.expire = expire + System.currentTimeMillis();
    }
 
    public TaskBase getData() {
        return data;
    }
 
    public long getExpire() {
        return expire;
    }
 
    @Override
    public boolean equals(Object obj) {
        if (obj instanceof DelayTask) {
            return this.data.getIdentifier().equals(((DelayTask) obj).getData().getIdentifier());
        }
        return false;
    }
 
    @Override
    public String toString() {
        return "{" + "data:" + data.toString() + "," + "延時(shí)時(shí)間:" +expire+ DateUtil.format(new Date(),"yyyy.MM.dd HH:mm:ss") + "}";
    }
 
    @Override
    public long getDelay(TimeUnit unit) {
        return unit.convert(this.expire - System.currentTimeMillis(), unit);
    }
 
    @Override
    public int compareTo(Delayed o) {
        long delta = getDelay(TimeUnit.NANOSECONDS) - o.getDelay(TimeUnit.NANOSECONDS);
        return (int) delta;
    }
}

** 任務(wù)管理器：**

import lombok.extern.slf4j.Slf4j;
import org.springframework.boot.CommandLineRunner;
import org.springframework.stereotype.Component;
import java.util.concurrent.DelayQueue;
import java.util.concurrent.Executors;
 
@Component
@Slf4j
public class DelayQueueManager implements CommandLineRunner {
 
    private final DelayQueue<DelayTask> delayQueue = new DelayQueue<>();
    private final Map<String, DelayTask> elements = new HashMap<>();
 
 
    /**
     * 加入到延時(shí)隊(duì)列中
     *
     * @param task
     */
    public void put(DelayTask task) {
        log.error("加入延時(shí)任務(wù)：{}", task);
        delayQueue.put(task);
    }
 /**
     * 查詢延時(shí)任務(wù)
     * @param taskID
     * @return
     */
    public DelayTask query(String taskID) {
        return elements.get(taskID);
    }
    /**
     * 取消延時(shí)任務(wù)
     *
     * @param task
     * @return
     */
    public boolean remove(DelayTask task) {
        log.error("取消延時(shí)任務(wù)：{}", task);
        return delayQueue.remove(task);
    }
 
    /**
     * 取消延時(shí)任務(wù)
     *
     * @param taskid
     * @return
     */
    public boolean remove(String taskid) {
        return remove(new DelayTask(new TaskBase(taskid), 0));
    }
 
    @Override
    public void run(String... args) throws Exception {
        log.info("初始化延時(shí)隊(duì)列");
        Executors.newSingleThreadExecutor().execute(new Thread(this::excuteThread));
    }
 
    /**
     * 延時(shí)任務(wù)執(zhí)行線程
     */
    private void excuteThread() {
        while (true) {
            try {
                DelayTask task = delayQueue.take();
                //執(zhí)行任務(wù)
                processTask(task);
            } catch (InterruptedException e) {
                break;
            }
        }
    }
 
    /**
     * 內(nèi)部執(zhí)行延時(shí)任務(wù)
     *
     * @param task
     */
    private void processTask(DelayTask task) {
        //獲取任務(wù)參數(shù)，執(zhí)行業(yè)務(wù)task.getData().getIdentifier()
        log.error("執(zhí)行延時(shí)任務(wù)：{}-{}", task, task.getData().getIdentifier());
    }
}

2.3 進(jìn)行測(cè)試

import com.example.demo.task.DelayQueueManager;
import com.example.demo.task.DelayTask;
import com.example.demo.task.TaskBase;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
 
@SpringBootTest
class DemoApplicationTests {
    @Autowired
    private DelayQueueManager delayQueueManager;
 
    @Test
    void contextLoads() throws InterruptedException {
        //新增任務(wù)
        delayQueueManager.put(new DelayTask(new TaskBase("abc"), 1000 * 1));
        //新增任務(wù)
        delayQueueManager.put(new DelayTask(new TaskBase("abc"), 1000 * 5));
        //新增任務(wù)
        delayQueueManager.put(new DelayTask(new TaskBase("abc"), 1000 * 6));
        //測(cè)試任務(wù)需要下邊代碼執(zhí)行，線上不用
        Thread.sleep(10 * 1000);
    }
 
}

到此這篇關(guān)于java使用DelayQueue實(shí)現(xiàn)延時(shí)任務(wù)的文章就介紹到這了,更多相關(guān)DelayQueue延時(shí)任務(wù)內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: