@Override public Protos.CommandInfo create(Map<String, String> additionalEnvironmentVariables) { Protos.CommandInfo.Builder builder = Protos.CommandInfo.newBuilder(); Optional<String> command = Optional.ofNullable(mesosConfig.getCommand()); builder.setShell(command.isPresent()); command.ifPresent(builder::setValue); builder.addAllUris(mesosConfig.getUri().stream().map(uri -> Protos.CommandInfo.URI.newBuilder().setValue(uri).build()).collect(Collectors.toList())); Map<String, String> environmentVariables = new LinkedHashMap<>(); environmentVariables.putAll(additionalEnvironmentVariables); environmentVariables.putAll(mesosConfig.getEnvironment()); //todo: migrate to foreach environmentVariables.entrySet().stream() .map(kv -> Protos.Environment.Variable.newBuilder().setName(kv.getKey()).setValue(kv.getValue()).build()) .collect(Collectors.collectingAndThen( Collectors.toList(), variables -> builder.setEnvironment(Protos.Environment.newBuilder().addAllVariables(variables)))); return builder.build(); } }
private Protos.CommandInfo buildCommand(final Protos.CommandInfo.URI uri, final String script, final ShardingContexts shardingContexts, final boolean isCommandExecutor) { Protos.CommandInfo.Builder result = Protos.CommandInfo.newBuilder().addUris(uri).setShell(true); if (isCommandExecutor) { CommandLine commandLine = CommandLine.parse(script); commandLine.addArgument(GsonFactory.getGson().toJson(shardingContexts), false); result.setValue(Joiner.on(" ").join(commandLine.getExecutable(), Joiner.on(" ").join(commandLine.getArguments()))); } else { result.setValue(script); } return result.build(); }
.setName(name) .setSlaveId(aggregatedOffers.getSlaveID()) .setCommand(commandInfoBuilder.build()) .addAllResources(resources) .build();
.setRole(diskRole) .setScalar(Value.Scalar.newBuilder().setValue(containerDisk))) .setCommand(commandInfo.build());
public static final double MEM = 1000.0; public static final double DISK = 2000.0; public static final Protos.CommandInfo CMD = Protos.CommandInfo.newBuilder().setValue("echo test-cmd").build();
.setEnvironment(envBuilder) .setValue(String.format("cd %s && %s", directory, command)) .addUris(CommandInfo.URI.newBuilder().setValue(uri)).build();
.setTaskId(taskId) .setSlaveId(offer.getSlaveId()) .setCommand(commandBuilder.build());
public static Protos.TaskInfo getDefaultTaskInfo() { return Protos.TaskInfo.newBuilder() .setName("dummyTaskName") .setTaskId(Protos.TaskID.newBuilder().setValue(SLAVE_ID)) .setSlaveId(getSlaveId()) .setExecutor(Protos.ExecutorInfo.newBuilder() .setExecutorId(getExecutorId()) .setCommand(Protos.CommandInfo.newBuilder().setValue("").build()) .build()) .setDiscovery( Protos.DiscoveryInfo.newBuilder() .setVisibility(Protos.DiscoveryInfo.Visibility.EXTERNAL) .setPorts(Protos.Ports.newBuilder() .addPorts(Discovery.CLIENT_PORT_INDEX, Protos.Port.newBuilder().setNumber(9200)) .addPorts(Discovery.TRANSPORT_PORT_INDEX, Protos.Port.newBuilder().setNumber(9300)) ) ) .build(); }
public static CommandInfo createCommandInfo( final String command, final List<String> arguments, final Set<String> uris, boolean cacheFetchedUris, final Map<String, String> environment) { return CommandInfo.newBuilder() .setValue(command) .addAllArguments(arguments) .addAllUris(createURIs(uris, cacheFetchedUris)) .setEnvironment(createEnvironment(environment)).build(); }
private static Protos.ExecutorInfo.Builder getExecutorInfoBuilder() { Protos.CommandInfo cmd = Protos.CommandInfo.newBuilder().build(); return Protos.ExecutorInfo.newBuilder() .setExecutorId(Protos.ExecutorID.newBuilder().setValue("")) .setName(TestConstants.EXECUTOR_NAME) .setCommand(cmd); }
public static Protos.TaskInfo getTaskInfoExternalVolume(Integer esNodeId) { return Protos.TaskInfo.newBuilder(getDefaultTaskInfo()) .setCommand(Protos.CommandInfo.newBuilder() .setValue("") .setEnvironment(Protos.Environment.newBuilder() .addVariables(Protos.Environment.Variable.newBuilder() .setName(ExecutorEnvironmentalVariables.ELASTICSEARCH_NODE_ID) .setValue(esNodeId.toString()))) .build()) .build(); }