延时任务代码保存

1 TaskBase

package com.lzkj.admin.infrastructure.util;

import com.lzkj.admin.domain.testOrder.TestFullDataPlus;
import com.lzkj.admin.domain.testOrder.TestPopDataPlus;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

@Data
@AllArgsConstructor
@NoArgsConstructor
public class TaskBase {

    //任务参数TestFullDataPlus
    private TestFullDataPlus testFullDataPlus;

    //任务参数TestFullDataPlus
    private TestPopDataPlus testPopDataPlus;




    public TaskBase(TestFullDataPlus testFullDataPlus) {
        this.testFullDataPlus = testFullDataPlus;
    }

    public TaskBase(TestPopDataPlus testPopDataPlus) {
        this.testPopDataPlus = testPopDataPlus;
    }
}

2 DelayTask

package com.lzkj.admin.infrastructure.util;

import java.util.concurrent.Delayed;
import java.util.concurrent.TimeUnit;

/**
 * 延时任务
 */
public class DelayTask implements Delayed{


    //任务参数
    final private TaskBase data;
    //任务的延时时间，单位毫秒
    final private long expire;

    /**
     * 构造延时任务
     *  @param data   业务数据
     * @param expire 任务延时时间（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 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;
    }

}

3 DelayQueueManager

修改此processTask代码即可。

package com.lzkj.admin.infrastructure.util;

import cn.hutool.core.date.DateTime;
import com.lzkj.admin.domain.testOrder.TestFullDataPlus;
import com.lzkj.admin.service.common.TestOrderCommonService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.boot.CommandLineRunner;
import java.util.concurrent.DelayQueue;
import java.util.concurrent.Executors;

@Component
@Slf4j
public class DelayQueueManager implements CommandLineRunner {




    private final DelayQueue<DelayTask> delayQueue = new DelayQueue<>();

    /**
     * 加入到延时队列中
     * @param task
     */
    public void put(DelayTask task) {
        log.error("加入延时任务：{}", task);
        delayQueue.put(task);
    }

    /**
     * 取消延时任务
     * @param task
     * @return
     */
    public boolean remove(DelayTask task) {
        log.error("取消延时任务：{}", task);
        return delayQueue.remove(task);
    }

    @Override
    public void run(String... args) throws Exception {
        log.info("初始化延时队列");
        Executors.newSingleThreadExecutor().execute(new Thread(this::excuteThread));
    }

    /**
     * 延时任务执行线程
     */
    private void excuteThread() {
        while (true) {
            try {
                DelayTask task = delayQueue.take();
                //执行任务
                processTask(task);
            } catch (InterruptedException e) {
                break;
            }
        }
    }

    /**
     * 内部执行延时任务
     * @param task
     */
    private void processTask(DelayTask task) {

        TestFullDataPlus testFullDataPlus = task.getData().getTestFullDataPlus();
        if (testFullDataPlus == null) {
            TestOrderCommonService.deletePopOrder(task.getData().getTestPopDataPlus());
        }else{
            TestOrderCommonService.deleteFullOrder(task.getData().getTestFullDataPlus());
        }
//        System.out.println("删除结束时间" + DateTime.now());
    }

}