org.apache.drill.yarn.appMaster

Class AMYarnFacadeImpl

java.lang.Object

org.apache.drill.yarn.appMaster.AMYarnFacadeImpl

All Implemented Interfaces:

AMYarnFacade

public class AMYarnFacadeImpl
extends Object
implements AMYarnFacade

Wrapper around the asynchronous versions of the YARN AM-RM and AM-NM interfaces. Allows strategy code to factor out the YARN-specific bits so that strategy code is simpler. Also allows replacing the actual YARN code with a mock for unit testing.

Nested Class Summary

Nested classes/interfaces inherited from interface org.apache.drill.yarn.appMaster.AMYarnFacade

AMYarnFacade.YarnAppHostReport

Constructor Summary

Constructors

Constructor and Description
AMYarnFacadeImpl(int pollPeriodMs)

Method Summary

Modifier and Type	Method and Description
void	blacklistNode(String nodeName)
void	finish(boolean succeeded, String msg)
AMYarnFacade.YarnAppHostReport	getAppHostReport()
int	getNodeCount()
List<org.apache.hadoop.yarn.api.records.NodeReport>	getNodeReports()
org.apache.hadoop.yarn.api.protocolrecords.RegisterApplicationMasterResponse	getRegistrationResponse()
org.apache.hadoop.yarn.api.records.Resource	getResources()
String	getTrackingUrl()
void	killContainer(org.apache.hadoop.yarn.api.records.Container container)
void	launchContainer(org.apache.hadoop.yarn.api.records.Container container, LaunchSpec taskSpec)
void	register(String trackingUrl)
void	releaseContainer(org.apache.hadoop.yarn.api.records.Container container)
void	removeBlacklist(String nodeName)
void	removeContainerRequest(org.apache.hadoop.yarn.client.api.AMRMClient.ContainerRequest containerRequest)
org.apache.hadoop.yarn.client.api.AMRMClient.ContainerRequest	requestContainer(ContainerRequestSpec containerSpec)
void	start(org.apache.hadoop.yarn.client.api.async.AMRMClientAsync.CallbackHandler resourceCallback, org.apache.hadoop.yarn.client.api.async.NMClientAsync.CallbackHandler nodeCallback)
boolean	supportsDiskResource()

Methods inherited from class java.lang.Object

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

Constructor Detail

AMYarnFacadeImpl

public AMYarnFacadeImpl(int pollPeriodMs)

Method Detail

start

public void start(org.apache.hadoop.yarn.client.api.async.AMRMClientAsync.CallbackHandler resourceCallback,
                  org.apache.hadoop.yarn.client.api.async.NMClientAsync.CallbackHandler nodeCallback)

Specified by:

start in interface AMYarnFacade

register

public void register(String trackingUrl)
              throws YarnFacadeException

Specified by:

register in interface AMYarnFacade

Throws:

YarnFacadeException

getTrackingUrl

public String getTrackingUrl()

Specified by:

getTrackingUrl in interface AMYarnFacade

supportsDiskResource

public boolean supportsDiskResource()

Specified by:

supportsDiskResource in interface AMYarnFacade

requestContainer

public org.apache.hadoop.yarn.client.api.AMRMClient.ContainerRequest requestContainer(ContainerRequestSpec containerSpec)

Specified by:

requestContainer in interface AMYarnFacade

launchContainer

public void launchContainer(org.apache.hadoop.yarn.api.records.Container container,
                            LaunchSpec taskSpec)
                     throws YarnFacadeException

Specified by:

launchContainer in interface AMYarnFacade

Throws:

YarnFacadeException

finish

public void finish(boolean succeeded,
                   String msg)
            throws YarnFacadeException

Specified by:

finish in interface AMYarnFacade

Throws:

YarnFacadeException

releaseContainer

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

Specified by:

releaseContainer in interface AMYarnFacade

killContainer

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

Specified by:

killContainer in interface AMYarnFacade

getNodeCount

public int getNodeCount()

Specified by:

getNodeCount in interface AMYarnFacade

getResources

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

Specified by:

getResources in interface AMYarnFacade

removeContainerRequest

public void removeContainerRequest(org.apache.hadoop.yarn.client.api.AMRMClient.ContainerRequest containerRequest)

Specified by:

removeContainerRequest in interface AMYarnFacade

getRegistrationResponse

public org.apache.hadoop.yarn.api.protocolrecords.RegisterApplicationMasterResponse getRegistrationResponse()

Specified by:

getRegistrationResponse in interface AMYarnFacade

blacklistNode

public void blacklistNode(String nodeName)

Specified by:

blacklistNode in interface AMYarnFacade

removeBlacklist

public void removeBlacklist(String nodeName)

Specified by:

removeBlacklist in interface AMYarnFacade

getNodeReports

public List<org.apache.hadoop.yarn.api.records.NodeReport> getNodeReports()
                                                                   throws YarnFacadeException

Specified by:

getNodeReports in interface AMYarnFacade

Throws:

YarnFacadeException

getAppHostReport

public AMYarnFacade.YarnAppHostReport getAppHostReport()

Specified by:

getAppHostReport in interface AMYarnFacade