edu.snu.coral.runtime.master.scheduler

Class SingleJobTaskGroupQueue

java.lang.Object

edu.snu.coral.runtime.master.scheduler.SingleJobTaskGroupQueue

All Implemented Interfaces:

PendingTaskGroupQueue

@ThreadSafe
public final class SingleJobTaskGroupQueue
extends Object
implements PendingTaskGroupQueue

Keep tracks of all pending task groups. This class provides two-level queue scheduling by prioritizing TaskGroups of certain stages to be scheduled first. Stages that are mutually independent alternate turns in scheduling each of their TaskGroups. This PQ assumes that stages/task groups of higher priorities are never enqueued without first removing those of lower priorities (which is how Scheduler behaves) for simplicity.

Constructor Summary

Constructors

Constructor and Description
SingleJobTaskGroupQueue()

Method Summary

Modifier and Type	Method and Description
void	close() Closes and cleans up this queue.
Optional<ScheduledTaskGroup>	dequeue() Dequeues the next TaskGroup to be scheduled according to job dependency priority.
void	enqueue(ScheduledTaskGroup scheduledTaskGroup) Enqueues a TaskGroup to this PQ.
boolean	isEmpty() Checks whether there are schedulable TaskGroups in the queue or not.
void	onJobScheduled(PhysicalPlan physicalPlanForJob) Registers a job to this queue in case the queue needs to understand the topology of the job DAG.
void	removeTaskGroupsAndDescendants(String stageId) Removes a stage and its descendant stages from this PQ.

Methods inherited from class java.lang.Object

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

Constructor Detail

SingleJobTaskGroupQueue

@Inject
public SingleJobTaskGroupQueue()

Method Detail

enqueue

public void enqueue(ScheduledTaskGroup scheduledTaskGroup)

Description copied from interface: PendingTaskGroupQueue

Enqueues a TaskGroup to this PQ.

Specified by:

enqueue in interface PendingTaskGroupQueue

Parameters:

scheduledTaskGroup - to enqueue.

dequeue

public Optional<ScheduledTaskGroup> dequeue()

Dequeues the next TaskGroup to be scheduled according to job dependency priority.

Specified by:

dequeue in interface PendingTaskGroupQueue

Returns:

the next TaskGroup to be scheduled

removeTaskGroupsAndDescendants

public void removeTaskGroupsAndDescendants(String stageId)

Removes a stage and its descendant stages from this PQ.

Specified by:

removeTaskGroupsAndDescendants in interface PendingTaskGroupQueue

Parameters:

stageId - for the stage to begin the removal recursively.

onJobScheduled

public void onJobScheduled(PhysicalPlan physicalPlanForJob)

Description copied from interface: PendingTaskGroupQueue

Registers a job to this queue in case the queue needs to understand the topology of the job DAG.

Specified by:

onJobScheduled in interface PendingTaskGroupQueue

Parameters:

physicalPlanForJob - the job to schedule.

isEmpty

public boolean isEmpty()

Description copied from interface: PendingTaskGroupQueue

Checks whether there are schedulable TaskGroups in the queue or not.

Specified by:

isEmpty in interface PendingTaskGroupQueue

Returns:

true if there are schedulable TaskGroups in the queue, false otherwise.

close

public void close()

Description copied from interface: PendingTaskGroupQueue

Closes and cleans up this queue.

Specified by:

close in interface PendingTaskGroupQueue