org.apache.tez.dag.app.rm

Class TaskSchedulerContextImpl

java.lang.Object

org.apache.tez.dag.app.rm.TaskSchedulerContextImpl

All Implemented Interfaces:

org.apache.tez.serviceplugins.api.ServicePluginContextBase, org.apache.tez.serviceplugins.api.TaskSchedulerContext

public class TaskSchedulerContextImpl
extends Object
implements org.apache.tez.serviceplugins.api.TaskSchedulerContext

Nested Class Summary

Nested classes/interfaces inherited from interface org.apache.tez.serviceplugins.api.TaskSchedulerContext

org.apache.tez.serviceplugins.api.TaskSchedulerContext.AMState, org.apache.tez.serviceplugins.api.TaskSchedulerContext.AppFinalStatus

Constructor Summary

Constructors

Constructor and Description
TaskSchedulerContextImpl(TaskSchedulerManager taskSchedulerManager, AppContext appContext, int schedulerId, String trackingUrl, long customClusterIdentifier, String appHostname, int clientPort, org.apache.tez.dag.api.UserPayload initialUserPayload)

Method Summary

Modifier and Type	Method and Description
void	appShutdownRequested()
void	containerBeingReleased(org.apache.hadoop.yarn.api.records.ContainerId containerId)
void	containerCompleted(Object taskLastAllocated, org.apache.hadoop.yarn.api.records.ContainerStatus containerStatus)
org.apache.tez.serviceplugins.api.TaskSchedulerContext.AMState	getAMState()
int	getAppClientPort()
String	getAppHostName()
org.apache.hadoop.yarn.api.records.ApplicationAttemptId	getApplicationAttemptId()
String	getAppTrackingUrl()
org.apache.tez.common.ContainerSignatureMatcher	getContainerSignatureMatcher()
org.apache.tez.serviceplugins.api.DagInfo	getCurrentDagInfo()
long	getCustomClusterIdentifier()
org.apache.tez.serviceplugins.api.TaskSchedulerContext.AppFinalStatus	getFinalAppStatus()
org.apache.tez.dag.api.UserPayload	getInitialUserPayload()
float	getProgress()
int	getVertexIndexForTask(Object task)
boolean	isSession()
void	nodesUpdated(List<org.apache.hadoop.yarn.api.records.NodeReport> updatedNodes)
void	preemptContainer(org.apache.hadoop.yarn.api.records.ContainerId containerId)
void	reportError(org.apache.tez.serviceplugins.api.ServicePluginError servicePluginError, String diagnostics, org.apache.tez.serviceplugins.api.DagInfo dagInfo)
void	setApplicationRegistrationData(org.apache.hadoop.yarn.api.records.Resource maxContainerCapability, Map<org.apache.hadoop.yarn.api.records.ApplicationAccessType,String> appAcls, ByteBuffer clientAMSecretKey, String queueName)
void	taskAllocated(Object task, Object appCookie, org.apache.hadoop.yarn.api.records.Container container)

Methods inherited from class java.lang.Object

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

Constructor Detail

TaskSchedulerContextImpl

public TaskSchedulerContextImpl(TaskSchedulerManager taskSchedulerManager,
                                AppContext appContext,
                                int schedulerId,
                                String trackingUrl,
                                long customClusterIdentifier,
                                String appHostname,
                                int clientPort,
                                org.apache.tez.dag.api.UserPayload initialUserPayload)

Method Detail

taskAllocated

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

Specified by:

taskAllocated in interface org.apache.tez.serviceplugins.api.TaskSchedulerContext

containerCompleted

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

Specified by:

containerCompleted in interface org.apache.tez.serviceplugins.api.TaskSchedulerContext

containerBeingReleased

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

Specified by:

containerBeingReleased in interface org.apache.tez.serviceplugins.api.TaskSchedulerContext

nodesUpdated

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

Specified by:

nodesUpdated in interface org.apache.tez.serviceplugins.api.TaskSchedulerContext

appShutdownRequested

public void appShutdownRequested()

Specified by:

appShutdownRequested in interface org.apache.tez.serviceplugins.api.TaskSchedulerContext

setApplicationRegistrationData

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

Specified by:

setApplicationRegistrationData in interface org.apache.tez.serviceplugins.api.TaskSchedulerContext

getProgress

public float getProgress()

Specified by:

getProgress in interface org.apache.tez.serviceplugins.api.TaskSchedulerContext

preemptContainer

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

Specified by:

preemptContainer in interface org.apache.tez.serviceplugins.api.TaskSchedulerContext

getFinalAppStatus

public org.apache.tez.serviceplugins.api.TaskSchedulerContext.AppFinalStatus getFinalAppStatus()

Specified by:

getFinalAppStatus in interface org.apache.tez.serviceplugins.api.TaskSchedulerContext

getInitialUserPayload

public org.apache.tez.dag.api.UserPayload getInitialUserPayload()

Specified by:

getInitialUserPayload in interface org.apache.tez.serviceplugins.api.ServicePluginContextBase

getAppTrackingUrl

public String getAppTrackingUrl()

Specified by:

getAppTrackingUrl in interface org.apache.tez.serviceplugins.api.TaskSchedulerContext

getCustomClusterIdentifier

public long getCustomClusterIdentifier()

Specified by:

getCustomClusterIdentifier in interface org.apache.tez.serviceplugins.api.TaskSchedulerContext

getContainerSignatureMatcher

public org.apache.tez.common.ContainerSignatureMatcher getContainerSignatureMatcher()

Specified by:

getContainerSignatureMatcher in interface org.apache.tez.serviceplugins.api.TaskSchedulerContext

getApplicationAttemptId

public org.apache.hadoop.yarn.api.records.ApplicationAttemptId getApplicationAttemptId()

Specified by:

getApplicationAttemptId in interface org.apache.tez.serviceplugins.api.TaskSchedulerContext

getCurrentDagInfo

@Nullable
public org.apache.tez.serviceplugins.api.DagInfo getCurrentDagInfo()

Specified by:

getCurrentDagInfo in interface org.apache.tez.serviceplugins.api.ServicePluginContextBase

getAppHostName

public String getAppHostName()

Specified by:

getAppHostName in interface org.apache.tez.serviceplugins.api.TaskSchedulerContext

getAppClientPort

public int getAppClientPort()

Specified by:

getAppClientPort in interface org.apache.tez.serviceplugins.api.TaskSchedulerContext

isSession

public boolean isSession()

Specified by:

isSession in interface org.apache.tez.serviceplugins.api.TaskSchedulerContext

getAMState

public org.apache.tez.serviceplugins.api.TaskSchedulerContext.AMState getAMState()

Specified by:

getAMState in interface org.apache.tez.serviceplugins.api.TaskSchedulerContext

getVertexIndexForTask

public int getVertexIndexForTask(Object task)

Specified by:

getVertexIndexForTask in interface org.apache.tez.serviceplugins.api.TaskSchedulerContext

reportError

public void reportError(org.apache.tez.serviceplugins.api.ServicePluginError servicePluginError,
                        String diagnostics,
                        org.apache.tez.serviceplugins.api.DagInfo dagInfo)

Specified by:

reportError in interface org.apache.tez.serviceplugins.api.ServicePluginContextBase