org.apache.flink.runtime.resourcemanager.slotmanager

Class SlotManagerImpl

java.lang.Object

org.apache.flink.runtime.resourcemanager.slotmanager.SlotManagerImpl

All Implemented Interfaces:

AutoCloseable, SlotManager

public class SlotManagerImpl
extends Object
implements SlotManager

Implementation of SlotManager.

Constructor Summary

Constructors

Constructor and Description
SlotManagerImpl(SlotMatchingStrategy slotMatchingStrategy, ScheduledExecutor scheduledExecutor, Time taskManagerRequestTimeout, Time slotRequestTimeout, Time taskManagerTimeout, boolean waitResultConsumedBeforeRelease)

Method Summary

Modifier and Type	Method and Description
void	close() Closes the slot manager.
void	freeSlot(SlotID slotId, AllocationID allocationId) Free the given slot from the given allocation.
int	getNumberAssignedPendingTaskManagerSlots()
int	getNumberFreeSlots()
int	getNumberFreeSlotsOf(InstanceID instanceId)
int	getNumberPendingSlotRequests()
int	getNumberPendingTaskManagerSlots()
int	getNumberRegisteredSlots()
int	getNumberRegisteredSlotsOf(InstanceID instanceId)
boolean	registerSlotRequest(SlotRequest slotRequest) Requests a slot with the respective resource profile.
void	registerTaskManager(TaskExecutorConnection taskExecutorConnection, SlotReport initialSlotReport) Registers a new task manager at the slot manager.
boolean	reportSlotStatus(InstanceID instanceId, SlotReport slotReport) Reports the current slot allocations for a task manager identified by the given instance id.
void	setFailUnfulfillableRequest(boolean failUnfulfillableRequest)
void	start(ResourceManagerId newResourceManagerId, Executor newMainThreadExecutor, ResourceActions newResourceActions) Starts the slot manager with the given leader id and resource manager actions.
void	suspend() Suspends the component.
boolean	unregisterSlotRequest(AllocationID allocationId) Cancels and removes a pending slot request with the given allocation id.
boolean	unregisterTaskManager(InstanceID instanceId, Exception cause) Unregisters the task manager identified by the given instance id and its associated slots from the slot manager.
void	unregisterTaskManagersAndReleaseResources()

Methods inherited from class java.lang.Object

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

Constructor Detail

SlotManagerImpl

public SlotManagerImpl(SlotMatchingStrategy slotMatchingStrategy,
                       ScheduledExecutor scheduledExecutor,
                       Time taskManagerRequestTimeout,
                       Time slotRequestTimeout,
                       Time taskManagerTimeout,
                       boolean waitResultConsumedBeforeRelease)

Method Detail

getNumberRegisteredSlots

public int getNumberRegisteredSlots()

Specified by:

getNumberRegisteredSlots in interface SlotManager

getNumberRegisteredSlotsOf

public int getNumberRegisteredSlotsOf(InstanceID instanceId)

Specified by:

getNumberRegisteredSlotsOf in interface SlotManager

getNumberFreeSlots

public int getNumberFreeSlots()

Specified by:

getNumberFreeSlots in interface SlotManager

getNumberFreeSlotsOf

public int getNumberFreeSlotsOf(InstanceID instanceId)

Specified by:

getNumberFreeSlotsOf in interface SlotManager

getNumberPendingTaskManagerSlots

public int getNumberPendingTaskManagerSlots()

Specified by:

getNumberPendingTaskManagerSlots in interface SlotManager

getNumberPendingSlotRequests

public int getNumberPendingSlotRequests()

Specified by:

getNumberPendingSlotRequests in interface SlotManager

getNumberAssignedPendingTaskManagerSlots

@VisibleForTesting
public int getNumberAssignedPendingTaskManagerSlots()

start

public void start(ResourceManagerId newResourceManagerId,
                  Executor newMainThreadExecutor,
                  ResourceActions newResourceActions)

Starts the slot manager with the given leader id and resource manager actions.

Specified by:

start in interface SlotManager

Parameters:

newResourceManagerId - to use for communication with the task managers

newMainThreadExecutor - to use to run code in the ResourceManager's main thread

newResourceActions - to use for resource (de-)allocations

suspend

public void suspend()

Suspends the component. This clears the internal state of the slot manager.

Specified by:

suspend in interface SlotManager

close

public void close()
           throws Exception

Closes the slot manager.

Specified by:

close in interface AutoCloseable

Throws:

Exception - if the close operation fails

registerSlotRequest

public boolean registerSlotRequest(SlotRequest slotRequest)
                            throws ResourceManagerException

Requests a slot with the respective resource profile.

Specified by:

registerSlotRequest in interface SlotManager

Parameters:

slotRequest - specifying the requested slot specs

Returns:

true if the slot request was registered; false if the request is a duplicate

Throws:

ResourceManagerException - if the slot request failed (e.g. not enough resources left)

unregisterSlotRequest

public boolean unregisterSlotRequest(AllocationID allocationId)

Cancels and removes a pending slot request with the given allocation id. If there is no such pending request, then nothing is done.

Specified by:

unregisterSlotRequest in interface SlotManager

Parameters:

allocationId - identifying the pending slot request

Returns:

True if a pending slot request was found; otherwise false

registerTaskManager

public void registerTaskManager(TaskExecutorConnection taskExecutorConnection,
                                SlotReport initialSlotReport)

Registers a new task manager at the slot manager. This will make the task managers slots known and, thus, available for allocation.

Specified by:

registerTaskManager in interface SlotManager

Parameters:

taskExecutorConnection - for the new task manager

initialSlotReport - for the new task manager

unregisterTaskManager

public boolean unregisterTaskManager(InstanceID instanceId,
                                     Exception cause)

Description copied from interface: SlotManager

Unregisters the task manager identified by the given instance id and its associated slots from the slot manager.

Specified by:

unregisterTaskManager in interface SlotManager

Parameters:

instanceId - identifying the task manager to unregister

cause - for unregistering the TaskManager

Returns:

True if there existed a registered task manager with the given instance id

reportSlotStatus

public boolean reportSlotStatus(InstanceID instanceId,
                                SlotReport slotReport)

Reports the current slot allocations for a task manager identified by the given instance id.

Specified by:

reportSlotStatus in interface SlotManager

Parameters:

instanceId - identifying the task manager for which to report the slot status

slotReport - containing the status for all of its slots

Returns:

true if the slot status has been updated successfully, otherwise false

freeSlot

public void freeSlot(SlotID slotId,
                     AllocationID allocationId)

Free the given slot from the given allocation. If the slot is still allocated by the given allocation id, then the slot will be marked as free and will be subject to new slot requests.

Specified by:

freeSlot in interface SlotManager

Parameters:

slotId - identifying the slot to free

allocationId - with which the slot is presumably allocated

setFailUnfulfillableRequest

public void setFailUnfulfillableRequest(boolean failUnfulfillableRequest)

Specified by:

setFailUnfulfillableRequest in interface SlotManager

unregisterTaskManagersAndReleaseResources

@VisibleForTesting
public void unregisterTaskManagersAndReleaseResources()

Specified by:

unregisterTaskManagersAndReleaseResources in interface SlotManager