com.codbex.kronos.xsjob.ds.scheduler.SchedulerManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of codbex-kronos-modules-engines-xsjob Show documentation
The newest version!
/*
 * Copyright (c) 2022 codbex or an codbex affiliate company and contributors
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v2.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v20.html
 *
 * SPDX-FileCopyrightText: 2022 codbex or an codbex affiliate company and contributors
 * SPDX-License-Identifier: EPL-2.0
 */
package com.codbex.kronos.xsjob.ds.scheduler;

import static org.quartz.CronScheduleBuilder.cronSchedule;
import static org.quartz.JobBuilder.newJob;
import static org.quartz.TriggerBuilder.newTrigger;

import com.codbex.kronos.utils.CommonsConstants;
import com.codbex.kronos.utils.CommonsUtils;
import com.codbex.kronos.xsjob.ds.api.IJobCoreService;
import com.codbex.kronos.xsjob.ds.model.JobDefinition;
import com.codbex.kronos.xsjob.ds.scheduler.handler.JobHandler;

import java.util.Date;
import java.util.Set;
import org.eclipse.dirigible.core.scheduler.api.SchedulerException;
import org.quartz.CronTrigger;
import org.quartz.JobDetail;
import org.quartz.JobKey;
import org.quartz.ObjectAlreadyExistsException;
import org.quartz.Scheduler;
import org.quartz.TriggerBuilder;
import org.quartz.TriggerKey;
import org.quartz.impl.matchers.GroupMatcher;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * The Class SchedulerManager.
 */
public class SchedulerManager {

  /** The Constant logger. */
  private static final Logger logger = LoggerFactory.getLogger(SchedulerManager.class);

  /** The scheduler. */
  private static Scheduler scheduler = org.eclipse.dirigible.core.scheduler.manager.SchedulerManager.getScheduler();

  /**
   * Schedule job.
   *
   * @param jobDefinition the job definition
   * @throws SchedulerException the scheduler exception
   */
  public static void scheduleJob(JobDefinition jobDefinition) throws SchedulerException {
    try {
      JobKey jobKey = new JobKey(jobDefinition.getName(), jobDefinition.getGroup());
      TriggerKey triggerKey = new TriggerKey(jobDefinition.getName(), jobDefinition.getGroup());
      if (!scheduler.checkExists(jobKey) && (!scheduler.checkExists(triggerKey))) {
        JobDetail job;
        if (IJobCoreService.KRONOS_DEFINED_GROUP.equals(jobDefinition.getGroup())) {
          // user defined jobs
          job = newJob(JobHandler.class).withIdentity(jobKey).withDescription(jobDefinition.getDescription()).build();
          job.getJobDataMap().put(IJobCoreService.KRONOS_JOB_PARAMETERS, jobDefinition.getParametersAsMap());
          job.getJobDataMap().put(IJobCoreService.KRONOS_JOB_FUNCTION, jobDefinition.getFunction());
          job.getJobDataMap().put(IJobCoreService.KRONOS_JOB_MODULE, jobDefinition.getModule());
        } else {
          return;
        }

        TriggerBuilder triggerBuilder = newTrigger().withIdentity(triggerKey)
            .withSchedule(cronSchedule(jobDefinition.getCronExpression()).withMisfireHandlingInstructionDoNothing());

        if (jobDefinition.getStartAt() != null) {
          triggerBuilder.startAt(new Date(jobDefinition.getStartAt().getTime()));
        }

        if (jobDefinition.getEndAt() != null) {
          triggerBuilder.endAt(new Date(jobDefinition.getEndAt().getTime()));
        }

        scheduler.scheduleJob(job, triggerBuilder.build());

        logger.info("Scheduled Job: [{}] of group: [{}] at: [{}]", jobDefinition.getName(), jobDefinition.getGroup(),
            jobDefinition.getCronExpression());
      }
    } catch (ObjectAlreadyExistsException e) {
      logger.warn(e.getMessage());
    } catch (org.quartz.SchedulerException e) {
      CommonsUtils.logProcessorErrors(e.getMessage(), CommonsConstants.SCHEDULER_ERROR, jobDefinition.getName(),
          CommonsConstants.JOB_PARSER);
      throw new SchedulerException(e);
    }
  }

  /**
   * Unschedule job.
   *
   * @param name the name
   * @param group the group
   * @throws SchedulerException the scheduler exception
   */
  public static void unscheduleJob(String name, String group) throws SchedulerException {
    if (!IJobCoreService.KRONOS_DEFINED_GROUP.equals(group)) {
      return;
    }
    try {
      JobKey jobKey = new JobKey(name, group);
      TriggerKey triggerKey = new TriggerKey(name, group);
      if (scheduler.checkExists(triggerKey)) {
        scheduler.unscheduleJob(triggerKey);
        scheduler.deleteJob(jobKey);
        logger.info("Unscheduled Job: [{}] of group: [{}]", name, group);
      }
    } catch (ObjectAlreadyExistsException e) {
      logger.warn(e.getMessage());
    } catch (org.quartz.SchedulerException e) {
      CommonsUtils.logProcessorErrors(e.getMessage(), CommonsConstants.SCHEDULER_ERROR, name, CommonsConstants.JOB_PARSER);
      throw new SchedulerException(e);
    }
  }

  /**
   * List jobs.
   *
   * @return the sets the
   * @throws SchedulerException the scheduler exception
   */
  public static Set listJobs() throws SchedulerException {
    try {
      Set triggerKeys = scheduler.getTriggerKeys(GroupMatcher.anyTriggerGroup());
      return triggerKeys;
    } catch (org.quartz.SchedulerException e) {
      CommonsUtils.logProcessorErrors(e.getMessage(), CommonsConstants.SCHEDULER_ERROR, "-", CommonsConstants.JOB_PARSER);
      throw new SchedulerException(e);
    }
  }

  /**
   * Exists job.
   *
   * @param name the name
   * @return true, if successful
   * @throws SchedulerException the scheduler exception
   */
  public static boolean existsJob(String name) throws SchedulerException {
    Set triggerKeys = listJobs();
    for (TriggerKey triggerKey : triggerKeys) {
      if (triggerKey.getName().equals(name) && IJobCoreService.KRONOS_DEFINED_GROUP.equals(triggerKey.getGroup())) {
        return true;
      }
    }
    return false;
  }
}