org.apache.ambari.server.actionmanager

Class ActionDBAccessorImpl

java.lang.Object

org.apache.ambari.server.actionmanager.ActionDBAccessorImpl

All Implemented Interfaces:

ActionDBAccessor

public class ActionDBAccessorImpl
extends Object
implements ActionDBAccessor

Constructor Summary

Constructors

Constructor and Description
ActionDBAccessorImpl(long cacheLimit, AmbariEventPublisher eventPublisher)

Method Summary

Modifier and Type	Method and Description
void	abortHostRole(String host, long requestId, long stageId, String role) Mark the task as to have been aborted
void	abortHostRole(String host, long requestId, long stageId, String role, String reason) Mark the task as to have been aborted.
Collection<HostRoleCommandEntity>	abortOperation(long requestId) Abort all outstanding operations associated with the given request.
void	bulkAbortHostRole(Stage s, Map<ExecutionCommand,String> commands) Bulk abort commands
void	bulkHostRoleScheduled(Stage s, List<ExecutionCommand> commands) Bulk update scheduled commands
void	endRequest(long requestId)
void	endRequestIfCompleted(long requestId)
List<Stage>	getAllStages(long requestId) Get all stages associated with a single request id
List<HostRoleCommand>	getAllTasksByRequestIds(Collection<Long> requestIds) Given a list of request ids, get all the tasks that belong to these requests
int	getCommandsInProgressCount() Gets the number of commands in progress.
List<Stage>	getFirstStageInProgressPerRequest() Returns the next stage which is in-progress for every in-progress request in the system.
long	getLastPersistedRequestIdWhenInitialized() Return the last persisted Request ID as seen when the DBAccessor object was initialized.
Request	getRequest(long requestId) Get request object by id.
Map<Long,String>	getRequestContext(List<Long> requestIds) Gets request contexts associated with the list of request id
String	getRequestContext(long requestId) Gets the request context associated with the request id
RequestEntity	getRequestEntity(long requestId) Gets the request entity by id.
List<Request>	getRequests(Collection<Long> requestIds) Gets request objects by ids
List<Long>	getRequestsByStatus(RequestStatus status, int maxResults, boolean ascOrder) Get first or last maxResults requests that are in the specified status
List<HostRoleCommand>	getRequestTasks(long requestId) Given a request id, get all the tasks that belong to this request
Stage	getStage(String actionId) Given an action id of the form requestId-stageId, retrieve the Stage
List<Stage>	getStagesInProgressForRequest(Long requestId) Returns all the pending stages in a request, including queued and not-queued.
HostRoleCommand	getTask(long taskId) Gets the host role command corresponding to the task id
List<HostRoleCommand>	getTasks(Collection<Long> taskIds) Given a list of task ids, get all the host role commands
List<HostRoleCommand>	getTasksByHostRoleAndStatus(String hostname, String role, HostRoleStatus status) Get a List of host role commands where the host, role and status are as specified
List<HostRoleCommand>	getTasksByRoleAndStatus(String role, HostRoleStatus status) Get a List of host role commands where the role and status are as specified
void	hostRoleScheduled(Stage s, String hostname, String roleStr) Updates scheduled stage.
void	invalidateCommandCacheOnHostRemove(HostsRemovedEvent event) Invalidate cached HostRoleCommands if a host is deleted.
void	persistActions(Request request) Persists all tasks for a given request
void	resubmitTasks(List<Long> taskIds) Resubmits a series of tasks
void	setSourceScheduleForRequest(long requestId, long scheduleId) Updates request with link to source schedule
void	startRequest(long requestId)
void	timeoutHostRole(String host, long requestId, long stageId, String role) Mark the task as to have timed out
void	timeoutHostRole(String host, long requestId, long stageId, String role, boolean skipSupported, boolean hostUnknownState) Mark the task as to have timed out
void	updateHostRoleState(String hostname, long requestId, long stageId, String role, CommandReport report) For the given host, update all the tasks based on the command report
void	updateHostRoleStates(Collection<CommandReport> reports) Update tasks according to command reports

Methods inherited from class java.lang.Object

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

Constructor Detail

ActionDBAccessorImpl

@Inject
public ActionDBAccessorImpl(long cacheLimit,
                                    AmbariEventPublisher eventPublisher)

Method Detail

getStage

public Stage getStage(String actionId)

Description copied from interface: ActionDBAccessor

Given an action id of the form requestId-stageId, retrieve the Stage

Specified by:

getStage in interface ActionDBAccessor

getAllStages

public List<Stage> getAllStages(long requestId)

Description copied from interface: ActionDBAccessor

Get all stages associated with a single request id

Specified by:

getAllStages in interface ActionDBAccessor

getRequestEntity

