org.springframework.yarn.am.allocate

Class DefaultAllocateCountTracker

java.lang.Object

org.springframework.yarn.am.allocate.DefaultAllocateCountTracker

public class DefaultAllocateCountTracker
extends java.lang.Object

Helper class tracking allocation counts. This separates counts for hosts into two states; in first state we have a counts of pending requests which are not yet sent into resource manager. in second state we have a counts which are sent into resource manager. These states allows us to loosely track needed counts sent during the allocation requests to minimise allocation garbage.

Author:

Janne Valkealahti

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	DefaultAllocateCountTracker.AllocateCountInfo

Constructor Summary

Constructors

Constructor and Description
DefaultAllocateCountTracker(org.apache.hadoop.conf.Configuration configuration) Instantiates a new default allocate count tracker.
DefaultAllocateCountTracker(java.lang.String id, org.apache.hadoop.conf.Configuration configuration)

Method Summary

Methods

Modifier and Type	Method and Description
void	addContainers(ContainerAllocateData containerAllocateData) Adds new count of containers into 'host', 'rack' and 'any' pending requests.
void	addContainers(int count) Adds new count of containers into 'any' pending requests.
DefaultAllocateCountTracker.AllocateCountInfo	getAllocateCounts() Gets the allocate counts which should be used to create allocate requests.
java.lang.String	getId()
org.apache.hadoop.yarn.api.records.Container	processAllocatedContainer(org.apache.hadoop.yarn.api.records.Container container)
void	setConfiguration(org.apache.hadoop.conf.Configuration configuration)
java.lang.String	toString()

Methods inherited from class java.lang.Object

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

Constructor Detail

DefaultAllocateCountTracker

public DefaultAllocateCountTracker(org.apache.hadoop.conf.Configuration configuration)

Instantiates a new default allocate count tracker.

Parameters:

configuration - the hadoop configuration

DefaultAllocateCountTracker

public DefaultAllocateCountTracker(java.lang.String id,
                           org.apache.hadoop.conf.Configuration configuration)

Method Detail

getId

public java.lang.String getId()

setConfiguration

public void setConfiguration(org.apache.hadoop.conf.Configuration configuration)

addContainers

public void addContainers(int count)

Adds new count of containers into 'any' pending requests.

Parameters:

count - the count to add

addContainers

public void addContainers(ContainerAllocateData containerAllocateData)

Adds new count of containers into 'host', 'rack' and 'any' pending requests.

Parameters:

containerAllocateData - the container allocate data

getAllocateCounts

public DefaultAllocateCountTracker.AllocateCountInfo getAllocateCounts()

Gets the allocate counts which should be used to create allocate requests.

Returns:

the allocate counts

processAllocatedContainer

public org.apache.hadoop.yarn.api.records.Container processAllocatedContainer(org.apache.hadoop.yarn.api.records.Container container)

toString

public java.lang.String toString()

Overrides:

toString in class java.lang.Object