com.hubspot.singularity.executor.shells.SingularityExecutorShellCommandRunnerCallable 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.shells;

import com.google.common.base.Charsets;
import com.hubspot.mesos.JavaUtils;
import com.hubspot.singularity.SingularityTaskShellCommandUpdate.UpdateType;
import com.hubspot.singularity.runner.base.shared.SafeProcessManager;
import java.io.File;
import java.io.FileOutputStream;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.util.Date;
import java.util.Optional;
import java.util.concurrent.Callable;
import org.slf4j.Logger;

public class SingularityExecutorShellCommandRunnerCallable
  extends SafeProcessManager
  implements Callable {
  private final SingularityExecutorShellCommandUpdater updater;
  private final ProcessBuilder processBuilder;
  private final File outputFile;

  public SingularityExecutorShellCommandRunnerCallable(
    Logger log,
    SingularityExecutorShellCommandUpdater updater,
    ProcessBuilder processBuilder,
    File outputFile
  ) {
    super(log);
    this.processBuilder = processBuilder;
    this.updater = updater;
    this.outputFile = outputFile;
  }

  @Override
  public Integer call() throws Exception {
    try (
      final PrintWriter outputWriter = new PrintWriter(
        new OutputStreamWriter(new FileOutputStream(outputFile, true), Charsets.UTF_8)
      )
    ) {
      outputWriter.println("#");
      outputWriter.println(
        String.format(
          "# %s -- Launching %s",
          new Date(),
          JavaUtils.SPACE_JOINER.join(processBuilder.command())
        )
      );
      outputWriter.println("#");
    }

    Process process = startProcess(processBuilder);

    Optional pid = getCurrentPid();

    updater.sendUpdate(
      UpdateType.STARTED,
      Optional.of(String.format("pid - %s", pid.orElse(null))),
      Optional.empty()
    );

    try {
      return process.waitFor();
    } catch (InterruptedException ie) {
      signalKillToProcessIfActive();
      throw ie;
    }
  }
}