springboot 延時任務

用途，某個時間發生后要在多久后去執行其他事

package com.xxx.zhugeyl.delayService;

import com.xxx.common.utils.DateUtils;
import com.xxx.zhugeyl.domain.Web.TaskBase;

import java.util.Date;
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 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() + "," + "延時時間:" + DateUtils.getTimestampToTime(expire,DateUtils.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;
    }
}



===================================================================================================

package com.xxx.zhugeyl.delayService;

import com.xxx.zhugeyl.domain.Web.TaskBase;
import com.xxx.zhugeyl.domain.ZTransaction;
import com.xxx.zhugeyl.service.IZTransactionService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
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<>();

    @Autowired
    private IZTransactionService transactionService;

    /**
     * 加入到延時隊列中
     * @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);
    }
    /**
     * 取消延時任務
     * @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("初始化延時隊列");
        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) {
        log.error("執行延時任務：{}-{}", task,task.getData().getIdentifier());
        ZTransaction transaction = transactionService.selectZTransactionByOrderId(task.getData().getIdentifier());
        if (transaction.getPayStatus() == 0){
            transaction.setPayStatus(3);
            transactionService.updateZTransaction(transaction);
        }

    }
}
==========================================================================================================================

package com.xxx.zhugeyl.domain.Web;

import com.alibaba.fastjson.JSON;

public class TaskBase {

    private String identifier;

    public TaskBase(String identifier) {
        this.identifier = identifier;
    }

    public String getIdentifier() {
        return identifier;
    }

    public void setIdentifier(String identifier) {
        this.identifier = identifier;
    }

    @Override
    public String toString() {
        return JSON.toJSONString(this);
    }
}
======================================================================================================================

public class DateUtils extends org.apache.commons.lang3.time.DateUtils
{

　　public static String YYYY_MM_DD_HH_MM_SS = "yyyy-MM-dd HH:mm:ss";

　　public static String getTimestampToTime(Long timeStamp,final String ts){
    　　SimpleDateFormat sdf=new SimpleDateFormat(ts);
    　　return sdf.format(new Date(Long.parseLong(String.valueOf(timeStamp))));
　　}

}

======================================================================================================================

ZTransaction 是我自己的訂單類

使用的時候

@Autowired
private DelayQueueManager delayQueueManager;

delayQueueManager.put(new DelayTask(new TaskBase(itrOrderId),1000*60*5));//新增任務