public RequestEntity getRequestEntity(long requestId)

Description copied from interface: ActionDBAccessor

Gets the request entity by id. Will not load the entire Request/Stage/HostRoleCommand object hierarchy.

Specified by:

getRequestEntity in interface ActionDBAccessor

getRequest

public Request getRequest(long requestId)

Description copied from interface: ActionDBAccessor

Get request object by id. USE WITH CAUTION! This method will result in loading the full object hierarchy, and can be expensive to construct.

Specified by:

getRequest in interface ActionDBAccessor

Parameters:

requestId - the request id

Returns:

the Request instance, with all Stages and HostRoleCommands constructed from the database

abortOperation

public Collection<HostRoleCommandEntity> abortOperation(long requestId)

Abort all outstanding operations associated with the given request. This method uses the HostRoleStatus.SCHEDULED_STATES to determine which HostRoleCommand instances to abort. Returns the list of the aborted operations.

Specified by:

abortOperation in interface ActionDBAccessor

timeoutHostRole

public void timeoutHostRole(String host,
                            long requestId,
                            long stageId,
                            String role)

Description copied from interface: ActionDBAccessor

Mark the task as to have timed out

Specified by:

timeoutHostRole in interface ActionDBAccessor

timeoutHostRole

public void timeoutHostRole(String host,
                            long requestId,
                            long stageId,
                            String role,
                            boolean skipSupported,
                            boolean hostUnknownState)

Description copied from interface: ActionDBAccessor

Mark the task as to have timed out

Specified by:

timeoutHostRole in interface ActionDBAccessor

getStagesInProgressForRequest

public List<Stage> getStagesInProgressForRequest(Long requestId)

Returns all the pending stages in a request, including queued and not-queued. A stage is considered in progress if it is in progress for any host.

The results will be sorted by stage ID making this call expensive in some scenarios. Use ActionDBAccessor.getCommandsInProgressCount() in order to determine if there are stages that are in progress before getting the stages from this method.

Specified by:

getStagesInProgressForRequest in interface ActionDBAccessor

See Also:

HostRoleStatus.IN_PROGRESS_STATUSES

getFirstStageInProgressPerRequest

public List<Stage> getFirstStageInProgressPerRequest()

Returns the next stage which is in-progress for every in-progress request in the system. Since stages are always synchronous, there is no reason to return more than the most recent stage per request. Returning every single stage in the requesrt would be extremely inffecient and wasteful. However, since requests can run in parallel, this method must return the most recent stage for every request. The results will be sorted by request ID.

Use ActionDBAccessor.getCommandsInProgressCount() in order to determine if there are stages that are in progress before getting the stages from this method.

Specified by:

getFirstStageInProgressPerRequest in interface ActionDBAccessor

See Also:

HostRoleStatus.IN_PROGRESS_STATUSES

getCommandsInProgressCount

public int getCommandsInProgressCount()

Gets the number of commands in progress.

Specified by:

getCommandsInProgressCount in interface ActionDBAccessor

Returns:

the number of commands in progress.

persistActions

public void persistActions(Request request)
                    throws org.apache.ambari.server.AmbariException

Description copied from interface: ActionDBAccessor

Persists all tasks for a given request

Specified by:

persistActions in interface ActionDBAccessor

Parameters:

request - request object

Throws:

org.apache.ambari.server.AmbariException

startRequest

public void startRequest(long requestId)

Specified by:

startRequest in interface ActionDBAccessor

endRequest

public void endRequest(long requestId)

Specified by:

endRequest in interface ActionDBAccessor

endRequestIfCompleted

public void endRequestIfCompleted(long requestId)

setSourceScheduleForRequest

public void setSourceScheduleForRequest(long requestId,
                                        long scheduleId)

Description copied from interface: ActionDBAccessor

Updates request with link to source schedule

Specified by:

setSourceScheduleForRequest in interface ActionDBAccessor

updateHostRoleStates

public void updateHostRoleStates(Collection<CommandReport> reports)

Description copied from interface: ActionDBAccessor

Update tasks according to command reports

Specified by:

updateHostRoleStates in interface ActionDBAccessor

updateHostRoleState

public void updateHostRoleState(String hostname,
                                long requestId,
                                long stageId,
                                String role,
                                CommandReport report)

Description copied from interface: ActionDBAccessor

For the given host, update all the tasks based on the command report

Specified by:

updateHostRoleState in interface ActionDBAccessor

abortHostRole

public void abortHostRole(String host,
                          long requestId,
                          long stageId,
                          String role)

Description copied from interface: ActionDBAccessor

Mark the task as to have been aborted

Specified by:

abortHostRole in interface ActionDBAccessor

abortHostRole

