.setTaskId(taskID) .setName(taskID.getValue());
.setTaskId(taskId) .setSlaveId(offer.getSlaveId()) .setCommand(builder)
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; }
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(); }
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(); }
@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(); } }
@Override public RepairTask updateId() { return new RepairTask(getBuilder().setTaskId(createId(getName())) .build()); }
@Override public RestoreSnapshotTask updateId() { return new RestoreSnapshotTask(getBuilder().setTaskId(createId(getName())) .build()); }
@Override public BackupUploadTask updateId() { return new BackupUploadTask(getBuilder().setTaskId(createId(getName())) .build()); }
@Override public RestoreSchemaTask updateId() { return new RestoreSchemaTask(getBuilder().setTaskId(createId(getName())) .build()); }
@Override public CleanupTask updateId() { return new CleanupTask(getBuilder().setTaskId(createId(getName())) .build()); }
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, final Protos.TaskID taskID) { Protos.TaskInfo.Builder taskBuilder = Protos.TaskInfo.newBuilder() .setName(taskName) .setTaskId(taskID); taskBuilder.getSlaveIdBuilder().setValue("proto-field-required"); return taskBuilder.build(); }
private static TaskInfo getTaskInfo(String taskName, String attrName, String attrVal) { TaskInfo.Builder infoBuilder = TaskTestUtils.getTaskInfo(Collections.emptyList()).toBuilder() .setName(taskName) .setTaskId(CommonIdUtils.toTaskId(TestConstants.SERVICE_NAME, taskName)); infoBuilder.setLabels(new TaskLabelWriter(infoBuilder) .setOfferAttributes(offerWithAttribute(attrName, attrVal)) .toProto()); return infoBuilder.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 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; }
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 getTestTaskInfo() { return Protos.TaskInfo.newBuilder() .setName(testTaskName) .setTaskId(Protos.TaskID.newBuilder().setValue(testTaskId)) .setSlaveId(Protos.SlaveID.newBuilder().setValue(testAgentId)) .build(); } }
@Override public UpgradeSSTableTask updateId() { return new UpgradeSSTableTask(getBuilder().setTaskId(createId(getName())) .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()); }