Java ContainerLaunch.getExitCodeFile方法代码示例

import org.apache.hadoop.yarn.server.nodemanager.containermanager.launcher.ContainerLaunch; //导入方法依赖的package包/类
@Override
public void writeLocalWrapperScript(Path launchDst, Path pidFile,
                                    PrintStream pout) {

  String exitCodeFile = ContainerLaunch.getExitCodeFile(
    pidFile.toString());
  String tmpFile = exitCodeFile + ".tmp";
  pout.println("#!/usr/bin/env bash");
  pout.println("bash \"" + sessionScriptPath.toString() + "\"");
  pout.println("rc=$?");
  pout.println("echo $rc > \"" + tmpFile + "\"");
  pout.println("mv -f \"" + tmpFile + "\" \"" + exitCodeFile + "\"");
  pout.println("exit $rc");
}
 

import org.apache.hadoop.yarn.server.nodemanager.containermanager.launcher.ContainerLaunch; //导入方法依赖的package包/类
@Override
public void writeLocalWrapperScript(Path launchDst, Path pidFile,
    PrintStream pout) {
  String exitCodeFile = ContainerLaunch.getExitCodeFile(
      pidFile.toString());
  String tmpFile = exitCodeFile + ".tmp";
  pout.println("#!/bin/bash");
  pout.println("/bin/bash \"" + sessionScriptPath.toString() + "\"");
  pout.println("rc=$?");
  pout.println("echo $rc > \"" + tmpFile + "\"");
  pout.println("/bin/mv -f \"" + tmpFile + "\" \"" + exitCodeFile + "\"");
  pout.println("exit $rc");
}
 

import org.apache.hadoop.yarn.server.nodemanager.containermanager.launcher.ContainerLaunch; //导入方法依赖的package包/类
@Override
public void writeLocalWrapperScript(Path launchDst, Path pidFile,
  PrintStream pout) {
  String exitCodeFile = ContainerLaunch.getExitCodeFile(
    pidFile.toString());
  String tmpFile = exitCodeFile + ".tmp";
  pout.println("#!/usr/bin/env bash");
  pout.println("bash \"" + sessionScriptPath.toString() + "\"");
  pout.println("rc=$?");
  pout.println("echo $rc > \"" + tmpFile + "\"");
  pout.println("mv -f \"" + tmpFile + "\" \"" + exitCodeFile + "\"");
  pout.println("exit $rc");
}
 

import org.apache.hadoop.yarn.server.nodemanager.containermanager.launcher.ContainerLaunch; //导入方法依赖的package包/类
/**
 * Recover an already existing container. This is a blocking call and returns
 * only when the container exits.  Note that the container must have been
 * activated prior to this call.
 * @param ctx encapsulates information necessary to reacquire container
 * @return The exit code of the pre-existing container
 * @throws IOException
 * @throws InterruptedException 
 */
public int reacquireContainer(ContainerReacquisitionContext ctx)
    throws IOException, InterruptedException {
  Container container = ctx.getContainer();
  String user = ctx.getUser();
  ContainerId containerId = ctx.getContainerId();


  Path pidPath = getPidFilePath(containerId);
  if (pidPath == null) {
    LOG.warn(containerId + " is not active, returning terminated error");
    return ExitCode.TERMINATED.getExitCode();
  }

  String pid = null;
  pid = ProcessIdFileReader.getProcessId(pidPath);
  if (pid == null) {
    throw new IOException("Unable to determine pid for " + containerId);
  }

  LOG.info("Reacquiring " + containerId + " with pid " + pid);
  ContainerLivenessContext livenessContext = new ContainerLivenessContext
      .Builder()
      .setContainer(container)
      .setUser(user)
      .setPid(pid)
      .build();
  while(isContainerAlive(livenessContext)) {
    Thread.sleep(1000);
  }

  // wait for exit code file to appear
  String exitCodeFile = ContainerLaunch.getExitCodeFile(pidPath.toString());
  File file = new File(exitCodeFile);
  final int sleepMsec = 100;
  int msecLeft = 2000;
  while (!file.exists() && msecLeft >= 0) {
    if (!isContainerActive(containerId)) {
      LOG.info(containerId + " was deactivated");
      return ExitCode.TERMINATED.getExitCode();
    }
    
    Thread.sleep(sleepMsec);
    
    msecLeft -= sleepMsec;
  }
  if (msecLeft < 0) {
    throw new IOException("Timeout while waiting for exit code from "
        + containerId);
  }

  try {
    return Integer.parseInt(FileUtils.readFileToString(file).trim());
  } catch (NumberFormatException e) {
    throw new IOException("Error parsing exit code from pid " + pid, e);
  }
}
 

import org.apache.hadoop.yarn.server.nodemanager.containermanager.launcher.ContainerLaunch; //导入方法依赖的package包/类
/**
 * Recover an already existing container. This is a blocking call and returns
 * only when the container exits.  Note that the container must have been
 * activated prior to this call.
 * @param user the user of the container
 * @param containerId The ID of the container to reacquire
 * @return The exit code of the pre-existing container
 * @throws IOException
 * @throws InterruptedException 
 */
public int reacquireContainer(String user, ContainerId containerId)
    throws IOException, InterruptedException {
  Path pidPath = getPidFilePath(containerId);
  if (pidPath == null) {
    LOG.warn(containerId + " is not active, returning terminated error");
    return ExitCode.TERMINATED.getExitCode();
  }

  String pid = null;
  pid = ProcessIdFileReader.getProcessId(pidPath);
  if (pid == null) {
    throw new IOException("Unable to determine pid for " + containerId);
  }

  LOG.info("Reacquiring " + containerId + " with pid " + pid);
  while(isContainerProcessAlive(user, pid)) {
    Thread.sleep(1000);
  }

  // wait for exit code file to appear
  String exitCodeFile = ContainerLaunch.getExitCodeFile(pidPath.toString());
  File file = new File(exitCodeFile);
  final int sleepMsec = 100;
  int msecLeft = 2000;
  while (!file.exists() && msecLeft >= 0) {
    if (!isContainerActive(containerId)) {
      LOG.info(containerId + " was deactivated");
      return ExitCode.TERMINATED.getExitCode();
    }
    
    Thread.sleep(sleepMsec);
    
    msecLeft -= sleepMsec;
  }
  if (msecLeft < 0) {
    throw new IOException("Timeout while waiting for exit code from "
        + containerId);
  }

  try {
    return Integer.parseInt(FileUtils.readFileToString(file).trim());
  } catch (NumberFormatException e) {
    throw new IOException("Error parsing exit code from pid " + pid, e);
  }
}