java 动态定时任务开启

工具类

package com.goboo.modules.job.task;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Lazy;
import org.springframework.scheduling.Trigger;
import org.springframework.scheduling.TriggerContext;
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;
import org.springframework.scheduling.support.CronTrigger;
import org.springframework.stereotype.Component;

import java.util.Date;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ScheduledFuture;
/**
 * @author lsx
 * @Classname DynamicTaskConfig
 * @Description TODO

 */
@Component
public class DynamicTaskConfig {

    @Lazy
    @Autowired
    private ThreadPoolTaskScheduler threadPoolTaskScheduler;
    /**
     *  用于存放对应的定时任务
     */
    private Map<String, ScheduledFuture<?>> taskFutures = new ConcurrentHashMap<String, ScheduledFuture<?>>();

    @Bean
    public ThreadPoolTaskScheduler threadPoolTaskScheduler() {
        return new ThreadPoolTaskScheduler();
    }

    /**
     * 开始定时任务
     * @param taskId
     * @param task
     * @param cron
     */
    public void startCron(String taskId, Runnable task, String cron) {
        //先把之前的定时任务取消
        stopCron(taskId);
        //提交任务
        ScheduledFuture<?> scheduledFuture = threadPoolTaskScheduler.schedule(task, new Trigger() {
            @Override
            public Date nextExecutionTime(TriggerContext triggerContext) {
                //设置定时时间
                CronTrigger trigger = new CronTrigger(cron);
                Date nextExecDate = trigger.nextExecutionTime(triggerContext);
                return nextExecDate;
            }
        });
        //保存任务,用于后面可以取消任务
        taskFutures.put(taskId, scheduledFuture);
    }

    /**
     * 取消定时任务
     * @param taskId
     */
    public void stopCron(String taskId) {
        ScheduledFuture<?> future = taskFutures.get(taskId);
        if (future != null) {
            future.cancel(true);
        }
    }



业务逻辑中调用:

public List<T>  executeAllTask() {
        //可执行的任务
            dynamicTaskConfig.startCron("uploadDataSets_" + dataset.getId(), new Runnable() {
                @SneakyThrows
                @Override
                public void run() {
                    //业务逻辑
                    uploadDataSet(T);
                }
            }, cron);
         
        }
        return new ArrayList<T>;
    }
}




评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值