| /** |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package org.apache.hadoop.mapred; |
| |
| import java.io.IOException; |
| import java.util.Collections; |
| import java.util.List; |
| import java.util.ArrayList; |
| import java.util.Map; |
| import java.util.HashMap; |
| import java.util.Set; |
| import java.util.HashSet; |
| import java.util.Iterator; |
| |
| import org.apache.commons.logging.Log; |
| import org.apache.commons.logging.LogFactory; |
| import org.apache.hadoop.mapred.TaskStatus.Phase; |
| import org.apache.hadoop.mapred.TaskStatus.State; |
| import org.apache.hadoop.tools.rumen.TaskAttemptInfo; |
| import org.apache.hadoop.tools.rumen.ReduceTaskAttemptInfo; |
| // Explicitly use the new api, older o.a.h.mapred.TaskAttemptID is deprecated |
| import org.apache.hadoop.mapreduce.TaskAttemptID; |
| |
| /** |
| * This class simulates a {@link TaskTracker}. Its main purpose is to call heartbeat() |
| * of the simulated Job Tracker with apropriately updated statuses of the |
| * tasks assigned to it. |
| * |
| * The events emitted and consumed are HeartbeatEvent and |
| * TaskAttemptCompletionEvent . |
| * |
| * Internal naming convention: accept() dispatches simulation events to |
| * process*Event() methods. heartbeat() dispactches task tracker actions to |
| * handle*Action() methods. |
| */ |
| public class SimulatorTaskTracker implements SimulatorEventListener { |
| /** The name of the task tracker. */ |
| protected final String taskTrackerName; |
| /** The name of the host the task tracker is running on. */ |
| protected final String hostName; |
| /** The http port the simulated task tracker reports to the jobtracker. */ |
| protected final int httpPort = 80; |
| /** Number of map slots. */ |
| protected final int maxMapSlots; |
| /** Number of reduce slots. */ |
| protected final int maxReduceSlots; |
| /** The job tracker this task tracker is a slave of. */ |
| protected final InterTrackerProtocol jobTracker; |
| |
| /** |
| * State of and bookkeeping information for all tasks assigned to the task |
| * tracker. Contains all the information about running or completed but |
| * not yet reported tasks. We manage it in a mark & sweep garbage collector |
| * manner. We insert tasks on launch, mark them on completion, and remove |
| * completed tasks at heartbeat() reports. |
| */ |
| protected Map<TaskAttemptID, SimulatorTaskInProgress> tasks = |
| new HashMap<TaskAttemptID, SimulatorTaskInProgress>(); |
| /** |
| * Number of map slots allocated to tasks in RUNNING state on this task |
| * tracker. Must be in sync with the tasks map above. |
| */ |
| private int usedMapSlots = 0; |
| /** |
| * Number of reduce slots allocated to tasks in RUNNING state on this task |
| * tracker. Must be in sync with the tasks map above. |
| */ |
| private int usedReduceSlots = 0; |
| /** |
| * True if the jobTracker.heartbeat() call to be made is the first one. |
| * We need this to mimick the InterTrackerProtocol properly. |
| */ |
| private boolean firstHeartbeat = true; |
| |
| // last heartbeat response recieved |
| private short heartbeatResponseId = -1; |
| |
| /** |
| * Task attempt ids for which TaskAttemptCompletionEvent was created but the |
| * task attempt got killed. |
| */ |
| private Set<TaskAttemptID> orphanTaskCompletions = |
| new HashSet<TaskAttemptID>(); |
| |
| /** The log object to send our messages to; only used for debugging. */ |
| private static final Log LOG = LogFactory.getLog(SimulatorTaskTracker.class); |
| |
| /** |
| * Constructs a task tracker. |
| * |
| * @param jobTracker the SimulatorJobTracker we talk to |
| * @param taskTrackerName the task tracker name to report, otherwise unused |
| * @param hostName the host name to report, otherwise unused |
| * @param maxMapTasks the number of map slots |
| * @param maxReduceTasks the number of reduce slots |
| */ |
| public SimulatorTaskTracker(InterTrackerProtocol jobTracker, |
| String taskTrackerName, String hostName, |
| int maxMapTasks, int maxReduceTasks) { |
| LOG.debug("SimulatorTaskTracker constructor, taskTrackerName=" + |
| taskTrackerName); |
| |
| this.jobTracker = jobTracker; |
| this.taskTrackerName = taskTrackerName; |
| this.hostName = hostName; |
| this.maxMapSlots = maxMapTasks; |
| this.maxReduceSlots = maxReduceTasks; |
| } |
| |
| /** |
| * Processes a simulation event. |
| * |
| * @param event the event to process, should be an instance of HeartbeatEvent |
| * or TaskAttemptCompletionEvent |
| * @return the list of new events generated in response |
| */ |
| @Override |
| public List<SimulatorEvent> accept(SimulatorEvent event) { |
| if (LOG.isDebugEnabled()) { |
| LOG.debug("Accepted event " + event); |
| } |
| |
| if (event instanceof HeartbeatEvent) { |
| return processHeartbeatEvent((HeartbeatEvent)event); |
| } else if (event instanceof TaskAttemptCompletionEvent) { |
| return processTaskAttemptCompletionEvent((TaskAttemptCompletionEvent) |
| event); |
| } else { |
| throw new IllegalArgumentException("Unhandled event "+event); |
| } |
| } |
| |
| /** |
| * Called once at the start of the simulation. |
| * |
| * @param when Time when the task tracker starts. |
| * @return the initial HeartbeatEvent for ourselves. |
| */ |
| public List<SimulatorEvent> init(long when) { |
| LOG.debug("TaskTracker starting up, current simulation time=" + when); |
| |
| return Collections.<SimulatorEvent>singletonList(new HeartbeatEvent(this, when)); |
| } |
| |
| /** |
| * Stops running a task attempt on the task tracker. It also updates the |
| * number of available slots accordingly. |
| * |
| * @param finalStatus the TaskStatus containing the task id and final |
| * status of the task attempt. This rountine asserts a lot of the |
| * finalStatus params, in case it is coming from a task attempt |
| * completion event sent to ourselves. Only the run state, finish time, |
| * and progress fields of the task attempt are updated. |
| * @param now Current simulation time, used for assert only |
| */ |
| private void finishRunningTask(TaskStatus finalStatus, long now) { |
| TaskAttemptID taskId = finalStatus.getTaskID(); |
| if (LOG.isDebugEnabled()) { |
| LOG.debug("Finishing running task id=" + taskId + ", now=" + now); |
| } |
| |
| SimulatorTaskInProgress tip = tasks.get(taskId); |
| if (tip == null) { |
| throw new IllegalArgumentException("Unknown task attempt " + taskId |
| + " completed"); |
| } |
| TaskStatus currentStatus = tip.getTaskStatus(); |
| if (currentStatus.getRunState() != State.RUNNING) { |
| throw new IllegalArgumentException( |
| "Task attempt to finish is not running: " + tip); |
| } |
| |
| // Check that finalStatus describes a task attempt that has just been |
| // completed |
| State finalRunState = finalStatus.getRunState(); |
| if (finalRunState != State.SUCCEEDED && finalRunState != State.FAILED |
| && finalRunState != State.KILLED) { |
| throw new IllegalArgumentException( |
| "Final run state for completed task can't be : " + finalRunState |
| + " " + tip); |
| } |
| |
| if (now != finalStatus.getFinishTime()) { |
| throw new IllegalArgumentException( |
| "Current time does not match task finish time: now=" + now |
| + ", finish=" + finalStatus.getFinishTime()); |
| } |
| |
| if (currentStatus.getIsMap() != finalStatus.getIsMap() |
| || currentStatus.getNumSlots() != finalStatus.getNumSlots() |
| || currentStatus.getPhase() != finalStatus.getPhase() |
| || currentStatus.getStartTime() != finalStatus.getStartTime()) { |
| throw new IllegalArgumentException( |
| "Current status does not match final status"); |
| } |
| |
| // We can't assert getShuffleFinishTime() and getSortFinishTime() for |
| // reduces as those were unknown when the task attempt completion event |
| // was created. We have not called setMapFinishTime() for maps either. |
| // If we were really thorough we could update the progress of the task |
| // and check if it is consistent with finalStatus. |
| |
| // If we've got this far it is safe to update the task status |
| currentStatus.setRunState(finalStatus.getRunState()); |
| currentStatus.setFinishTime(finalStatus.getFinishTime()); |
| currentStatus.setProgress(finalStatus.getProgress()); |
| |
| // and update the free slots |
| int numSlots = currentStatus.getNumSlots(); |
| if (tip.isMapTask()) { |
| usedMapSlots -= numSlots; |
| if (usedMapSlots < 0) { |
| throw new IllegalStateException( |
| "TaskTracker reaches negative map slots: " + usedMapSlots); |
| } |
| } else { |
| usedReduceSlots -= numSlots; |
| if (usedReduceSlots < 0) { |
| throw new IllegalStateException( |
| "TaskTracker reaches negative reduce slots: " + usedReduceSlots); |
| } |
| } |
| } |
| |
| /** |
| * Records that a task attempt has completed. Ignores the event for tasks |
| * that got killed after the creation of the completion event. |
| * |
| * @param event the TaskAttemptCompletionEvent the tracker sent to itself |
| * @return the list of response events, empty |
| */ |
| private List<SimulatorEvent> processTaskAttemptCompletionEvent( |
| TaskAttemptCompletionEvent event) { |
| if (LOG.isDebugEnabled()) { |
| LOG.debug("Processing task attempt completion event" + event); |
| } |
| |
| long now = event.getTimeStamp(); |
| TaskStatus finalStatus = event.getStatus(); |
| TaskAttemptID taskID = finalStatus.getTaskID(); |
| boolean killedEarlier = orphanTaskCompletions.remove(taskID); |
| if (!killedEarlier) { |
| finishRunningTask(finalStatus, now); |
| } |
| return SimulatorEngine.EMPTY_EVENTS; |
| } |
| |
| |
| /** |
| * Creates a signal for itself marking the completion of a task attempt. |
| * It assumes that the task attempt hasn't made any progress in the user |
| * space code so far, i.e. it is called right at launch for map tasks and |
| * immediately after all maps completed for reduce tasks. |
| * |
| * @param tip the simulator task in progress |
| * @param now the current simulation time |
| * @return the TaskAttemptCompletionEvent we are sending to ourselves |
| */ |
| private TaskAttemptCompletionEvent createTaskAttemptCompletionEvent( |
| SimulatorTaskInProgress tip, long now) { |
| // We need to clone() status as we modify and it goes into an Event |
| TaskStatus status = (TaskStatus)tip.getTaskStatus().clone(); |
| long delta = tip.getUserSpaceRunTime(); |
| assert delta >= 0 : "TaskAttempt " + tip.getTaskStatus().getTaskID() |
| + " has negative UserSpaceRunTime = " + delta; |
| long finishTime = now + delta; |
| status.setFinishTime(finishTime); |
| if (tip.isMapTask()) { |
| status.setMapFinishTime(finishTime); |
| } |
| status.setProgress(1.0f); |
| status.setRunState(tip.getFinalRunState()); |
| TaskAttemptCompletionEvent event = |
| new TaskAttemptCompletionEvent(this, status); |
| if (LOG.isDebugEnabled()) { |
| LOG.debug("Created task attempt completion event " + event); |
| } |
| return event; |
| } |
| |
| /** |
| * Launches a task on the simulated task tracker. |
| * |
| * @param action SimulatorLaunchTaskAction sent by the job tracker |
| * @param now current simulation time |
| * @return new events generated, a TaskAttemptCompletionEvent for map |
| * tasks, empty otherwise |
| */ |
| private List<SimulatorEvent> handleSimulatorLaunchTaskAction( |
| SimulatorLaunchTaskAction action, long now) { |
| if (LOG.isDebugEnabled()) { |
| LOG.debug("Handling launch task action " + action); |
| } |
| // First, create statuses and update used slots for map and reduce |
| // task separately |
| Task task = action.getTask(); |
| TaskAttemptID taskId = task.getTaskID(); |
| if (tasks.containsKey(taskId)) { |
| throw new IllegalArgumentException("Multiple launch of task id =" + taskId); |
| } |
| |
| // Ctor of MapTaskStatus and ReduceTaskStatus need deprecated |
| // o.a.h.mapred.TaskAttemptID, hence the downgrade |
| org.apache.hadoop.mapred.TaskAttemptID taskIdOldApi = |
| org.apache.hadoop.mapred.TaskAttemptID.downgrade(taskId); |
| TaskStatus status; |
| int numSlotsRequired = task.getNumSlotsRequired(); |
| Counters emptyCounters = new Counters(); |
| if (task.isMapTask()) { |
| status = new MapTaskStatus(taskIdOldApi, 0f, numSlotsRequired, |
| State.RUNNING, "", "", taskTrackerName, |
| Phase.MAP, emptyCounters); |
| usedMapSlots += numSlotsRequired; |
| if (usedMapSlots > maxMapSlots) { |
| throw new IllegalStateException("usedMapSlots exceeds maxMapSlots: " + |
| usedMapSlots + " > " + maxMapSlots); |
| } |
| } else { |
| status = new ReduceTaskStatus(taskIdOldApi, 0f, numSlotsRequired, |
| State.RUNNING, "", "", taskTrackerName, |
| Phase.SHUFFLE, emptyCounters); |
| usedReduceSlots += numSlotsRequired; |
| if (usedReduceSlots > maxReduceSlots) { |
| throw new IllegalStateException("usedReduceSlots exceeds usedReduceSlots: " + |
| usedReduceSlots + " > " + usedReduceSlots); |
| } |
| } |
| |
| // Second, create and store a TIP |
| status.setStartTime(now); |
| SimulatorTaskInProgress tip = |
| new SimulatorTaskInProgress(action, status, now); |
| tasks.put(taskId, tip); |
| |
| // Third, schedule events for ourselves |
| if (task.isMapTask()) { |
| // we know when this task attempts ends iff it is a map |
| TaskAttemptCompletionEvent e = createTaskAttemptCompletionEvent(tip, now); |
| return Collections.<SimulatorEvent>singletonList(e); |
| } else { |
| // reduce, completion time can only be determined when all maps are done |
| return SimulatorEngine.EMPTY_EVENTS; |
| } |
| } |
| |
| /** |
| * Kills a task attempt. |
| * |
| * @param action contains the task attempt to kill |
| * @param now current simulation time |
| * @return new events generated in response, empty |
| */ |
| private List<SimulatorEvent> handleKillTaskAction(KillTaskAction action, long now) { |
| TaskAttemptID taskId = action.getTaskID(); |
| // we don't have a nice(r) toString() in Hadoop's TaskActions |
| if (LOG.isDebugEnabled()) { |
| LOG.debug("Handling kill task action, taskId=" + taskId + ", now=" + now); |
| } |
| |
| SimulatorTaskInProgress tip = tasks.get(taskId); |
| |
| // Safety check: We might get a KillTaskAction even for completed reduces |
| if (tip == null) { |
| return SimulatorEngine.EMPTY_EVENTS; |
| } |
| |
| progressTaskStatus(tip, now); // make progress up to date |
| TaskStatus finalStatus = (TaskStatus)tip.getTaskStatus().clone(); |
| finalStatus.setFinishTime(now); |
| finalStatus.setRunState(State.KILLED); |
| finishRunningTask(finalStatus, now); |
| |
| if (finalStatus.getIsMap() || finalStatus.getPhase() == Phase.REDUCE) { |
| // if we have already created a task attempt completion event we remember |
| // the task id, so that we can safely ignore the event when its delivered |
| orphanTaskCompletions.add(taskId); |
| } |
| return SimulatorEngine.EMPTY_EVENTS; |
| } |
| |
| /** |
| * Starts "running" the REDUCE phase of reduce upon being notified that |
| * all map tasks are (successfully) done. |
| * |
| * @param action contains the notification for one of the reduce tasks |
| * @param now current simulation time |
| * @return new events generated, a single TaskAttemptCompletionEvent for the |
| * reduce |
| */ |
| private List<SimulatorEvent> handleAllMapsCompletedTaskAction( |
| AllMapsCompletedTaskAction action, long now) { |
| if (LOG.isDebugEnabled()) { |
| LOG.debug("Handling all maps completed task action " + action); |
| } |
| |
| TaskAttemptID taskId = action.getTaskID(); |
| SimulatorTaskInProgress tip = tasks.get(taskId); |
| // If tip is null here it is because the task attempt to be notified is |
| // unknown to this TaskTracker. |
| TaskStatus status = tip.getTaskStatus(); |
| if (status.getIsMap()) { |
| throw new IllegalStateException( |
| "Invalid AllMapsCompletedTaskAction, task attempt " |
| + "to be notified is a map: " + taskId + " " + status); |
| } |
| if (status.getPhase() != Phase.SHUFFLE) { |
| throw new IllegalArgumentException( |
| "Reducer task attempt already notified: " + taskId + " " + status); |
| } |
| |
| // Warning: setPhase() uses System.currentTimeMillis() internally to |
| // set shuffle and sort times, but we overwrite that manually anyway |
| status.setPhase(Phase.REDUCE); |
| status.setShuffleFinishTime(now); |
| status.setSortFinishTime(now); |
| |
| // Forecast the completion of this reduce |
| TaskAttemptCompletionEvent e = createTaskAttemptCompletionEvent(tip, now); |
| return Collections.<SimulatorEvent>singletonList(e); |
| } |
| |
| /** |
| * Updates the progress indicator of a task if it is running. |
| * |
| * @param tip simulator task in progress whose progress is to be updated |
| * @param now current simulation time |
| */ |
| private void progressTaskStatus(SimulatorTaskInProgress tip, long now) { |
| TaskStatus status = tip.getTaskStatus(); |
| if (status.getRunState() != State.RUNNING) { |
| return; // nothing to be done |
| } |
| |
| boolean isMap = tip.isMapTask(); |
| // Time when the user space code started |
| long startTime = -1; |
| // Time spent in map or just in the REDUCE phase of a reduce task |
| long runTime = tip.getUserSpaceRunTime(); |
| float progress = 0.0f; |
| if (isMap) { |
| // We linearly estimate the progress of maps since their start |
| startTime = status.getStartTime(); |
| progress = ((float)(now - startTime)) / runTime; |
| } else { |
| // We don't model reduce progress in the SHUFFLE or SORT phases |
| // We use linear estimate for the 3rd, REDUCE phase |
| Phase reducePhase = status.getPhase(); |
| switch (reducePhase) { |
| case SHUFFLE: |
| progress = 0.0f; // 0 phase is done out of 3 |
| break; |
| case SORT: |
| progress = 1.0f/3; // 1 phase is done out of 3 |
| break; |
| case REDUCE: { |
| // REDUCE phase with the user code started when sort finished |
| startTime = status.getSortFinishTime(); |
| // 0.66f : 2 phases are done out of of 3 |
| progress = 2.0f/3 + (((float) (now - startTime)) / runTime) / 3.0f; |
| } |
| break; |
| default: |
| // should never get here |
| throw new IllegalArgumentException("Invalid reducePhase=" + reducePhase); |
| } |
| } |
| |
| final float EPSILON = 0.0001f; |
| if (progress < -EPSILON || progress > 1 + EPSILON) { |
| throw new IllegalStateException("Task progress out of range: " + progress); |
| } |
| progress = Math.max(Math.min(1.0f, progress), 0.0f); |
| status.setProgress(progress); |
| |
| if (LOG.isDebugEnabled()) { |
| LOG.debug("Updated task progress, taskId=" + status.getTaskID() |
| + ", progress=" + status.getProgress()); |
| } |
| } |
| |
| /** |
| * Updates the progress indicator of all running tasks. |
| * |
| * @param now current simulation time |
| */ |
| private void progressTaskStatuses(long now) { |
| for (SimulatorTaskInProgress tip : tasks.values()) { |
| progressTaskStatus(tip, now); |
| } |
| } |
| |
| /** |
| * Frees up bookkeping memory used by completed tasks. |
| * Has no effect on the events or logs produced by the SimulatorTaskTracker. |
| * We need this in order not to report completed task multiple times and |
| * to ensure that we do not run out of Java heap memory in larger |
| * simulations. |
| */ |
| private void garbageCollectCompletedTasks() { |
| for (Iterator<TaskAttemptID> iter = tasks.keySet().iterator(); |
| iter.hasNext();) { |
| TaskAttemptID taskId = iter.next(); |
| SimulatorTaskInProgress tip = tasks.get(taskId); |
| if (tip.getTaskStatus().getRunState() != State.RUNNING) { |
| iter.remove(); |
| if (LOG.isDebugEnabled()) { |
| LOG.debug("Garbage collected SimulatorTIP, taskId=" + taskId); |
| } |
| // We don't have to / must not touch usedMapSlots and usedReduceSlots |
| // as those were already updated by processTaskAttemptCompletionEvent() |
| // when the task switched its state from running |
| } |
| } |
| } |
| |
| /** |
| * Creates a list of task statuses suitable for transmission via heartbeat(). |
| * The task statuses are cloned() so that the heartbeat() callee, the job |
| * tracker, can't mess up the SimulatorTaskTracker's internal data. |
| * |
| * @return the list of running and recently completed task statuses |
| * on the tracker |
| */ |
| private List<TaskStatus> collectAndCloneTaskStatuses() { |
| ArrayList<TaskStatus> statuses = new ArrayList<TaskStatus>(); |
| for (SimulatorTaskInProgress tip : tasks.values()) { |
| statuses.add((TaskStatus)tip.getTaskStatus().clone()); |
| } |
| return statuses; |
| } |
| |
| /** |
| * Handles the HeartbeatResponse received from the job tracker upon |
| * heartbeat(). Dispatches to handle*Action() methods. |
| * |
| * @param response HeartbeatResponse received from the job tracker |
| * @param now current simulation time |
| * @return list of new events generated in response to the task actions |
| */ |
| private List<SimulatorEvent> handleHeartbeatResponse(HeartbeatResponse response, |
| long now) { |
| if (LOG.isDebugEnabled()) { |
| LOG.debug("Handling heartbeat response " + response); |
| } |
| |
| List<SimulatorEvent> events = new ArrayList<SimulatorEvent>(); |
| TaskTrackerAction[] actions = response.getActions(); |
| for (TaskTrackerAction action : actions) { |
| List<SimulatorEvent> actionEvents; |
| if (action instanceof SimulatorLaunchTaskAction) { |
| actionEvents = handleSimulatorLaunchTaskAction( |
| (SimulatorLaunchTaskAction)action, now); |
| } else if(action instanceof KillTaskAction) { |
| actionEvents = handleKillTaskAction((KillTaskAction)action, now); |
| } else if(action instanceof AllMapsCompletedTaskAction) { |
| // our extra task action for notifying the reducers |
| actionEvents = handleAllMapsCompletedTaskAction( |
| (AllMapsCompletedTaskAction)action, now); |
| } else { |
| // Should never get here. |
| // CommitTaskAction is not implemented in the simulator |
| // LaunchTaskAction has to be SimulatorLaunchTaskAction |
| throw new UnsupportedOperationException("Unimplemented TaskAction: " |
| + action); |
| } |
| events.addAll(actionEvents); |
| } |
| return events; |
| } |
| |
| /** |
| * Transmits a heartbeat event to the jobtracker and processes the response. |
| * |
| * @param event HeartbeatEvent to process |
| * @return list of new events generated in response |
| */ |
| private List<SimulatorEvent> processHeartbeatEvent(HeartbeatEvent event) { |
| if (LOG.isDebugEnabled()) { |
| LOG.debug("Processing heartbeat event " + event); |
| } |
| |
| long now = event.getTimeStamp(); |
| |
| // Create the TaskTrackerStatus to report |
| progressTaskStatuses(now); |
| List<TaskStatus> taskStatuses = collectAndCloneTaskStatuses(); |
| boolean askForNewTask = (usedMapSlots < maxMapSlots || |
| usedReduceSlots < maxReduceSlots); |
| |
| // 0 means failures == 0 here. Undocumented in TaskTracker, but does not |
| // seem to be used at all in org.apache.hadoop.mapred . |
| TaskTrackerStatus taskTrackerStatus = |
| new SimulatorTaskTrackerStatus(taskTrackerName, hostName, httpPort, |
| taskStatuses, 0, |
| maxMapSlots, maxReduceSlots, now); |
| |
| // This is the right, and only, place to release bookkeping memory held |
| // by completed tasks: after collectAndCloneTaskStatuses() and before |
| // heartbeat(). |
| // The status of TIPs to be purged is already cloned & copied to |
| // taskStatuses for reporting |
| // We shouldn't run the gc after heartbeat() since KillTaskAction might |
| // produce new completed tasks that we have not yet reported back and |
| // don't want to purge immediately. |
| garbageCollectCompletedTasks(); |
| |
| // Transmit the heartbeat |
| HeartbeatResponse response = null; |
| try { |
| response = |
| jobTracker.heartbeat(taskTrackerStatus, false, firstHeartbeat, |
| askForNewTask, heartbeatResponseId); |
| } catch (IOException ioe) { |
| throw new IllegalStateException("Internal error", ioe); |
| } |
| firstHeartbeat = false; |
| |
| // The heartbeat got through successfully! |
| heartbeatResponseId = response.getResponseId(); |
| |
| // Process the heartbeat response |
| List<SimulatorEvent> events = handleHeartbeatResponse(response, now); |
| |
| // Next heartbeat |
| events.add(new HeartbeatEvent(this, now + response.getHeartbeatInterval())); |
| |
| return events; |
| } |
| |
| /** |
| * Internal helper class used for storing the current status and other |
| * auxilliary information associated with a task attempt assigned to |
| * a simulator task tracker. |
| * WARNING: This is a completely different inner class than the one with |
| * the same name in SimulatorJobTracker. |
| */ |
| static class SimulatorTaskInProgress { |
| |
| /** |
| * Current status of the task attempt. |
| * We store the start time, the start time of reduce phases and the |
| * run state of the task in this object. |
| */ |
| private TaskStatus taskStatus; |
| |
| /** |
| * Object storing the run time and the final state of the task attempt. |
| * It is never read directly by the SimulatorTaskTracker. |
| */ |
| private TaskAttemptInfo taskAttempInfo; |
| |
| /** |
| * Runtime of the user-space code of the task attempt. This is the full |
| * runtime for map tasks, and only that of the REDUCE phase for reduce |
| * tasks. |
| */ |
| private final long userSpaceRunTime; |
| |
| /** |
| * Constructs an object by copying most of the fields from a |
| * SimulatorTaskAction. |
| */ |
| public SimulatorTaskInProgress(SimulatorLaunchTaskAction action, |
| TaskStatus taskStatus, long now) { |
| this.taskStatus = taskStatus; |
| this.taskAttempInfo = action.getTaskAttemptInfo(); |
| if (taskStatus.getIsMap()) { |
| this.userSpaceRunTime = taskAttempInfo.getRuntime(); |
| } else { |
| this.userSpaceRunTime = |
| ((ReduceTaskAttemptInfo)taskAttempInfo).getReduceRuntime(); |
| } |
| } |
| |
| /** |
| * Returns whether the task attempt is a map. |
| * |
| * @return true iff the task attempt is a map |
| */ |
| public boolean isMapTask() { |
| return taskStatus.getIsMap(); |
| } |
| |
| /* |
| * Returns the current status of the task attempt. |
| * |
| * @return current task status |
| */ |
| public TaskStatus getTaskStatus() { |
| return taskStatus; |
| } |
| |
| /** |
| * Sets the status of the task attempt. |
| * |
| * @param status the new task status |
| */ |
| public void setTaskStatus(TaskStatus status) { |
| this.taskStatus = status; |
| } |
| |
| /** |
| * Returns the final state of the completed task. |
| * |
| * @return the final state of the completed task; |
| * it is either State.SUCCEEDED or State.FAILED |
| */ |
| public State getFinalRunState() { |
| return taskAttempInfo.getRunState(); |
| } |
| |
| /** |
| * Gets the time spent in the user space code of the task attempt. |
| * This is the full runtime for map tasks, and only that of the REDUCE |
| * phase for reduce tasks. |
| * |
| * @return the user space runtime |
| */ |
| public long getUserSpaceRunTime() { |
| return userSpaceRunTime; |
| } |
| } |
| } |
