package com.atlassian.messagequeue.internal.scheduler;
   
   import com.atlassian.messagequeue.registry.MessageContext;
   import com.atlassian.scheduler.JobRunnerRequest;
   import com.atlassian.scheduler.SchedulerService;
   import com.atlassian.scheduler.config.JobId;
   
   import javax.annotation.Nonnull;
   import javax.annotation.Nullable;
  import java.util.Optional;
  
  import static java.util.Objects.requireNonNull;
  
  /**
   * Scheduler {@code MessageContext}.
   */
  public class SchedulerMessageContext implements MessageContext {
      private final JobId jobId;
      private final String payload;
      private final SchedulerService schedulerService;
      private final JobRunnerRequest jobRunnerRequest;
      private boolean autoAcknowledgeMessage = true;
  
      public SchedulerMessageContext(@Nonnull JobRunnerRequest jobRunnerRequest, @Nullable String payload, SchedulerService schedulerService) {
          this.jobRunnerRequest = requireNonNull(jobRunnerRequest, "jobRunnerRequest");
          this.jobId = requireNonNull(jobRunnerRequest.getJobId(), "jobId");
          this.payload = payload;
          this.schedulerService = requireNonNull(schedulerService, "schedulerService");
      }
  
      @Override
      public Optional<String> getMessageId() {
          return Optional.of(jobId.toString());
      }
  
      @Override
      public Optional<String> getPayload() {
          return Optional.ofNullable(payload);
      }
  
      @Override
      public void acknowledge() {
          schedulerService.unscheduleJob(jobId);
      }
  
      @Override
      public boolean isCancellationRequested() {
          return jobRunnerRequest.isCancellationRequested();
      }
  
      @Override
      public boolean shouldAutoAcknowledgeMessage() {
          return autoAcknowledgeMessage;
      }
  
      @Override
      public void cancelAutoAcknowledgementOfMessage() {
          autoAcknowledgeMessage = false;
      }
  }