/**
 * @author Created by zhuzqc on 2023/1/30 15:28
 */
@Slf4j
@Component
@EnableScheduling
public class ScheduleTaskDemo implements SchedulingConfigurer {

    private Logger logger = LoggerFactory.getLogger(this.getClass());

    @Override
    public void configureTasks(ScheduledTaskRegistrar taskRegistrar) {

        //Runnable線(xiàn)程注冊(cè)任務(wù)
        Runnable taskOne = () -> {
            //需要執(zhí)行的業(yè)務(wù)邏輯，一般會(huì)在這里封裝好
            logger.info("----------業(yè)務(wù)執(zhí)行結(jié)束----------");
        };
        //任務(wù)的觸發(fā)時(shí)間，一般使用 cron 表達(dá)式
        Trigger triggerOne = triggerContext -> {
            Date nextExecTime = null;
            try {
                // 此處指定 cron 表達(dá)式
                String cron = "0 00 12 ? * *";
                if (StringUtils.isBlank(cron)) {
                    // 提示參數(shù)為空
                    logger.info("trigger定時(shí)器的 cron 參數(shù)為空！");
                    // 如果為空則賦默認(rèn)值，每天中午12點(diǎn)
                    cron = "0 00 12 ? * *";
                }
                logger.info("---------->定時(shí)任務(wù)執(zhí)行中<---------");
                CronTrigger cronTrigger = new CronTrigger(cron);
                nextExecTime = cronTrigger.nextExecutionTime(triggerContext);
            } catch (Exception e) {
                e.printStackTrace();
                log.info(e.getMessage());
            }
            return nextExecTime;
        };
        taskRegistrar.addTriggerTask(taskOne, triggerOne);
    }
}

 	@Override
	@Nullable
	public ScheduledFuture<?> schedule(Runnable task, Trigger trigger) {
		ScheduledExecutorService executor = getScheduledExecutor();
		try {
			ErrorHandler errorHandler = this.errorHandler;
			if (errorHandler == null) {
				errorHandler = TaskUtils.getDefaultErrorHandler(true);
			}
			return new ReschedulingRunnable(task, trigger, this.clock, executor, errorHandler).schedule();
		}
		catch (RejectedExecutionException ex) {
			throw new TaskRejectedException("Executor [" + executor + "] did not accept task: " + task, ex);
		}
	}

/**
 * @author @author Created by zhuzqc on 2023/1/30 15:39
 * 任務(wù)線(xiàn)程池管理工具
 */
public class TaskSchedulerUtil {

    private static final Logger logger = LoggerFactory.getLogger(TaskSchedulerUtil.class);

    /**
     * 線(xiàn)程調(diào)度工具對(duì)象，作為該類(lèi)的成員變量
     */
    private static ThreadPoolTaskScheduler threadPoolTaskScheduler = new ThreadPoolTaskScheduler();

    /**
     *初始化 map 對(duì)象，裝配 schedule 方法的返回對(duì)象為 value 值
     */
    private static Map<String, ScheduledFuture<?>> scheduledFutureMap = new HashMap<String, ScheduledFuture<?>>();

    static {
        threadPoolTaskScheduler.initialize();
    }

    /**
     * 將Runnable對(duì)象和Trigger對(duì)象作為參數(shù)傳入該靜態(tài)方法
     * @param runnable
     * @param trigger
     * @param 定時(shí)任務(wù)id
     */
    public static void put(Runnable runnable, Trigger trigger, String id) {
        // 將攜帶有Runnable和Trigger的ScheduledFuture類(lèi)對(duì)象作為 Map 的 value 進(jìn)行裝配
        ScheduledFuture<?> scheduledFuture = threadPoolTaskScheduler.schedule(runnable, trigger);
        // 放入 Map 中作為 value
        scheduledFutureMap.put(id, scheduledFuture);
        logger.info("---添加定時(shí)任務(wù)--->" + id);
    }

    /**
     * 通過(guò)上述 put 方法的參數(shù)id（定時(shí)任務(wù)id）標(biāo)識(shí)，將定時(shí)任務(wù)移除出 map
     * @param id
     */
    public static void delete(String id) {
        ScheduledFuture<?> scheduledFuture = scheduledFutureMap.get(id);
        // 條件判斷
        if (scheduledFuture != null && scheduledFuture.isCancelled()) {
            scheduledFuture.cancel(true);
        }
        scheduledFutureMap.remove(id);
        logger.info("---取消定時(shí)任務(wù)--->" + id);
    }
}

/**
 * @author Created by zhuzqc on 2023/1/30 15:58
 */
@Slf4j
@Component
@EnableScheduling
public class TaskScheduleDemo{

    @Resource
    private AAAMapper aaaMapper;
    @Resource
    private BBBService bbbService;

    private Logger logger = LoggerFactory.getLogger(this.getClass());

    // 引入外部的業(yè)務(wù)數(shù)據(jù)對(duì)象
    public void putHiredTask(CCCEntity cccEntity){
        //TODO: 將業(yè)務(wù)線(xiàn)程和定時(shí)觸發(fā)器交由線(xiàn)程池工具管理：創(chuàng)建業(yè)務(wù)線(xiàn)程對(duì)象，并對(duì)屬性賦初始化值（有參構(gòu)造）
        TaskSchedulerUtil.put(new TaskThreadDemo(cccEntity,aaaMapper,bbbService), new Trigger() {
            @Override
            public Date nextExecutionTime(TriggerContext triggerContext) {
                //獲取定時(shí)觸發(fā)器，這里可以獲取頁(yè)面的更新記錄，實(shí)現(xiàn)定時(shí)間隔的動(dòng)態(tài)調(diào)整
                Date nextExecTime = TaskTransUtils.StringToDateTime(cccEntity.getSendTime());
                //cron 表達(dá)式轉(zhuǎn)換工具類(lèi)
                String cron = TaskTransUtils.getDateCronTime(nextExecTime);
                try {
                    if (StringUtils.isBlank(cron)) {
                        // 提示參數(shù)為空
                        logger.info("trackScheduler定時(shí)器的 cron 參數(shù)為空！");
                        // 如果為空則賦默認(rèn)值，每天早上9：00
                        cron = "0 00 09 ? * *";
                    }
                    logger.info("-------定時(shí)任務(wù)執(zhí)行中：" + cron + "--------");
                    CronTrigger cronTrigger = new CronTrigger(cron);
                    nextExecTime = cronTrigger.nextExecutionTime(triggerContext);
                } catch (Exception e) {
                    e.printStackTrace();
                    log.info(e.getMessage());
                }
                return nextExecTime;
            }
        },"該定時(shí)任務(wù)的id");
    }
}