Package org.apache.hadoop.yarn.server

Class AMHeartbeatRequestHandler

java.lang.Object

java.lang.Thread

org.apache.hadoop.yarn.server.AMHeartbeatRequestHandler

All Implemented Interfaces:

java.lang.Runnable

public class AMHeartbeatRequestHandler
extends java.lang.Thread

Extends Thread and provides an implementation that is used for processing the AM heart beat request asynchronously and sending back the response using the callback method registered with the system.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	AMHeartbeatRequestHandler.AsyncAllocateRequestInfo	Data structure that encapsulates AllocateRequest and AsyncCallback instance.
class	AMHeartbeatRequestHandler.HeartBeatThreadUncaughtExceptionHandler	Uncaught exception handler for the background heartbeat thread.

Nested classes/interfaces inherited from class java.lang.Thread

java.lang.Thread.State, java.lang.Thread.UncaughtExceptionHandler

Field Summary

Fields

Modifier and Type	Field	Description
static org.slf4j.Logger	LOG

Fields inherited from class java.lang.Thread

MAX_PRIORITY, MIN_PRIORITY, NORM_PRIORITY

Constructor Summary

Constructors

Constructor	Description
AMHeartbeatRequestHandler(org.apache.hadoop.conf.Configuration conf, org.apache.hadoop.yarn.api.records.ApplicationId applicationId, AMRMClientRelayer rmProxyRelayer)

Method Summary

Modifier and Type	Method	Description
void	allocateAsync(org.apache.hadoop.yarn.api.protocolrecords.AllocateRequest request, org.apache.hadoop.yarn.util.AsyncCallback<org.apache.hadoop.yarn.api.protocolrecords.AllocateResponse> callback)	Sends the specified heart beat request to the resource manager and invokes the callback asynchronously with the response.
void	drainHeartbeatThread()
int	getRequestQueueSize()
void	resetLastResponseId()	Reset the lastResponseId to zero.
void	run()
void	setUGI(org.apache.hadoop.security.UserGroupInformation ugi)	Set the UGI for RM connection.
void	shutdown()	Shutdown the thread.

Methods inherited from class java.lang.Thread

activeCount, checkAccess, clone, countStackFrames, currentThread, dumpStack, enumerate, getAllStackTraces, getContextClassLoader, getDefaultUncaughtExceptionHandler, getId, getName, getPriority, getStackTrace, getState, getThreadGroup, getUncaughtExceptionHandler, holdsLock, interrupt, interrupted, isAlive, isDaemon, isInterrupted, join, join, join, onSpinWait, resume, setContextClassLoader, setDaemon, setDefaultUncaughtExceptionHandler, setName, setPriority, setUncaughtExceptionHandler, sleep, sleep, start, stop, suspend, toString, yield

Methods inherited from class java.lang.Object

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

Field Detail

LOG

public static final org.slf4j.Logger LOG

Constructor Detail

AMHeartbeatRequestHandler

public AMHeartbeatRequestHandler(org.apache.hadoop.conf.Configuration conf,
                                 org.apache.hadoop.yarn.api.records.ApplicationId applicationId,
                                 AMRMClientRelayer rmProxyRelayer)

Method Detail

shutdown

public void shutdown()

Shutdown the thread.

run

public void run()

Specified by:

run in interface java.lang.Runnable

Overrides:

run in class java.lang.Thread

resetLastResponseId

public void resetLastResponseId()

Reset the lastResponseId to zero.

setUGI

public void setUGI(org.apache.hadoop.security.UserGroupInformation ugi)

Set the UGI for RM connection.

Parameters:

ugi - UserGroupInformation.

allocateAsync

public void allocateAsync(org.apache.hadoop.yarn.api.protocolrecords.AllocateRequest request,
                          org.apache.hadoop.yarn.util.AsyncCallback<org.apache.hadoop.yarn.api.protocolrecords.AllocateResponse> callback)
                   throws org.apache.hadoop.yarn.exceptions.YarnException

Sends the specified heart beat request to the resource manager and invokes the callback asynchronously with the response.

Parameters:

request - the allocate request

callback - the callback method for the request

Throws:

org.apache.hadoop.yarn.exceptions.YarnException - if registerAM is not called yet

drainHeartbeatThread

@VisibleForTesting
public void drainHeartbeatThread()

getRequestQueueSize

@VisibleForTesting
public int getRequestQueueSize()