org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity

Class CapacityScheduler

java.lang.Object

org.apache.hadoop.service.AbstractService

org.apache.hadoop.yarn.server.resourcemanager.scheduler.AbstractYarnScheduler<FiCaSchedulerApp,FiCaSchedulerNode>

org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity.CapacityScheduler

All Implemented Interfaces:

Closeable, AutoCloseable, org.apache.hadoop.conf.Configurable, org.apache.hadoop.service.Service, org.apache.hadoop.yarn.event.EventHandler<SchedulerEvent>, Recoverable, CapacitySchedulerContext, PreemptableResourceScheduler, ResourceScheduler, YarnScheduler

@InterfaceAudience.LimitedPrivate(value="yarn")
@InterfaceStability.Evolving
public class CapacityScheduler
extends AbstractYarnScheduler<FiCaSchedulerApp,FiCaSchedulerNode>
implements PreemptableResourceScheduler, CapacitySchedulerContext, org.apache.hadoop.conf.Configurable

Nested Class Summary

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
static String	ROOT_QUEUE
protected long	THREAD_JOIN_TIMEOUT_MS

Fields inherited from class org.apache.hadoop.yarn.server.resourcemanager.scheduler.AbstractYarnScheduler

applications, clusterResource, EMPTY_ALLOCATION, EMPTY_CONTAINER_LIST, maximumAllocation, minimumAllocation, nodes, rmContext

Constructor Summary

Constructors

Constructor and Description
CapacityScheduler()

Method Summary

Methods

Modifier and Type	Method and Description
Allocation	allocate(org.apache.hadoop.yarn.api.records.ApplicationAttemptId applicationAttemptId, List<org.apache.hadoop.yarn.api.records.ResourceRequest> ask, List<org.apache.hadoop.yarn.api.records.ContainerId> release, List<String> blacklistAdditions, List<String> blacklistRemovals) The main api between the ApplicationMaster and the Scheduler.
boolean	checkAccess(org.apache.hadoop.security.UserGroupInformation callerUGI, org.apache.hadoop.yarn.api.records.QueueACL acl, String queueName) Check if the user has permission to perform the operation.
void	dropContainerReservation(RMContainer container) If the scheduler support container reservations, this method is used to ask the scheduler to drop the reservation for the given container.
FiCaSchedulerApp	getApplicationAttempt(org.apache.hadoop.yarn.api.records.ApplicationAttemptId applicationAttemptId)
Comparator<FiCaSchedulerApp>	getApplicationComparator()
List<org.apache.hadoop.yarn.api.records.ApplicationAttemptId>	getAppsInQueue(String queueName) Gets the apps under a given queue
org.apache.hadoop.conf.Configuration	getConf() Get the yarn configuration.
CapacitySchedulerConfiguration	getConfiguration()
RMContainerTokenSecretManager	getContainerTokenSecretManager()
int	getNumClusterNodes() Get the number of nodes available in the cluster.
Comparator<CSQueue>	getQueueComparator()
org.apache.hadoop.yarn.api.records.QueueInfo	getQueueInfo(String queueName, boolean includeChildQueues, boolean recursive) Get queue information
List<org.apache.hadoop.yarn.api.records.QueueUserACLInfo>	getQueueUserAclInfo() Get acls for queues for current user.
org.apache.hadoop.yarn.util.resource.ResourceCalculator	getResourceCalculator()
RMContext	getRMContext()
CSQueue	getRootQueue()
QueueMetrics	getRootQueueMetrics() Get the root queue for the scheduler.
void	handle(SchedulerEvent event)
void	killContainer(RMContainer cont) Ask the scheduler to forcibly interrupt the container given as input
void	preemptContainer(org.apache.hadoop.yarn.api.records.ApplicationAttemptId aid, RMContainer cont) Ask the scheduler to obtain back the container from a specific application by issuing a preemption request
void	recover(RMStateStore.RMState state)
void	reinitialize(org.apache.hadoop.conf.Configuration conf, RMContext rmContext) Re-initialize the ResourceScheduler.
void	serviceInit(org.apache.hadoop.conf.Configuration conf)
void	serviceStart()
void	serviceStop()
void	setConf(org.apache.hadoop.conf.Configuration conf)
void	setRMContext(RMContext rmContext) Set RMContext for ResourceScheduler.

