利用 ScheduledThreadPoolExecutor 实现简单定时/延时队列


import java.time.LocalTime;
import java.util.concurrent.ScheduledThreadPoolExecutor;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;


public class MyScheduleUtils {
    static class TaskThreadFactory implements ThreadFactory {
        private static final AtomicInteger threadNumber = new AtomicInteger(1);
        private final String namePrefix;
        public TaskThreadFactory(String namePrefix) {
            this.namePrefix = namePrefix + "-";
        }

        @Override
        public Thread newThread(Runnable r) {
            Thread t = new Thread(Thread.currentThread().getThreadGroup(), r, this.namePrefix + threadNumber.getAndIncrement());
            t.setDaemon(true);
            t.setPriority(Thread.NORM_PRIORITY);
            return t;
        }
    }

    public static ScheduledThreadPoolExecutor scheduleAtFixedRate(Runnable runnable, long delaySecs, long spanSecs, String taskName) {
        if (spanSecs < 0) {
            throw new IllegalArgumentException("spanSecs[" + spanSecs + "] must >0");
        }
        if (delaySecs <= 0) {
            delaySecs = spanSecs;
        }
        ScheduledThreadPoolExecutor task = new ScheduledThreadPoolExecutor(1, new TaskThreadFactory(taskName));
        // 第二个参数为首次执行的延时时间,第三个参数为定时执行的间隔时间  
        task.scheduleAtFixedRate(runnable, delaySecs, spanSecs, TimeUnit.SECONDS);
        return task;
    }

    /**
     * 功能描述: 间隔定时执行
     */
    public static ScheduledThreadPoolExecutor scheduleAtFixedRate(Runnable runnable, long spanSecs, String taskName) {
        return scheduleAtFixedRate(runnable, spanSecs, spanSecs, taskName);
    }

    /**
     * 功能描述: 每天定时执行
     */
    public static ScheduledThreadPoolExecutor scheduleEveryDayAtTime(Runnable runnable, LocalTime time, String taskName) {
        long secs = time.toSecondOfDay() - LocalTime.now().toSecondOfDay();
        if (secs < 0) {
            secs += TimeUnit.DAYS.toSeconds(1);
        }
        return scheduleAtFixedRate(runnable, secs, TimeUnit.DAYS.toSeconds(1), taskName);
    }
}

调用的时候可以使用定时运行:

ScheduledThreadPoolExecutor queue = MyScheduleUtils.scheduleAtFixedRate(() -> 具体的方法名, 延时时间,秒, "队列名称");

或者定时运行(例:凌晨两点运行):

ScheduledThreadPoolExecutor task = AppScheduleUtils.scheduleEveryDayAtTime(
          () -> ApplicationContextHolder.getContext().getBean(xxx.class).方法(),LocalTime.of(2,0,0),"task");

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值