final Protos.TaskInfo.Builder taskInfo = Protos.TaskInfo.newBuilder() .setSlaveId(slaveId) .setTaskId(taskID)
+ " -J-Xms" + String.valueOf((int)igniteTask.mem()) + "m"); return Protos.TaskInfo.newBuilder() .setName("Ignite node " + taskId.getValue()) .setTaskId(taskId)
private OfferRequirement getExistingOfferRequirement(String type, Protos.TaskInfo taskInfo) { LOGGER.info("Getting existing OfferRequirement for task: {}", taskInfo); ExecutorInfo execInfo = taskInfo.getExecutor(); taskInfo = Protos.TaskInfo.newBuilder(taskInfo).clearExecutor().build(); try { return OfferRequirement.create(type, Arrays.asList(taskInfo), Optional.of(execInfo)); } catch (InvalidRequirementException e) { LOGGER.error("Failed to construct OfferRequirement with Exception: ", e); return null; } } }
@Override public OfferRequirement getReplacementOfferRequirement(String type, Protos.TaskInfo taskInfo) { LOGGER.info("Getting replacement requirement for task: {}", taskInfo.getTaskId().getValue()); ExecutorInfo execInfo = taskInfo.getExecutor(); taskInfo = Protos.TaskInfo.newBuilder(taskInfo).clearExecutor().build(); try { return OfferRequirement.create(type, Arrays.asList(taskInfo), Optional.of(execInfo)); } catch (InvalidRequirementException e) { LOGGER.error("Failed to construct OfferRequirement with Exception: ", e); return null; } }
public static Protos.TaskInfo createTask(String taskName) { return Protos.TaskInfo.newBuilder() .setName(taskName) .setTaskId(CommonIdUtils.toTaskId(TestConstants.SERVICE_NAME, taskName)) .setSlaveId(Protos.SlaveID.newBuilder().setValue("ignored")) // proto field required .build(); }
private static Protos.TaskInfo newTaskInfo(final String taskName, final Protos.TaskID taskID) { Protos.TaskInfo.Builder taskBuilder = Protos.TaskInfo.newBuilder() .setName(taskName) .setTaskId(taskID); taskBuilder.getSlaveIdBuilder().setValue("proto-field-required"); return taskBuilder.build(); }
public static Protos.TaskInfo createTask(String taskName) { return Protos.TaskInfo.newBuilder() .setName(taskName) .setTaskId(CommonIdUtils.toTaskId(TestConstants.SERVICE_NAME, taskName)) .setSlaveId(Protos.SlaveID.newBuilder().setValue("ignored")) // proto field required .build(); }
private OfferRequirement getCreateOfferRequirement(String type, Protos.TaskInfo taskInfo) { ExecutorInfo execInfo = taskInfo.getExecutor(); taskInfo = Protos.TaskInfo.newBuilder(taskInfo).clearExecutor().build(); try { return OfferRequirement.create(type, Arrays.asList(taskInfo), Optional.of(execInfo)); } catch (InvalidRequirementException e) { LOGGER.error("Failed to construct OfferRequirement with Exception: ", e); return null; } }
private Protos.TaskInfo buildCommandExecutorTaskInfo(final TaskContext taskContext, final CloudJobConfiguration jobConfig, final ShardingContexts shardingContexts, final Protos.Offer offer, final Protos.CommandInfo command) { Protos.TaskInfo.Builder result = Protos.TaskInfo.newBuilder().setTaskId(Protos.TaskID.newBuilder().setValue(taskContext.getId()).build()) .setName(taskContext.getTaskName()).setSlaveId(offer.getSlaveId()) .addResources(buildResource("cpus", jobConfig.getCpuCount(), offer.getResourcesList())) .addResources(buildResource("mem", jobConfig.getMemoryMB(), offer.getResourcesList())) .setData(ByteString.copyFrom(new TaskInfoData(shardingContexts, jobConfig).serialize())); return result.setCommand(command).build(); }
private static Protos.TaskInfo newTaskInfo(final String taskName) { Protos.TaskInfo.Builder taskBuilder = Protos.TaskInfo.newBuilder() .setName(taskName) .setTaskId(CommonIdUtils.toTaskId(TestConstants.SERVICE_NAME, taskName)); taskBuilder.getSlaveIdBuilder().setValue("proto-field-required"); return taskBuilder.build(); }
private static List<Protos.TaskInfo> clearTaskIds(Collection<Protos.TaskInfo> taskInfos) { List<Protos.TaskInfo> outTaskInfos = new ArrayList<>(); for (Protos.TaskInfo restartTaskInfo : taskInfos) { outTaskInfos.add( Protos.TaskInfo.newBuilder(restartTaskInfo) .setTaskId(Protos.TaskID.newBuilder().setValue("")) .build()); } return outTaskInfos; }
private Protos.TaskInfo normalizeCassandraTaskInfo(CassandraDaemonTask daemonTask) { Protos.TaskInfo daemonTaskInfo = daemonTask.getTaskInfo(); Protos.ExecutorInfo expectedExecutorInfo = Protos.ExecutorInfo.newBuilder(daemonTaskInfo.getExecutor()) .setExecutorId(Protos.ExecutorID.newBuilder().setValue("")) .build(); daemonTaskInfo = Protos.TaskInfo.newBuilder(daemonTaskInfo) .setTaskId(Protos.TaskID.newBuilder().setValue("")) .setExecutor(expectedExecutorInfo) .build(); return daemonTaskInfo; }
@Override public Protos.TaskInfo create(String taskId, Protos.Offer offer, List<Protos.Resource> resources, ExecutionParameters executionParameters) { logger.debug("Creating Mesos task for taskId=" + taskId); return Protos.TaskInfo.newBuilder() .setName(applicationName + ".task") .setSlaveId(offer.getSlaveId()) .setTaskId(Protos.TaskID.newBuilder().setValue(taskId)) .addAllResources(resources) .setCommand(commandInfoMesosProtoFactory.create(executionParameters.getEnvironmentVariables())) .build(); } }
private static Protos.TaskInfo getTestTaskInfo() { return Protos.TaskInfo.newBuilder() .setName(testTaskName) .setTaskId(Protos.TaskID.newBuilder().setValue(testTaskId)) .setSlaveId(Protos.SlaveID.newBuilder().setValue(testAgentId)) .build(); } }
private Protos.TaskInfo createTask(String hostname) throws InvalidProtocolBufferException { return Protos.TaskInfo.newBuilder() .setName("Test") .setTaskId(Protos.TaskID.newBuilder().setValue("TestId").build()) .setSlaveId(Protos.SlaveID.newBuilder().setValue(hostname).build()) .build(); }
private static Protos.TaskInfo createTaskInfoWithZone(String taskName, String zone) { return Protos.TaskInfo.newBuilder(createTask(taskName)) .setCommand(Protos.CommandInfo.newBuilder() .setEnvironment(Protos.Environment.newBuilder() .addVariables(Protos.Environment.Variable.newBuilder() .setName(EnvConstants.ZONE_TASKENV) .setValue(zone)))) .build(); }
private static Protos.TaskInfo newTaskInfo(final String taskName, final Protos.TaskID taskID) { Protos.TaskInfo.Builder taskBuilder = Protos.TaskInfo.newBuilder() .setName(taskName) .setTaskId(taskID); taskBuilder.getSlaveIdBuilder().setValue("proto-field-required"); return taskBuilder.build(); }
public Collection<Protos.TaskInfo> getTaskInfos() { return Arrays.asList( Protos.TaskInfo.newBuilder(daemonTask.getTaskInfo()).clearExecutor().build(), Protos.TaskInfo.newBuilder(clusterTemplateTask.getTaskInfo()).clearExecutor().build()); }
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(); }
@Test public void shouldReturnCorrectNumberOfExecutors() throws IOException { ArrayList<Protos.TaskInfo> mock = Mockito.spy(new ArrayList<>()); mock.add(ProtoTestUtil.getDefaultTaskInfo()); Protos.TaskInfo defaultTaskInfo = Protos.TaskInfo.newBuilder().mergeFrom(ProtoTestUtil.getDefaultTaskInfo()).setTaskId(Protos.TaskID.newBuilder().setValue("Task2")).build(); mock.add(defaultTaskInfo); when(state.get(contains(ESTaskStatus.STATE_KEY))).thenReturn(ProtoTestUtil.getDefaultTaskStatus(Protos.TaskState.TASK_RUNNING)); when(state.get(contains(ClusterState.STATE_LIST))).thenReturn(mock); // Be careful, the state list and state key both have the word state in them. Order is important. assertEquals(2, clusterState.getGuiTaskList().size()); clusterState.removeTask(defaultTaskInfo); assertEquals(1, clusterState.getGuiTaskList().size()); }