org.apache.tez.dag.app.rm

Class TaskSchedulerEventHandler

java.lang.Object

org.apache.hadoop.service.AbstractService

org.apache.tez.dag.app.rm.TaskSchedulerEventHandler

All Implemented Interfaces:

Closeable, AutoCloseable, org.apache.hadoop.service.Service, org.apache.hadoop.yarn.event.EventHandler<AMSchedulerEvent>, TaskSchedulerService.TaskSchedulerAppCallback

public class TaskSchedulerEventHandler
extends org.apache.hadoop.service.AbstractService
implements TaskSchedulerService.TaskSchedulerAppCallback, org.apache.hadoop.yarn.event.EventHandler<AMSchedulerEvent>

Nested Class Summary

Nested classes/interfaces inherited from interface org.apache.tez.dag.app.rm.TaskSchedulerService.TaskSchedulerAppCallback

TaskSchedulerService.TaskSchedulerAppCallback.AppFinalStatus

Nested classes/interfaces inherited from interface org.apache.hadoop.service.Service

org.apache.hadoop.service.Service.STATE

Field Summary

Fields

Modifier and Type	Field and Description
protected AppContext	appContext
protected boolean	isSignalled
protected TaskSchedulerService	taskScheduler

Constructor Summary

Constructors

Constructor and Description
TaskSchedulerEventHandler(AppContext appContext, DAGClientServer clientService, org.apache.hadoop.yarn.event.EventHandler eventHandler, ContainerSignatureMatcher containerSignatureMatcher)

Method Summary

Methods

Modifier and Type	Method and Description
void	appShutdownRequested()
void	containerBeingReleased(org.apache.hadoop.yarn.api.records.ContainerId containerId)
void	containerCompleted(Object task, org.apache.hadoop.yarn.api.records.ContainerStatus containerStatus)
protected TaskSchedulerService	createTaskScheduler(String host, int port, String trackingUrl, AppContext appContext)
void	dagCompleted()
Map<org.apache.hadoop.yarn.api.records.ApplicationAccessType,String>	getApplicationAcls()
org.apache.hadoop.yarn.api.records.Resource	getAvailableResources()
TaskSchedulerService.TaskSchedulerAppCallback.AppFinalStatus	getFinalAppStatus()
int	getNumClusterNodes()
float	getProgress()
org.apache.hadoop.yarn.api.records.Resource	getTotalResources()
void	handle(AMSchedulerEvent event)
void	handleEvent(AMSchedulerEvent sEvent)
boolean	hasUnregistered()
void	nodesUpdated(List<org.apache.hadoop.yarn.api.records.NodeReport> updatedNodes)
protected void	notifyForTest()
void	onError(Throwable t)
void	preemptContainer(org.apache.hadoop.yarn.api.records.ContainerId containerId)
void	serviceStart()
void	serviceStop()
void	setApplicationRegistrationData(org.apache.hadoop.yarn.api.records.Resource maxContainerCapability, Map<org.apache.hadoop.yarn.api.records.ApplicationAccessType,String> appAcls, ByteBuffer clientAMSecretKey)
void	setShouldUnregisterFlag()
void	setSignalled(boolean isSignalled)
void	taskAllocated(Object task, Object appCookie, org.apache.hadoop.yarn.api.records.Container container)

Methods inherited from class org.apache.hadoop.service.AbstractService

close, getBlockers, getConfig, getFailureCause, getFailureState, getLifecycleHistory, getName, getServiceState, getStartTime, init, isInState, noteFailure, putBlocker, registerGlobalListener, registerServiceListener, removeBlocker, serviceInit, setConfig, start, stop, toString, unregisterGlobalListener, unregisterServiceListener, waitForServiceToStop

Methods inherited from class java.lang.Object

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

Field Detail

appContext

protected final AppContext appContext

taskScheduler

protected TaskSchedulerService taskScheduler

isSignalled

protected volatile boolean isSignalled

Constructor Detail

TaskSchedulerEventHandler

public TaskSchedulerEventHandler(AppContext appContext,
                         DAGClientServer clientService,
                         org.apache.hadoop.yarn.event.EventHandler eventHandler,
                         ContainerSignatureMatcher containerSignatureMatcher)

Method Detail

getApplicationAcls

public Map<org.apache.hadoop.yarn.api.records.ApplicationAccessType,String> getApplicationAcls()

setSignalled

public void setSignalled(boolean isSignalled)

getNumClusterNodes

public int getNumClusterNodes()

getAvailableResources

public org.apache.hadoop.yarn.api.records.Resource getAvailableResources()

getTotalResources

public org.apache.hadoop.yarn.api.records.Resource getTotalResources()

handleEvent

public void handleEvent(AMSchedulerEvent sEvent)

handle

public void handle(AMSchedulerEvent event)

Specified by:

handle in interface org.apache.hadoop.yarn.event.EventHandler<AMSchedulerEvent>

createTaskScheduler

protected TaskSchedulerService createTaskScheduler(String host,
                                       int port,
                                       String trackingUrl,
                                       AppContext appContext)

serviceStart

public void serviceStart()

Overrides:

serviceStart in class org.apache.hadoop.service.AbstractService

notifyForTest

protected void notifyForTest()

serviceStop

public void serviceStop()

Overrides:

serviceStop in class org.apache.hadoop.service.AbstractService

taskAllocated

public void taskAllocated(Object task,
                 Object appCookie,
                 org.apache.hadoop.yarn.api.records.Container container)

Specified by:

taskAllocated in interface TaskSchedulerService.TaskSchedulerAppCallback

containerCompleted

public void containerCompleted(Object task,
                      org.apache.hadoop.yarn.api.records.ContainerStatus containerStatus)

Specified by:

containerCompleted in interface TaskSchedulerService.TaskSchedulerAppCallback

containerBeingReleased

public void containerBeingReleased(org.apache.hadoop.yarn.api.records.ContainerId containerId)

Specified by:

containerBeingReleased in interface TaskSchedulerService.TaskSchedulerAppCallback

nodesUpdated

public void nodesUpdated(List<org.apache.hadoop.yarn.api.records.NodeReport> updatedNodes)

Specified by:

nodesUpdated in interface TaskSchedulerService.TaskSchedulerAppCallback

appShutdownRequested

public void appShutdownRequested()

Specified by:

appShutdownRequested in interface TaskSchedulerService.TaskSchedulerAppCallback

setApplicationRegistrationData

public void setApplicationRegistrationData(org.apache.hadoop.yarn.api.records.Resource maxContainerCapability,
                                  Map<org.apache.hadoop.yarn.api.records.ApplicationAccessType,String> appAcls,
                                  ByteBuffer clientAMSecretKey)

Specified by:

setApplicationRegistrationData in interface TaskSchedulerService.TaskSchedulerAppCallback

getFinalAppStatus

public TaskSchedulerService.TaskSchedulerAppCallback.AppFinalStatus getFinalAppStatus()

Specified by:

getFinalAppStatus in interface TaskSchedulerService.TaskSchedulerAppCallback

getProgress

public float getProgress()

Specified by:

getProgress in interface TaskSchedulerService.TaskSchedulerAppCallback

onError

public void onError(Throwable t)

Specified by:

onError in interface TaskSchedulerService.TaskSchedulerAppCallback

dagCompleted

public void dagCompleted()

preemptContainer

public void preemptContainer(org.apache.hadoop.yarn.api.records.ContainerId containerId)

Specified by:

preemptContainer in interface TaskSchedulerService.TaskSchedulerAppCallback

setShouldUnregisterFlag

public void setShouldUnregisterFlag()

hasUnregistered

public boolean hasUnregistered()