edu.snu.coral.runtime.master

Class JobStateManager

java.lang.Object

edu.snu.coral.runtime.master.JobStateManager

public final class JobStateManager
extends Object

Manages the states related to a job. This class can be used to track a job's execution status to task level in the future. The methods of this class are synchronized.

Constructor Summary

Constructors

Constructor and Description
JobStateManager(PhysicalPlan physicalPlan, BlockManagerMaster blockManagerMaster, MetricMessageHandler metricMessageHandler, int maxScheduleAttempt)

Method Summary

Modifier and Type	Method and Description
boolean	checkJobTermination()
boolean	checkStageCompletion(String stageId)
int	getAttemptCountForStage(String stageId)
Map<String,StageState>	getIdToStageStates()
Map<String,TaskGroupState>	getIdToTaskGroupStates()
String	getJobId()
JobState	getJobState()
StageState	getStageState(String stageId)
TaskGroupState	getTaskGroupState(String taskGroupId)
void	onJobStateChanged(JobState.State newState) Updates the state of the job.
void	onStageStateChanged(String stageId, StageState.State newState) Updates the state of a stage.
void	onTaskGroupStateChanged(String taskGroupId, TaskGroupState.State newState) Updates the state of a task group.
void	storeJSON(String directory, String suffix) Stores JSON representation of job state into a file.
String	toString()
String	toStringWithPhysicalPlan()
JobState	waitUntilFinish() Wait for this job to be finished and return the final state.
JobState	waitUntilFinish(long timeout, TimeUnit unit) Wait for this job to be finished and return the final state.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Constructor Detail

JobStateManager

public JobStateManager(PhysicalPlan physicalPlan,
                       BlockManagerMaster blockManagerMaster,
                       MetricMessageHandler metricMessageHandler,
                       int maxScheduleAttempt)

Method Detail

onJobStateChanged

public void onJobStateChanged(JobState.State newState)

Updates the state of the job.

Parameters:

newState - of the job.

onStageStateChanged

public void onStageStateChanged(String stageId,
                                StageState.State newState)

Updates the state of a stage. Stage state changes only occur in master.

Parameters:

stageId - of the stage.

newState - of the stage.

onTaskGroupStateChanged

public void onTaskGroupStateChanged(String taskGroupId,
                                    TaskGroupState.State newState)

Updates the state of a task group. Task group state changes can occur both in master and executor. State changes that occur in master are initiated in BatchSingleJobScheduler. State changes that occur in executors are sent to master as a control message, and the call to this method is initiated in BatchSingleJobScheduler when the message/event is received. A task group completion implies completion of all its tasks.

Parameters:

taskGroupId - the ID of the task group.

newState - the new state of the task group.

checkStageCompletion

public boolean checkStageCompletion(String stageId)

checkJobTermination

public boolean checkJobTermination()

getAttemptCountForStage

public int getAttemptCountForStage(String stageId)

waitUntilFinish

public JobState waitUntilFinish()

Wait for this job to be finished and return the final state.

Returns:

the final state of this job.

waitUntilFinish

public JobState waitUntilFinish(long timeout,
                                TimeUnit unit)

Wait for this job to be finished and return the final state. It wait for at most the given time.

Parameters:

timeout - of waiting.

unit - of the timeout.

Returns:

the final state of this job.

getJobId

public String getJobId()

getJobState

public JobState getJobState()

getStageState

public StageState getStageState(String stageId)

getIdToStageStates

public Map<String,StageState> getIdToStageStates()

getTaskGroupState

public TaskGroupState getTaskGroupState(String taskGroupId)

getIdToTaskGroupStates

public Map<String,TaskGroupState> getIdToTaskGroupStates()

storeJSON

public void storeJSON(String directory,
                      String suffix)

Stores JSON representation of job state into a file.

Parameters:

directory - the directory which JSON representation is saved to

suffix - suffix for file name

toStringWithPhysicalPlan

public String toStringWithPhysicalPlan()

toString

public String toString()

Overrides:

toString in class Object