public void abortHostRole(String host,
                          long requestId,
                          long stageId,
                          String role,
                          String reason)

Description copied from interface: ActionDBAccessor

Mark the task as to have been aborted. Reason should be specified manually.

Specified by:

abortHostRole in interface ActionDBAccessor

getLastPersistedRequestIdWhenInitialized

public long getLastPersistedRequestIdWhenInitialized()

Description copied from interface: ActionDBAccessor

Return the last persisted Request ID as seen when the DBAccessor object was initialized. Value should remain unchanged through the lifetime of the object instance.

Specified by:

getLastPersistedRequestIdWhenInitialized in interface ActionDBAccessor

Returns:

Request Id seen at init time

bulkHostRoleScheduled

public void bulkHostRoleScheduled(Stage s,
                                  List<ExecutionCommand> commands)

Description copied from interface: ActionDBAccessor

Bulk update scheduled commands

Specified by:

bulkHostRoleScheduled in interface ActionDBAccessor

bulkAbortHostRole

public void bulkAbortHostRole(Stage s,
                              Map<ExecutionCommand,String> commands)

Description copied from interface: ActionDBAccessor

Bulk abort commands

Specified by:

bulkAbortHostRole in interface ActionDBAccessor

hostRoleScheduled

public void hostRoleScheduled(Stage s,
                              String hostname,
                              String roleStr)

Description copied from interface: ActionDBAccessor

Updates scheduled stage.

Specified by:

hostRoleScheduled in interface ActionDBAccessor

getRequestTasks

public List<HostRoleCommand> getRequestTasks(long requestId)

Description copied from interface: ActionDBAccessor

Given a request id, get all the tasks that belong to this request

Specified by:

getRequestTasks in interface ActionDBAccessor

getAllTasksByRequestIds

public List<HostRoleCommand> getAllTasksByRequestIds(Collection<Long> requestIds)

Description copied from interface: ActionDBAccessor

Given a list of request ids, get all the tasks that belong to these requests

Specified by:

getAllTasksByRequestIds in interface ActionDBAccessor

getTasks

public List<HostRoleCommand> getTasks(Collection<Long> taskIds)

Description copied from interface: ActionDBAccessor

Given a list of task ids, get all the host role commands

Specified by:

getTasks in interface ActionDBAccessor

getTasksByHostRoleAndStatus

public List<HostRoleCommand> getTasksByHostRoleAndStatus(String hostname,
                                                         String role,
                                                         HostRoleStatus status)

Description copied from interface: ActionDBAccessor

Get a List of host role commands where the host, role and status are as specified

Specified by:

getTasksByHostRoleAndStatus in interface ActionDBAccessor

getTasksByRoleAndStatus

public List<HostRoleCommand> getTasksByRoleAndStatus(String role,
                                                     HostRoleStatus status)

Description copied from interface: ActionDBAccessor

Get a List of host role commands where the role and status are as specified

Specified by:

getTasksByRoleAndStatus in interface ActionDBAccessor

getTask

public HostRoleCommand getTask(long taskId)

Description copied from interface: ActionDBAccessor

Gets the host role command corresponding to the task id

Specified by:

getTask in interface ActionDBAccessor

getRequestsByStatus

public List<Long> getRequestsByStatus(RequestStatus status,
                                      int maxResults,
                                      boolean ascOrder)

Description copied from interface: ActionDBAccessor

Get first or last maxResults requests that are in the specified status

Specified by:

getRequestsByStatus in interface ActionDBAccessor

Parameters:

status - Desired request status

maxResults - maximal number of returned id's

ascOrder - defines sorting order for database query result

Returns:

First or last maxResults request id's if ascOrder is true or false, respectively

getRequestContext

public Map<Long,String> getRequestContext(List<Long> requestIds)

Description copied from interface: ActionDBAccessor

Gets request contexts associated with the list of request id

Specified by:

getRequestContext in interface ActionDBAccessor

getRequestContext

public String getRequestContext(long requestId)

Description copied from interface: ActionDBAccessor

Gets the request context associated with the request id

Specified by:

getRequestContext in interface ActionDBAccessor

getRequests

public List<Request> getRequests(Collection<Long> requestIds)

Description copied from interface: ActionDBAccessor

Gets request objects by ids

Specified by:

getRequests in interface ActionDBAccessor

resubmitTasks

public void resubmitTasks(List<Long> taskIds)

Description copied from interface: ActionDBAccessor

Resubmits a series of tasks

Specified by:

resubmitTasks in interface ActionDBAccessor

invalidateCommandCacheOnHostRemove

public void invalidateCommandCacheOnHostRemove(HostsRemovedEvent event)

Invalidate cached HostRoleCommands if a host is deleted.

Parameters:

event - @HostRemovedEvent