Methods inherited from class org.apache.hadoop.yarn.server.resourcemanager.scheduler.AbstractYarnScheduler

getAppResourceUsageReport, getClusterResource, getCurrentAttemptForContainer, getMaximumResourceCapability, getMinimumResourceCapability, getNodeReport, getRMContainer, getSchedulerAppInfo, getSchedulerApplications, getSchedulerNode, getTransferredContainers, moveApplication, recoverContainersOnNode, recoverResourceRequestForContainer

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, setConfig, start, stop, toString, unregisterGlobalListener, unregisterServiceListener, waitForServiceToStop

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.apache.hadoop.yarn.server.resourcemanager.scheduler.YarnScheduler

getAppResourceUsageReport, getClusterResource, getMaximumResourceCapability, getMinimumResourceCapability, getNodeReport, getRMContainer, getSchedulerAppInfo, moveApplication

Methods inherited from interface org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity.CapacitySchedulerContext

getClusterResource, getMaximumResourceCapability, getMinimumResourceCapability

Field Detail

THREAD_JOIN_TIMEOUT_MS

protected final long THREAD_JOIN_TIMEOUT_MS

See Also:

Constant Field Values

ROOT_QUEUE

@InterfaceAudience.Private
public static final String ROOT_QUEUE

See Also:

Constant Field Values

Constructor Detail

CapacityScheduler

public CapacityScheduler()

Method Detail

setConf

public void setConf(org.apache.hadoop.conf.Configuration conf)

Specified by:

setConf in interface org.apache.hadoop.conf.Configurable

getConf

public org.apache.hadoop.conf.Configuration getConf()

Description copied from interface: CapacitySchedulerContext

Get the yarn configuration.

Specified by:

getConf in interface org.apache.hadoop.conf.Configurable

Specified by:

getConf in interface CapacitySchedulerContext

getRootQueueMetrics

public QueueMetrics getRootQueueMetrics()

Description copied from interface: YarnScheduler

Get the root queue for the scheduler.

Specified by:

getRootQueueMetrics in interface YarnScheduler

Returns:

the root queue for the scheduler.

getRootQueue

public CSQueue getRootQueue()

getConfiguration

public CapacitySchedulerConfiguration getConfiguration()

Specified by:

getConfiguration in interface CapacitySchedulerContext

getContainerTokenSecretManager

public RMContainerTokenSecretManager getContainerTokenSecretManager()

Specified by:

getContainerTokenSecretManager in interface CapacitySchedulerContext

getApplicationComparator

public Comparator<FiCaSchedulerApp> getApplicationComparator()

Specified by:

getApplicationComparator in interface CapacitySchedulerContext

getResourceCalculator

public org.apache.hadoop.yarn.util.resource.ResourceCalculator getResourceCalculator()

Specified by:

getResourceCalculator in interface CapacitySchedulerContext

getQueueComparator

public Comparator<CSQueue> getQueueComparator()

Specified by:

getQueueComparator in interface CapacitySchedulerContext

getNumClusterNodes

public int getNumClusterNodes()

Description copied from interface: YarnScheduler

Get the number of nodes available in the cluster.

Specified by:

getNumClusterNodes in interface CapacitySchedulerContext

Specified by:

getNumClusterNodes in interface YarnScheduler

Returns:

the number of available nodes.

getRMContext

public RMContext getRMContext()

Specified by:

getRMContext in interface CapacitySchedulerContext

setRMContext

public void setRMContext(RMContext rmContext)

Description copied from interface: ResourceScheduler

Set RMContext for ResourceScheduler. This method should be called immediately after instantiating a scheduler once.

Specified by:

setRMContext in interface ResourceScheduler

Parameters:

rmContext - created by ResourceManager

serviceInit

public void serviceInit(org.apache.hadoop.conf.Configuration conf)
                 throws Exception

Overrides:

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

Throws:

Exception

serviceStart

public void serviceStart()
                  throws Exception

Overrides:

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

Throws:

Exception

serviceStop

public void serviceStop()
                 throws Exception

Overrides:

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

Throws:

Exception

reinitialize

public void reinitialize(org.apache.hadoop.conf.Configuration conf,
                RMContext rmContext)
                  throws IOException

Description copied from interface: ResourceScheduler

Re-initialize the ResourceScheduler.

Specified by:

reinitialize in interface ResourceScheduler

Parameters:

conf - configuration

