com.hubspot.singularity.executor.models.DockerEnvironmentContext 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.models;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import org.apache.mesos.Protos.Environment.Variable;
import org.apache.mesos.Protos.TaskInfo;

public class DockerEnvironmentContext extends EnvironmentContext {
  private final List inheritVariables;

  public DockerEnvironmentContext(TaskInfo taskInfo, List inheritVariables) {
    super(taskInfo);
    this.inheritVariables = inheritVariables;
  }

  @Override
  public List getEnv() {
    List variables = new ArrayList<>();
    Set keys = new HashSet<>();

    inheritVariables.forEach(
      v -> {
        if (!keys.contains(v)) {
          String val = System.getenv(v);
          if (val != null) {
            variables.add(Variable.newBuilder().setName(v).setValue(val).build());
            keys.add(v);
          }
        }
      }
    );

    taskInfo
      .getExecutor()
      .getCommand()
      .getEnvironment()
      .getVariablesList()
      .forEach(
        v -> {
          if (!keys.contains(v.getName())) {
            variables.add(v);
            keys.add(v.getName());
          }
        }
      );
    return variables;
  }
}