hadoop DockerCommand 源码

haddop DockerCommand 代码

文件路径：/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-nodemanager/src/main/java/org/apache/hadoop/yarn/server/nodemanager/containermanager/linux/runtime/docker/DockerCommand.java

/*
 * *
 *  Licensed to the Apache Software Foundation (ASF) under one
 *  or more contributor license agreements.  See the NOTICE file
 *  distributed with this work for additional information
 *  regarding copyright ownership.  The ASF licenses this file
 *  to you under the Apache License, Version 2.0 (the
 *  "License"); you may not use this file except in compliance
 *  with the License.  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 * /
 */

package org.apache.hadoop.yarn.server.nodemanager.containermanager.linux.runtime.docker;

import org.apache.hadoop.classification.InterfaceAudience;
import org.apache.hadoop.classification.InterfaceStability;
import org.apache.hadoop.util.StringUtils;
import org.apache.hadoop.yarn.api.records.ContainerId;
import org.apache.hadoop.yarn.server.nodemanager.Context;
import org.apache.hadoop.yarn.server.nodemanager.containermanager.linux.privileged.PrivilegedOperation;
import org.apache.hadoop.yarn.server.nodemanager.containermanager.runtime.ContainerExecutionException;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;

@InterfaceAudience.Private
@InterfaceStability.Unstable

/** Represents a docker sub-command
 * e.g 'run', 'load', 'inspect' etc.,
 */

public abstract class DockerCommand {
  private final String command;
  private final Map<String, List<String>> commandArguments;

  protected DockerCommand(String command) {
    String dockerCommandKey = "docker-command";
    this.command = command;
    this.commandArguments = new TreeMap<>();
    commandArguments.put(dockerCommandKey, new ArrayList<>());
    commandArguments.get(dockerCommandKey).add(command);
  }

  /**
   * Returns the docker sub-command string being used
   * e.g 'run'.
   */
  public final String getCommandOption() {
    return this.command;
  }

  /**
   * Add command commandWithArguments - this method is only meant for use by
   * sub-classes.
   *
   * @param key   name of the key to be added
   * @param value value of the key
   */
  protected final void addCommandArguments(String key, String value) {
    List<String> list = commandArguments.get(key);
    if (list != null) {
      list.add(value);
      return;
    }
    list = new ArrayList<>();
    list.add(value);
    this.commandArguments.put(key, list);
  }

  public Map<String, List<String>> getDockerCommandWithArguments() {
    return Collections.unmodifiableMap(commandArguments);
  }

  @Override
  public String toString() {
    StringBuffer ret = new StringBuffer(this.command);
    for (Map.Entry<String, List<String>> entry : commandArguments.entrySet()) {
      ret.append(" ").append(entry.getKey());
      ret.append("=").append(StringUtils.join(",", entry.getValue()));
    }
    return ret.toString();
  }

  /**
   * Add the client configuration directory to the docker command.
   *
   * The client configuration option proceeds any of the docker subcommands
   * (such as run, load, pull, etc). Ordering will be handled by
   * container-executor. Docker expects the value to be a directory containing
   * the file config.json. This file is typically generated via docker login.
   *
   * @param clientConfigDir - directory containing the docker client config.
   */
  public void setClientConfigDir(String clientConfigDir) {
    if (clientConfigDir != null) {
      addCommandArguments("docker-config", clientConfigDir);
    }
  }

  /**
   * Prepare the privileged operation object that will be used to invoke
   * the container-executor.
   *
   * @param dockerCommand Specific command to be run by docker.
   * @param containerName
   * @param env
   * @param nmContext
   * @return Returns the PrivilegedOperation object to be used.
   * @throws ContainerExecutionException
   */
  public PrivilegedOperation preparePrivilegedOperation(
      DockerCommand dockerCommand, String containerName, Map<String,
      String> env, Context nmContext)
      throws ContainerExecutionException {
    DockerClient dockerClient = new DockerClient();
    String commandFile =
        dockerClient.writeCommandToTempFile(dockerCommand,
        ContainerId.fromString(containerName),
        nmContext);
    PrivilegedOperation dockerOp = new PrivilegedOperation(
        PrivilegedOperation.OperationType.RUN_DOCKER_CMD);
    dockerOp.appendArgs(commandFile);
    return dockerOp;
  }
}

相关信息

hadoop 源码目录

相关文章

hadoop DockerClient 源码

hadoop DockerCommandExecutor 源码

hadoop DockerExecCommand 源码

hadoop DockerImagesCommand 源码

hadoop DockerInspectCommand 源码

hadoop DockerKillCommand 源码

hadoop DockerLoadCommand 源码

hadoop DockerPullCommand 源码

hadoop DockerRmCommand 源码

hadoop DockerRunCommand 源码