Throws:

IOException

allocate

@Lock(value=org.apache.hadoop.yarn.server.utils.Lock.NoLock.class)
public Allocation allocate(org.apache.hadoop.yarn.api.records.ApplicationAttemptId applicationAttemptId,
                                                                            List<org.apache.hadoop.yarn.api.records.ResourceRequest> ask,
                                                                            List<org.apache.hadoop.yarn.api.records.ContainerId> release,
                                                                            List<String> blacklistAdditions,
                                                                            List<String> blacklistRemovals)

Description copied from interface: YarnScheduler

The main api between the ApplicationMaster and the Scheduler. The ApplicationMaster is updating his future resource requirements and may release containers he doens't need.

Specified by:

allocate in interface YarnScheduler

Returns:

the Allocation for the application

getQueueInfo

@Lock(value=org.apache.hadoop.yarn.server.utils.Lock.NoLock.class)
public org.apache.hadoop.yarn.api.records.QueueInfo getQueueInfo(String queueName,
                                                                                                                  boolean includeChildQueues,
                                                                                                                  boolean recursive)
                                                          throws IOException

Description copied from interface: YarnScheduler

Get queue information

Specified by:

getQueueInfo in interface YarnScheduler

Parameters:

queueName - queue name

includeChildQueues - include child queues?

recursive - get children queues?

Returns:

queue information

Throws:

IOException

getQueueUserAclInfo

@Lock(value=org.apache.hadoop.yarn.server.utils.Lock.NoLock.class)
public List<org.apache.hadoop.yarn.api.records.QueueUserACLInfo> getQueueUserAclInfo()

Description copied from interface: YarnScheduler

Get acls for queues for current user.

Specified by:

getQueueUserAclInfo in interface YarnScheduler

Returns:

acls for queues for current user

handle

public void handle(SchedulerEvent event)

Specified by:

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

getApplicationAttempt

@Lock(value=org.apache.hadoop.yarn.server.utils.Lock.NoLock.class)
public FiCaSchedulerApp getApplicationAttempt(org.apache.hadoop.yarn.api.records.ApplicationAttemptId applicationAttemptId)

Overrides:

getApplicationAttempt in class AbstractYarnScheduler<FiCaSchedulerApp,FiCaSchedulerNode>

recover

@Lock(value=org.apache.hadoop.yarn.server.utils.Lock.NoLock.class)
public void recover(RMStateStore.RMState state)
             throws Exception

Specified by:

recover in interface Recoverable

Throws:

Exception

dropContainerReservation

public void dropContainerReservation(RMContainer container)

Description copied from interface: PreemptableResourceScheduler

If the scheduler support container reservations, this method is used to ask the scheduler to drop the reservation for the given container.

Specified by:

dropContainerReservation in interface PreemptableResourceScheduler

Parameters:

container - Reference to reserved container allocation.

preemptContainer

public void preemptContainer(org.apache.hadoop.yarn.api.records.ApplicationAttemptId aid,
                    RMContainer cont)

Description copied from interface: PreemptableResourceScheduler

Ask the scheduler to obtain back the container from a specific application by issuing a preemption request

Specified by:

preemptContainer in interface PreemptableResourceScheduler

Parameters:

aid - the application from which we want to get a container back

cont - the container we want back

killContainer

public void killContainer(RMContainer cont)

Description copied from interface: PreemptableResourceScheduler

Ask the scheduler to forcibly interrupt the container given as input

Specified by:

killContainer in interface PreemptableResourceScheduler

checkAccess

public boolean checkAccess(org.apache.hadoop.security.UserGroupInformation callerUGI,
                  org.apache.hadoop.yarn.api.records.QueueACL acl,
                  String queueName)

Description copied from interface: YarnScheduler

Check if the user has permission to perform the operation. If the user has QueueACL.ADMINISTER_QUEUE permission, this user can view/modify the applications in this queue

Specified by:

checkAccess in interface YarnScheduler

Returns:

true if the user has the permission, false otherwise

getAppsInQueue

public List<org.apache.hadoop.yarn.api.records.ApplicationAttemptId> getAppsInQueue(String queueName)

Description copied from interface: YarnScheduler

Gets the apps under a given queue

Specified by:

getAppsInQueue in interface YarnScheduler

Parameters:

queueName - the name of the queue.

Returns:

a collection of app attempt ids in the given queue.