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

   1 package org.onap.vid.job.impl;
   2
   3 import org.apache.commons.lang3.StringUtils;
   4 import org.onap.vid.job.Job;
   5 import org.onap.vid.job.JobCommand;
   6 import org.onap.vid.job.JobsBrokerService;
   7 import org.onap.vid.job.NextCommand;
   8 import org.onap.vid.job.command.JobCommandFactory;
   9 import org.onap.vid.properties.Features;
  10 import org.onap.portalsdk.core.logging.logic.EELFLoggerDelegate;
  11 import org.quartz.JobExecutionContext;
  12 import org.springframework.scheduling.quartz.QuartzJobBean;
  13 import org.springframework.stereotype.Component;
  14 import org.togglz.core.manager.FeatureManager;
  15
  16 import java.util.Optional;
  17 import java.util.UUID;
  18
  19 import static org.onap.vid.job.Job.JobStatus.FAILED;
  20 import static org.onap.vid.job.Job.JobStatus.STOPPED;
  21
  22 @Component
  23 public class JobWorker extends QuartzJobBean {
  24
  25     private static final EELFLoggerDelegate LOGGER = EELFLoggerDelegate.getLogger(JobWorker.class);
  26
  27     private JobsBrokerService jobsBrokerService;
  28     private FeatureManager featureManager;
  29     private JobCommandFactory jobCommandFactory;
  30     private Job.JobStatus topic;
  31
  32     @Override
  33     protected void executeInternal(JobExecutionContext context) {
  34         Optional<Job> job;
  35
  36         if (!isMsoNewApiActive()) {
  37             return;
  38         }
  39
  40         job = pullJob();
  41
  42         while (job.isPresent()) {
  43             Job nextJob = executeJobAndGetNext(job.get());
  44             pushBack(nextJob);
  45
  46             job = pullJob();
  47         }
  48     }
  49
  50     private Optional<Job> pullJob() {
  51         try {
  52             return jobsBrokerService.pull(topic, UUID.randomUUID().toString());
  53         } catch (Exception e) {
  54             LOGGER.error(EELFLoggerDelegate.errorLogger, "failed to pull job from queue, breaking: {}", e, e);
  55             return Optional.empty();
  56         }
  57     }
  58
  59     private void pushBack(Job nextJob) {
  60         try {
  61             jobsBrokerService.pushBack(nextJob);
  62         } catch (Exception e) {
  63             LOGGER.error(EELFLoggerDelegate.errorLogger, "failed pushing back job to queue: {}", e, e);
  64         }
  65     }
  66
  67     protected Job executeJobAndGetNext(Job job) {
  68         LOGGER.debug(EELFLoggerDelegate.debugLogger, "going to execute job {} of {}: {}/{}",
  69                 StringUtils.substring(String.valueOf(job.getUuid()), 0, 8),
  70                 StringUtils.substring(String.valueOf(job.getTemplateId()), 0, 8),
  71                 job.getStatus(), job.getType());
  72
  73         NextCommand nextCommand = executeCommandAndGetNext(job);
  74
  75         Job nextJob = setNextCommandInJob(nextCommand, job);
  76
  77         return nextJob;
  78     }
  79
  80     private NextCommand executeCommandAndGetNext(Job job) {
  81         NextCommand nextCommand;
  82         try {
  83             final JobCommand jobCommand = jobCommandFactory.toCommand(job);
  84             nextCommand = jobCommand.call();
  85         } catch (Exception e) {
  86             LOGGER.error(EELFLoggerDelegate.errorLogger, "error while executing job from queue: {}", e, e);
  87             nextCommand = new NextCommand(FAILED);
  88         }
  89
  90         if (nextCommand == null) {
  91             nextCommand = new NextCommand(STOPPED);
  92         }
  93         return nextCommand;
  94     }
  95
  96     private Job setNextCommandInJob(NextCommand nextCommand, Job job) {
  97         LOGGER.debug(EELFLoggerDelegate.debugLogger, "transforming job {} of {}: {}/{} -> {}{}",
  98                 StringUtils.substring(String.valueOf(job.getUuid()), 0, 8),
  99                 StringUtils.substring(String.valueOf(job.getTemplateId()), 0, 8),
 100                 job.getStatus(), job.getType(),
 101                 nextCommand.getStatus(),
 102                 nextCommand.getCommand() != null ? ("/" + nextCommand.getCommand().getType()) : "");
 103
 104         job.setStatus(nextCommand.getStatus());
 105
 106         if (nextCommand.getCommand() != null) {
 107             job.setTypeAndData(nextCommand.getCommand().getType(), nextCommand.getCommand().getData());
 108         }
 109
 110         return job;
 111     }
 112
 113     private boolean isMsoNewApiActive() {
 114         return featureManager.isActive(Features.FLAG_ASYNC_INSTANTIATION);
 115     }
 116
 117
 118     //used by quartz to inject JobsBrokerService into the job
 119     //see JobSchedulerInitializer
 120     public void setJobsBrokerService(JobsBrokerService jobsBrokerService) {
 121         this.jobsBrokerService = jobsBrokerService;
 122     }
 123
 124     public void setFeatureManager(FeatureManager featureManager) {
 125         this.featureManager = featureManager;
 126     }
 127
 128     public void setJobCommandFactory(JobCommandFactory jobCommandFactory) {
 129         this.jobCommandFactory = jobCommandFactory;
 130     }
 131
 132     public void setTopic(Job.JobStatus topic) {
 133         this.topic = topic;
 134     }
 135 }