vid-app-common/src/main/java/org/onap/vid/job/impl/JobSchedulerInitializer.java

   1 package org.onap.vid.job.impl;
   2
   3 import com.google.common.collect.ImmutableMap;
   4 import org.onap.vid.exceptions.GenericUncheckedException;
   5 import org.onap.vid.job.Job;
   6 import org.onap.vid.job.JobsBrokerService;
   7 import org.onap.vid.job.command.JobCommandFactory;
   8 import org.onap.vid.properties.Features;
   9 import org.onap.portalsdk.core.logging.logic.EELFLoggerDelegate;
  10 import org.quartz.*;
  11 import org.springframework.beans.factory.annotation.Autowired;
  12 import org.springframework.scheduling.quartz.SchedulerFactoryBean;
  13 import org.springframework.stereotype.Component;
  14 import org.togglz.core.manager.FeatureManager;
  15
  16 import javax.annotation.PostConstruct;
  17
  18 import static org.quartz.SimpleScheduleBuilder.simpleSchedule;
  19
  20 @Component
  21 public class JobSchedulerInitializer {
  22
  23     private JobsBrokerService jobsBrokerService;
  24     private SchedulerFactoryBean schedulerFactoryBean;
  25     private FeatureManager featureManager;
  26     private JobCommandFactory jobCommandFactory;
  27     private EELFLoggerDelegate logger = EELFLoggerDelegate.getLogger(JobSchedulerInitializer.class);
  28
  29     @Autowired
  30     public JobSchedulerInitializer(
  31             JobsBrokerService jobsBrokerService,
  32             SchedulerFactoryBean schedulerFactoryBean,
  33             FeatureManager featureManager,
  34             JobCommandFactory JobCommandFactory
  35     ) {
  36         this.jobsBrokerService = jobsBrokerService;
  37         this.schedulerFactoryBean = schedulerFactoryBean;
  38         this.featureManager = featureManager;
  39         this.jobCommandFactory = JobCommandFactory;
  40
  41     }
  42
  43     @PostConstruct
  44     public void init() {
  45         if (!featureManager.isActive(Features.FLAG_ASYNC_JOBS)) {
  46             return;
  47         }
  48         scheduleJobWorker(Job.JobStatus.PENDING, 1);
  49         scheduleJobWorker(Job.JobStatus.IN_PROGRESS, 1);
  50     }
  51
  52     private void scheduleJobWorker(Job.JobStatus topic, int intervalInSeconds) {
  53         Scheduler scheduler = schedulerFactoryBean.getScheduler();
  54         JobDetail jobDetail = JobBuilder.newJob().ofType(JobWorker.class)
  55                 .withIdentity("AsyncWorkersJob" + topic)
  56                 .withDescription("Job that run async worker for " + topic)
  57                 .setJobData(new JobDataMap(ImmutableMap.of(
  58                         "jobsBrokerService", jobsBrokerService,
  59                         "jobCommandFactory", jobCommandFactory,
  60                         "featureManager", featureManager,
  61                         "topic", topic
  62                 )))
  63                 .build();
  64         Trigger asyncWorkerTrigger = TriggerBuilder.newTrigger().forJob(jobDetail)
  65                 .withIdentity("AsyncWorkersTrigger" + topic)
  66                 .withDescription("Trigger to run async worker for " + topic)
  67                 .withSchedule(simpleSchedule().repeatForever().withIntervalInSeconds(intervalInSeconds))
  68                 .build();
  69         try {
  70             scheduler.scheduleJob(jobDetail, asyncWorkerTrigger);
  71         } catch (SchedulerException e) {
  72             logger.error(EELFLoggerDelegate.errorLogger, "Failed to schedule trigger for async worker jobs: {}", e.getMessage());
  73             throw new GenericUncheckedException(e);
  74         }
  75     }
  76 }