@Override public int compareTo(TaskId other) { int jobIdComp = this.getJobId().compareTo(other.getJobId()); if (jobIdComp == 0) { if (this.getTaskType() == other.getTaskType()) { return this.getId() - other.getId(); } else { return this.getTaskType().compareTo(other.getTaskType()); } } else { return jobIdComp; } } }
public static String toString(TaskId tid) { return tid.toString(); }
public static TaskId newTaskId(JobId jobId, int id, TaskType taskType) { TaskId taskId = Records.newRecord(TaskId.class); taskId.setJobId(jobId); taskId.setId(id); taskId.setTaskType(taskType); return taskId; }
private AtomicInteger containerNeed(TaskId taskID) { JobId jobID = taskID.getJobId(); TaskType taskType = taskID.getTaskType(); ConcurrentMap<JobId, AtomicInteger> relevantMap = taskType == TaskType.MAP ? mapContainerNeeds : reduceContainerNeeds; AtomicInteger result = relevantMap.get(jobID); if (result == null) { relevantMap.putIfAbsent(jobID, new AtomicInteger(0)); result = relevantMap.get(jobID); } return result; }
@Override public GetTaskReportResponse getTaskReport(GetTaskReportRequest request) throws IOException { TaskId taskId = request.getTaskId(); Job job = verifyAndGetJob(taskId.getJobId(), true); GetTaskReportResponse response = recordFactory.newRecordInstance(GetTaskReportResponse.class); response.setTaskReport(job.getTask(taskId).getReport()); return response; }
@Override public TaskType getType() { return taskID.getTaskType(); }
taskId.setId(0); taskId.setTaskType(TaskType.REDUCE); taskId.setJobId(jobId); Task task = mock(Task.class); when(task.getID()).thenReturn(taskId); pWriter.flush(); assertTrue(data.toString().contains( "<a href='" + block.url("task",task.getID().toString()) +"'>" +"attempt_0_0001_r_000000_0</a>"));
@Override public long estimatedRuntime(TaskAttemptId id) { if ((id.getTaskId().getId() == 0) && (id.getId() == 0)) { return SPECULATE_THIS; } return super.estimatedRuntime(id); } }
if (taskId.getTaskType() == TaskType.REDUCE) { job.handle(new JobTaskEvent(taskId, TaskState.SUCCEEDED)); Assert.assertEquals("Number of killed events", 2, killedEvents.size()); Assert.assertEquals("AttemptID", "task_1234567890000_0001_m_000000", killedEvents.get(0).getTaskID().toString()); Assert.assertEquals("AttemptID", "task_1234567890000_0001_m_000001", killedEvents.get(1).getTaskID().toString()); } else { Assert.assertEquals("Number of killed events", 0, killedEvents.size());
@Override public int compareTo(TaskAttemptId other) { int taskIdComp = this.getTaskId().compareTo(other.getTaskId()); if (taskIdComp == 0) { return this.getId() - other.getId(); } else { return taskIdComp; } } }
@Override public boolean equals(Object obj) { if (this == obj) return true; if (obj == null) return false; if (getClass() != obj.getClass()) return false; TaskAttemptId other = (TaskAttemptId) obj; if (getId() != other.getId()) return false; if (!getTaskId().equals(other.getTaskId())) return false; return true; }
private AtomicInteger containerNeed(TaskId taskID) { JobId jobID = taskID.getJobId(); TaskType taskType = taskID.getTaskType(); ConcurrentMap<JobId, AtomicInteger> relevantMap = taskType == TaskType.MAP ? mapContainerNeeds : reduceContainerNeeds; AtomicInteger result = relevantMap.get(jobID); if (result == null) { relevantMap.putIfAbsent(jobID, new AtomicInteger(0)); result = relevantMap.get(jobID); } return result; }
@Override public GetTaskAttemptReportResponse getTaskAttemptReport( GetTaskAttemptReportRequest request) throws IOException { TaskAttemptId taskAttemptId = request.getTaskAttemptId(); Job job = verifyAndGetJob(taskAttemptId.getTaskId().getJobId(), true); GetTaskAttemptReportResponse response = recordFactory.newRecordInstance(GetTaskAttemptReportResponse.class); response.setTaskAttemptReport(job.getTask(taskAttemptId.getTaskId()).getAttempt(taskAttemptId).getReport()); return response; }
void add(Container container, TaskAttemptId tId) { LOG.info("Assigned container " + container.getId().toString() + " to " + tId); containerToAttemptMap.put(container.getId(), tId); if (tId.getTaskId().getTaskType().equals(TaskType.MAP)) { maps.put(tId, container); } else { reduces.put(tId, container); } }
@Override public void handle(ContainerLauncherEvent event) { TaskAttemptId taskAttemptID = event.getTaskAttemptID(); // Pass everything except the 2nd attempt of the first task. if (taskAttemptID.getId() != 1 || taskAttemptID.getTaskId().getId() != 0) { super.handle(event); } } };
@Override public int compareTo(TaskAttemptId other) { int taskIdComp = this.getTaskId().compareTo(other.getTaskId()); if (taskIdComp == 0) { return this.getId() - other.getId(); } else { return taskIdComp; } } }
@Override public boolean equals(Object obj) { if (this == obj) return true; if (obj == null) return false; if (getClass() != obj.getClass()) return false; TaskAttemptId other = (TaskAttemptId) obj; if (getId() != other.getId()) return false; if (!getTaskId().equals(other.getTaskId())) return false; return true; }
@Override public int compareTo(TaskId other) { int jobIdComp = this.getJobId().compareTo(other.getJobId()); if (jobIdComp == 0) { if (this.getTaskType() == other.getTaskType()) { return this.getId() - other.getId(); } else { return this.getTaskType().compareTo(other.getTaskType()); } } else { return jobIdComp; } } }
private AtomicInteger containerNeed(TaskId taskID) { JobId jobID = taskID.getJobId(); TaskType taskType = taskID.getTaskType(); ConcurrentMap<JobId, AtomicInteger> relevantMap = taskType == TaskType.MAP ? mapContainerNeeds : reduceContainerNeeds; AtomicInteger result = relevantMap.get(jobID); if (result == null) { relevantMap.putIfAbsent(jobID, new AtomicInteger(0)); result = relevantMap.get(jobID); } return result; }
public static TaskId newTaskId(JobId jobId, int id, TaskType taskType) { TaskId taskId = Records.newRecord(TaskId.class); taskId.setJobId(jobId); taskId.setId(id); taskId.setTaskType(taskType); return taskId; }