org.noear.luffy.event.schedule.controller.ScheduleHelper Maven / Gradle / Ivy

Go to download
package org.noear.luffy.event.schedule.controller;

import org.noear.luffy.model.AFileModel;
import org.noear.luffy.task.cron.CronExpressionPlus;
import org.noear.luffy.task.cron.CronUtils;
import org.noear.luffy.utils.Datetime;

import java.text.ParseException;
import java.util.Date;

/**
 * @author noear
 */
public class ScheduleHelper {
    public static ScheduleNext getNextTimeByCron(AFileModel task, Date baseTime) throws ParseException {
        ScheduleNext next = new ScheduleNext();

        Datetime now_time = Datetime.Now();

        CronExpressionPlus cron = CronUtils.get(task.plan_interval);

        next.datetime = cron.getNextValidTimeAfter(baseTime);

        if (task.plan_state != 1) {
            //如果，限制特定的小时
            if (cron.getHours().size() < 24) {
                int now_hour = now_time.getHours();
                next.allow = false;

                for (Integer h : cron.getHours()) {
                    if (now_hour == h) {
                        next.allow = true;
                        break;
                    }
                }
            }

            //如果，限制特定的分
            if (next.allow && cron.getMinutes().size() < 60) {
                int now_minute = now_time.getMinutes();
                next.allow = false;

                for (Integer m : cron.getMinutes()) {
                    if (now_minute == m) {
                        next.allow = true;
                        break;
                    }
                }
            }
        } else {
            next.allow = true;
        }

        return next;
    }

    public static ScheduleNext getNextTimeBySimple(AFileModel task, Date baseTime) {
        ScheduleNext next = new ScheduleNext();

        Datetime begin_time = new Datetime(task.plan_begin_time);
        Datetime next_time = new Datetime(baseTime);
        Datetime now_time = Datetime.Now();

        String s1 = task.plan_interval.substring(0, task.plan_interval.length() - 1);
        String s2 = task.plan_interval.substring(task.plan_interval.length() - 1);

        switch (s2) {
            case "s": //秒
                next_time.addSecond(Integer.parseInt(s1));
                break;
            case "m": //分
                next_time.setSecond(begin_time.getSeconds());
                next_time.addMinute(Integer.parseInt(s1));
                break;
            case "h": //时
                next_time.setMinute(begin_time.getMinutes());
                next_time.setSecond(begin_time.getSeconds());

                next_time.addHour(Integer.parseInt(s1));
                break;
            case "d": //日
                next.intervalOfDay = true;
                next.allow = (now_time.getHours() == begin_time.getHours());

                next_time.setHour(begin_time.getHours());
                next_time.setMinute(begin_time.getMinutes());
                next_time.setSecond(begin_time.getSeconds());

                next_time.addDay(Integer.parseInt(s1));
                break;
            default:
                next.allow = false; //不支持的格式，不允许执行
                next_time.addDay(1);
                break;
        }

        next.datetime = next_time.getFulltime();

        if (task.plan_state == 1) {
            next.allow = true;
        }

        return next;
    }
}