org.apache.drill.exec.resourcemgr.config

Class ResourcePoolTreeImpl

java.lang.Object

org.apache.drill.exec.resourcemgr.config.ResourcePoolTreeImpl

All Implemented Interfaces:

ResourcePoolTree

public class ResourcePoolTreeImpl
extends Object
implements ResourcePoolTree

Parses and initializes configuration for ResourceManagement in Drill. It takes care of creating all the ResourcePools recursively maintaining the n-ary tree hierarchy defined in the configuration.

Field Summary

Fields

Modifier and Type	Field and Description
static String	ROOT_POOL_CONFIG_KEY
static String	ROOT_POOL_QUEUE_SELECTION_POLICY_KEY

Constructor Summary

Constructors

Constructor and Description
ResourcePoolTreeImpl(com.typesafe.config.Config rmConfig, long totalNodeMemory, int totalNodePhysicalCpu, int vFactor)

Method Summary

Modifier and Type	Method and Description
Map<String,QueryQueueConfig>	getAllLeafQueues()
double	getResourceShare()
ResourcePool	getRootPool()
QueueSelectionPolicy	getSelectionPolicyInUse()
QueueAssignmentResult	selectAllQueues(QueryContext queryContext) Creates QueueAssignmentResult which contains list of all selected leaf ResourcePools and all the rejected ResourcePools for the provided query.
QueryQueueConfig	selectOneQueue(QueryContext queryContext, NodeResources queryMaxNodeResource) Selects a leaf queue out of all the possibles leaf queues returned by ResourcePoolTree.selectAllQueues(QueryContext) for a given query based on the configured QueueSelectionPolicy.
String	toString()

Methods inherited from class java.lang.Object

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

Field Detail

ROOT_POOL_QUEUE_SELECTION_POLICY_KEY

public static final String ROOT_POOL_QUEUE_SELECTION_POLICY_KEY

See Also:

Constant Field Values

ROOT_POOL_CONFIG_KEY

public static final String ROOT_POOL_CONFIG_KEY

See Also:

Constant Field Values

Constructor Detail

ResourcePoolTreeImpl

public ResourcePoolTreeImpl(com.typesafe.config.Config rmConfig,
                            long totalNodeMemory,
                            int totalNodePhysicalCpu,
                            int vFactor)
                     throws RMConfigException

Throws:

RMConfigException

Method Detail

getRootPool

public ResourcePool getRootPool()

Specified by:

getRootPool in interface ResourcePoolTree

Returns:

root ResourcePool

getAllLeafQueues

public Map<String,QueryQueueConfig> getAllLeafQueues()

Specified by:

getAllLeafQueues in interface ResourcePoolTree

Returns:

Map containing all the configured leaf queues

getResourceShare

public double getResourceShare()

Specified by:

getResourceShare in interface ResourcePoolTree

selectAllQueues

public QueueAssignmentResult selectAllQueues(QueryContext queryContext)

Creates QueueAssignmentResult which contains list of all selected leaf ResourcePools and all the rejected ResourcePools for the provided query. Performs DFS of the ResourcePoolTree to traverse and find selected/rejected ResourcePools.

Specified by:

selectAllQueues in interface ResourcePoolTree

Parameters:

queryContext - QueryContext which contains metadata required for the given query

Returns:

QueueAssignmentResult populated with selected/rejected ResourcePools

selectOneQueue

public QueryQueueConfig selectOneQueue(QueryContext queryContext,
                                       NodeResources queryMaxNodeResource)
                                throws QueueSelectionException

Selects a leaf queue out of all the possibles leaf queues returned by ResourcePoolTree.selectAllQueues(QueryContext) for a given query based on the configured QueueSelectionPolicy. If none of the queue qualifies then it throws QueueSelectionException

Specified by:

selectOneQueue in interface ResourcePoolTree

Parameters:

queryContext - QueryContext which contains metadata for given query

queryMaxNodeResource - Max resources on a node required for given query

Returns:

QueryQueueConfig for the selected leaf queue

Throws:

QueueSelectionException - If no leaf queue is selected

getSelectionPolicyInUse

public QueueSelectionPolicy getSelectionPolicyInUse()

Specified by:

getSelectionPolicyInUse in interface ResourcePoolTree

Returns:

QueueSelectionPolicy which is used to chose one queue out of all the available options for a query

toString

public String toString()

Overrides:

toString in class Object