org.apache.flink.runtime.resourcemanager.slotmanager

Class DefaultSlotTracker

java.lang.Object

org.apache.flink.runtime.resourcemanager.slotmanager.DefaultSlotTracker

public class DefaultSlotTracker
extends Object

Default SlotTracker implementation.

Constructor Summary

Constructors

Constructor and Description
DefaultSlotTracker()

Method Summary

Modifier and Type	Method and Description
void	addSlot(SlotID slotId, ResourceProfile resourceProfile, TaskExecutorConnection taskManagerConnection, JobID assignedJob) Adds the given slot to this tracker.
Collection<TaskManagerSlotInformation>	getFreeSlots() Returns a view over free slots.
Collection<TaskExecutorConnection>	getTaskExecutorsWithAllocatedSlotsForJob(JobID jobId) Returns all task executors that have at least 1 pending/completed allocation for the given job.
void	notifyAllocationComplete(SlotID slotId, JobID jobId) Notifies the tracker that the allocation for the given slot, for the given job, has completed successfully.
void	notifyAllocationStart(SlotID slotId, JobID jobId) Notifies the tracker that the allocation for the given slot, for the given job, has started.
void	notifyFree(SlotID slotId) Notifies the tracker that the given slot was freed.
boolean	notifySlotStatus(Iterable<SlotStatus> slotStatuses) Notifies the tracker about the slot statuses.
void	registerSlotStatusUpdateListener(org.apache.flink.runtime.resourcemanager.slotmanager.SlotStatusUpdateListener slotStatusUpdateListener) Registers the given listener with this tracker.
void	removeSlots(Iterable<SlotID> slotsToRemove) Removes the given set of slots from the slot manager.

Methods inherited from class java.lang.Object

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

Constructor Detail

DefaultSlotTracker

public DefaultSlotTracker()

Method Detail

registerSlotStatusUpdateListener

public void registerSlotStatusUpdateListener(org.apache.flink.runtime.resourcemanager.slotmanager.SlotStatusUpdateListener slotStatusUpdateListener)

Registers the given listener with this tracker.

Parameters:

slotStatusUpdateListener - listener to register

addSlot

public void addSlot(SlotID slotId,
                    ResourceProfile resourceProfile,
                    TaskExecutorConnection taskManagerConnection,
                    @Nullable
                    JobID assignedJob)

Adds the given slot to this tracker. The given slot may already be allocated for a job. This method must be called before the tracker is notified of any state transition or slot status notification.

Parameters:

slotId - ID of the slot

resourceProfile - resource of the slot

taskManagerConnection - connection to the hosting task executor

assignedJob - job that the slot is allocated for, or null if it is free

removeSlots

public void removeSlots(Iterable<SlotID> slotsToRemove)

Removes the given set of slots from the slot manager. If a removed slot was not free at the time of removal, then this method will automatically transition the slot to a free state.

Parameters:

slotsToRemove - identifying the slots to remove from the slot manager

notifyFree

public void notifyFree(SlotID slotId)

Notifies the tracker that the given slot was freed.

Parameters:

slotId - slot being freed

notifyAllocationStart

public void notifyAllocationStart(SlotID slotId,
                                  JobID jobId)

Notifies the tracker that the allocation for the given slot, for the given job, has started.

Parameters:

slotId - slot being allocated

jobId - job for which the slot is being allocated

notifyAllocationComplete

public void notifyAllocationComplete(SlotID slotId,
                                     JobID jobId)

Notifies the tracker that the allocation for the given slot, for the given job, has completed successfully.

Parameters:

slotId - slot being allocated

jobId - job for which the slot is being allocated

notifySlotStatus

public boolean notifySlotStatus(Iterable<SlotStatus> slotStatuses)

Notifies the tracker about the slot statuses.

Parameters:

slotStatuses - slot statues

Returns:

whether any slot status has changed

getFreeSlots

public Collection<TaskManagerSlotInformation> getFreeSlots()

Returns a view over free slots. The returned collection cannot be modified directly, but reflects changes to the set of free slots.

Returns:

free slots

getTaskExecutorsWithAllocatedSlotsForJob

public Collection<TaskExecutorConnection> getTaskExecutorsWithAllocatedSlotsForJob(JobID jobId)

Returns all task executors that have at least 1 pending/completed allocation for the given job.

Parameters:

jobId - the job for which the task executors must have a slot

Returns:

task executors with at least 1 slot for the job