org.apache.tez.dag.app.rm

Class LocalTaskSchedulerService

java.lang.Object

org.apache.tez.serviceplugins.api.TaskScheduler

org.apache.tez.dag.app.rm.LocalTaskSchedulerService

All Implemented Interfaces:

org.apache.tez.common.ServicePluginLifecycle

public class LocalTaskSchedulerService
extends org.apache.tez.serviceplugins.api.TaskScheduler

Nested Class Summary

Nested classes/interfaces inherited from class org.apache.tez.serviceplugins.api.TaskScheduler

org.apache.tez.serviceplugins.api.TaskScheduler.SchedulerTaskState

Constructor Summary

Constructors

Constructor and Description
LocalTaskSchedulerService(org.apache.tez.serviceplugins.api.TaskSchedulerContext taskSchedulerContext)

Method Summary

Modifier and Type	Method and Description
void	allocateTask(Object task, org.apache.hadoop.yarn.api.records.Resource capability, org.apache.hadoop.yarn.api.records.ContainerId containerId, org.apache.hadoop.yarn.api.records.Priority priority, Object containerSignature, Object clientCookie)
void	allocateTask(Object task, org.apache.hadoop.yarn.api.records.Resource capability, String[] hosts, String[] racks, org.apache.hadoop.yarn.api.records.Priority priority, Object containerSignature, Object clientCookie)
void	blacklistNode(org.apache.hadoop.yarn.api.records.NodeId nodeId)
protected org.apache.tez.dag.app.rm.LocalTaskSchedulerService.AsyncDelegateRequestHandler	createRequestHandler(org.apache.hadoop.conf.Configuration conf)
void	dagComplete()
Object	deallocateContainer(org.apache.hadoop.yarn.api.records.ContainerId containerId)
boolean	deallocateTask(Object task, boolean taskSucceeded, org.apache.tez.serviceplugins.api.TaskAttemptEndReason endReason, String diagnostics)
org.apache.hadoop.yarn.api.records.Resource	getAvailableResources()
int	getClusterNodeCount()
org.apache.hadoop.yarn.api.records.Resource	getTotalResources()
boolean	hasUnregistered()
void	initialize()
void	initiateStop()
void	setShouldUnregister()
void	shutdown()
void	start()
void	unblacklistNode(org.apache.hadoop.yarn.api.records.NodeId nodeId)

Methods inherited from class org.apache.tez.serviceplugins.api.TaskScheduler

getContext, taskStateUpdated

Methods inherited from class java.lang.Object

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

Constructor Detail

LocalTaskSchedulerService

public LocalTaskSchedulerService(org.apache.tez.serviceplugins.api.TaskSchedulerContext taskSchedulerContext)

Method Detail

getAvailableResources

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

Specified by:

getAvailableResources in class org.apache.tez.serviceplugins.api.TaskScheduler

getClusterNodeCount

public int getClusterNodeCount()

Specified by:

getClusterNodeCount in class org.apache.tez.serviceplugins.api.TaskScheduler

dagComplete

public void dagComplete()

Specified by:

dagComplete in class org.apache.tez.serviceplugins.api.TaskScheduler

getTotalResources

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

Specified by:

getTotalResources in class org.apache.tez.serviceplugins.api.TaskScheduler

blacklistNode

public void blacklistNode(org.apache.hadoop.yarn.api.records.NodeId nodeId)

Specified by:

blacklistNode in class org.apache.tez.serviceplugins.api.TaskScheduler

unblacklistNode

public void unblacklistNode(org.apache.hadoop.yarn.api.records.NodeId nodeId)

Specified by:

unblacklistNode in class org.apache.tez.serviceplugins.api.TaskScheduler

allocateTask

public void allocateTask(Object task,
                         org.apache.hadoop.yarn.api.records.Resource capability,
                         String[] hosts,
                         String[] racks,
                         org.apache.hadoop.yarn.api.records.Priority priority,
                         Object containerSignature,
                         Object clientCookie)

Specified by:

allocateTask in class org.apache.tez.serviceplugins.api.TaskScheduler

allocateTask

public void allocateTask(Object task,
                         org.apache.hadoop.yarn.api.records.Resource capability,
                         org.apache.hadoop.yarn.api.records.ContainerId containerId,
                         org.apache.hadoop.yarn.api.records.Priority priority,
                         Object containerSignature,
                         Object clientCookie)

Specified by:

allocateTask in class org.apache.tez.serviceplugins.api.TaskScheduler

deallocateTask

public boolean deallocateTask(Object task,
                              boolean taskSucceeded,
                              org.apache.tez.serviceplugins.api.TaskAttemptEndReason endReason,
                              String diagnostics)

Specified by:

deallocateTask in class org.apache.tez.serviceplugins.api.TaskScheduler

deallocateContainer

public Object deallocateContainer(org.apache.hadoop.yarn.api.records.ContainerId containerId)

Specified by:

deallocateContainer in class org.apache.tez.serviceplugins.api.TaskScheduler

initialize

public void initialize()

Specified by:

initialize in interface org.apache.tez.common.ServicePluginLifecycle

Overrides:

initialize in class org.apache.tez.serviceplugins.api.TaskScheduler

createRequestHandler

protected org.apache.tez.dag.app.rm.LocalTaskSchedulerService.AsyncDelegateRequestHandler createRequestHandler(org.apache.hadoop.conf.Configuration conf)

start

public void start()

Specified by:

start in interface org.apache.tez.common.ServicePluginLifecycle

Overrides:

start in class org.apache.tez.serviceplugins.api.TaskScheduler

shutdown

public void shutdown()
              throws InterruptedException

Specified by:

shutdown in interface org.apache.tez.common.ServicePluginLifecycle

Overrides:

shutdown in class org.apache.tez.serviceplugins.api.TaskScheduler

Throws:

InterruptedException

setShouldUnregister

public void setShouldUnregister()

Specified by:

setShouldUnregister in class org.apache.tez.serviceplugins.api.TaskScheduler

hasUnregistered

public boolean hasUnregistered()

Specified by:

hasUnregistered in class org.apache.tez.serviceplugins.api.TaskScheduler

initiateStop

public void initiateStop()

Overrides:

initiateStop in class org.apache.tez.serviceplugins.api.TaskScheduler