com.hubspot.singularity.executor.task.SingularityExecutorTaskDefinition Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of SingularityExecutor Show documentation
The newest version!
package com.hubspot.singularity.executor.task;

import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonIgnore;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.hubspot.deploy.HealthcheckOptions;
import com.hubspot.singularity.SingularityTaskExecutorData;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Optional;

public class SingularityExecutorTaskDefinition {
  private final SingularityTaskExecutorData executorData;
  private final String taskId;
  private final Path taskDirectoryPath;
  private final String executorBashOut;
  private final String serviceLogOut;
  private final String serviceLogOutExtension;
  private final String serviceFinishedTailLog;
  private final String taskAppDirectory;
  private final String logrotateStateFile;
  private final String executorPid;
  private final String signatureVerifyOut;

  @JsonCreator
  public SingularityExecutorTaskDefinition(
    @JsonProperty("taskId") String taskId,
    @JsonProperty("executorData") SingularityTaskExecutorData executorData,
    @JsonProperty("taskDirectory") String taskDirectory,
    @JsonProperty("executorPid") String executorPid,
    @JsonProperty("serviceLogOut") String serviceLogOut,
    @JsonProperty("serviceLogOutExtension") String serviceLogOutExtension,
    @JsonProperty("serviceFinishedTailLog") String serviceFinishedTailLog,
    @JsonProperty("taskAppDirectory") String taskAppDirectory,
    @JsonProperty("executorBashOut") String executorBashOut,
    @JsonProperty("logrotateStateFilePath") String logrotateStateFile,
    @JsonProperty("signatureVerifyOut") String signatureVerifyOut
  ) {
    this.executorData = executorData;
    this.taskId = taskId;
    this.taskDirectoryPath = Paths.get(taskDirectory);
    this.executorPid = executorPid;
    this.executorBashOut = executorBashOut;
    this.serviceLogOut = serviceLogOut;
    this.serviceLogOutExtension = serviceLogOutExtension;
    this.serviceFinishedTailLog = serviceFinishedTailLog;
    this.taskAppDirectory = taskAppDirectory;
    this.logrotateStateFile = logrotateStateFile;
    this.signatureVerifyOut = signatureVerifyOut;
  }

  @JsonIgnore
  public Path getTaskDirectoryPath() {
    return taskDirectoryPath;
  }

  @JsonIgnore
  public Path getExecutorBashOutPath() {
    return taskDirectoryPath.resolve(executorBashOut);
  }

  @JsonIgnore
  public Path getServiceLogOutPath() {
    return taskDirectoryPath.resolve(serviceLogOut);
  }

  @JsonIgnore
  public Path getTaskAppDirectoryPath() {
    return taskDirectoryPath.resolve(taskAppDirectory);
  }

  @JsonIgnore
  public Path getLogrotateStateFilePath() {
    return taskDirectoryPath.resolve(logrotateStateFile);
  }

  @JsonIgnore
  /**
   * Convenience method for handling skipLogrotateAndCompress
   */
  public boolean shouldLogrotateLogFile() {
    return !executorData
      .getSkipLogrotateAndCompress()
      .orElse(Boolean.FALSE)
      .booleanValue();
  }

  @JsonIgnore
  public Path getSignatureVerifyOutPath() {
    return taskDirectoryPath.resolve(signatureVerifyOut);
  }

  public String getTaskDirectory() {
    return taskDirectoryPath.toString();
  }

  public String getExecutorBashOut() {
    return getExecutorBashOutPath().toString();
  }

  public String getServiceLogOut() {
    return getServiceLogOutPath().toString();
  }

  @JsonIgnore
  public String getServiceLogFileName() {
    return serviceLogOut;
  }

  @JsonIgnore
  public String getServiceFinishedTailLogFileName() {
    return serviceFinishedTailLog;
  }

  public String getServiceLogOutExtension() {
    return serviceLogOutExtension;
  }

  public String getServiceFinishedTailLog() {
    return getServiceFinishedTailLogPath().toString();
  }

  @JsonIgnore
  public Path getServiceFinishedTailLogPath() {
    return taskDirectoryPath.resolve(serviceFinishedTailLog);
  }

  public String getTaskAppDirectory() {
    return getTaskAppDirectoryPath().toString();
  }

  public String getLogrotateStateFile() {
    return getLogrotateStateFilePath().toString();
  }

  public SingularityTaskExecutorData getExecutorData() {
    return executorData;
  }

  public String getTaskId() {
    return taskId;
  }

  public String getExecutorPid() {
    return executorPid;
  }

  public String getSignatureVerifyOut() {
    return signatureVerifyOut;
  }

  public Optional getHealthcheckResultFilePath() {
    if (executorData.getHealthcheckOptions().isPresent()) {
      return executorData.getHealthcheckOptions().get().getHealthcheckResultFilePath();
    }
    return Optional.empty();
  }

  public Optional getHealthcheckOptions() {
    return executorData.getHealthcheckOptions();
  }

  @JsonIgnore
  public Optional getExecutorPidSafe() {
    try {
      return Optional.of(Integer.parseInt(executorPid));
    } catch (NumberFormatException nfe) {
      return Optional.empty();
    }
  }

  @Override
  public String toString() {
    return (
      "SingularityExecutorTaskDefinition{" +
      "executorData=" +
      executorData +
      ", taskId='" +
      taskId +
      '\'' +
      ", taskDirectoryPath=" +
      taskDirectoryPath +
      ", executorBashOut='" +
      executorBashOut +
      '\'' +
      ", serviceLogOut='" +
      serviceLogOut +
      '\'' +
      ", serviceLogOutExtension='" +
      serviceLogOutExtension +
      '\'' +
      ", serviceFinishedTailLog='" +
      serviceFinishedTailLog +
      '\'' +
      ", taskAppDirectory='" +
      taskAppDirectory +
      '\'' +
      ", logrotateStateFile='" +
      logrotateStateFile +
      '\'' +
      ", executorPid='" +
      executorPid +
      '\'' +
      ", signatureVerifyOut='" +
      signatureVerifyOut +
      '\'' +
      '}'
    );
  }
}