org.apache.tez.runtime.api.impl

Class TezTaskContextImpl

java.lang.Object

org.apache.tez.runtime.api.impl.TezTaskContextImpl

All Implemented Interfaces:

Closeable, AutoCloseable, TaskContext

Direct Known Subclasses:

TezInputContextImpl, TezOutputContextImpl, TezProcessorContextImpl

public abstract class TezTaskContextImpl
extends Object
implements TaskContext, Closeable

Field Summary

Fields

Modifier and Type	Field and Description
protected EntityDescriptor<?>	descriptor
protected MemoryDistributor	initialMemoryDistributor
protected LogicalIOProcessorRuntimeTask	runtimeTask
protected TezTaskAttemptID	taskAttemptID
protected String	taskVertexName
protected TezUmbilical	tezUmbilical

Constructor Summary

Constructors

Constructor and Description

TezTaskContextImpl(org.apache.hadoop.conf.Configuration conf, String[] workDirs, int appAttemptNumber, String dagName, String taskVertexName, int vertexParallelism, TezTaskAttemptID taskAttemptID, TezCounters counters, LogicalIOProcessorRuntimeTask runtimeTask, TezUmbilical tezUmbilical, Map<String,ByteBuffer> serviceConsumerMetadata, Map<String,String> auxServiceEnv, MemoryDistributor memDist, EntityDescriptor<?> descriptor, ObjectRegistry objectRegistry, ExecutionContext ExecutionContext, long memAvailable, TezExecutors sharedExecutor)

Method Summary

Modifier and Type	Method and Description
void	close()
ExecutorService	createTezFrameworkExecutorService(int parallelism, String threadNameFormat)
org.apache.hadoop.yarn.api.records.ApplicationId	getApplicationId()
org.apache.hadoop.conf.Configuration	getContainerConfiguration()
TezCounters	getCounters()
int	getDAGAttemptNumber()
int	getDagIdentifier()
String	getDAGName()
ExecutionContext	getExecutionContext()
ObjectRegistry	getObjectRegistry()
ByteBuffer	getServiceConsumerMetaData(String serviceName)
ByteBuffer	getServiceProviderMetaData(String serviceName)
int	getTaskAttemptNumber()
int	getTaskIndex()
int	getTaskVertexIndex()
String	getTaskVertexName()
long	getTotalMemoryAvailableToTask()
String	getUniqueIdentifier()
int	getVertexParallelism()
String[]	getWorkDirs()
void	notifyProgress()
void	requestInitialMemory(long size, MemoryUpdateCallback callbackHandler)
protected void	signalFailure(TaskFailureType taskFailureType, Throwable t, String message, EventMetaData sourceInfo)
protected void	signalFatalError(Throwable t, String message, EventMetaData sourceInfo)
protected void	signalKillSelf(Throwable t, String message, EventMetaData sourceInfo)

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.apache.tez.runtime.api.TaskContext

fatalError, getUserPayload, killSelf, reportFailure, sendEvents

Field Detail

taskVertexName

protected final String taskVertexName

taskAttemptID

protected final TezTaskAttemptID taskAttemptID

runtimeTask

protected final LogicalIOProcessorRuntimeTask runtimeTask

tezUmbilical

protected final TezUmbilical tezUmbilical

initialMemoryDistributor

protected volatile MemoryDistributor initialMemoryDistributor

descriptor

protected final EntityDescriptor<?> descriptor

Constructor Detail

TezTaskContextImpl

@InterfaceAudience.Private
public TezTaskContextImpl(org.apache.hadoop.conf.Configuration conf,
                                                     String[] workDirs,
                                                     int appAttemptNumber,
                                                     String dagName,
                                                     String taskVertexName,
                                                     int vertexParallelism,
                                                     TezTaskAttemptID taskAttemptID,
                                                     TezCounters counters,
                                                     LogicalIOProcessorRuntimeTask runtimeTask,
                                                     TezUmbilical tezUmbilical,
                                                     Map<String,ByteBuffer> serviceConsumerMetadata,
                                                     Map<String,String> auxServiceEnv,
                                                     MemoryDistributor memDist,
                                                     EntityDescriptor<?> descriptor,
                                                     ObjectRegistry objectRegistry,
                                                     ExecutionContext ExecutionContext,
                                                     long memAvailable,
                                                     TezExecutors sharedExecutor)

Method Detail

getApplicationId

public org.apache.hadoop.yarn.api.records.ApplicationId getApplicationId()

Specified by:

getApplicationId in interface TaskContext

getTaskIndex

public int getTaskIndex()

Specified by:

getTaskIndex in interface TaskContext

getDAGAttemptNumber

public int getDAGAttemptNumber()

Specified by:

getDAGAttemptNumber in interface TaskContext

getTaskAttemptNumber

public int getTaskAttemptNumber()

Specified by:

getTaskAttemptNumber in interface TaskContext

getContainerConfiguration

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

Specified by:

getContainerConfiguration in interface TaskContext

getDAGName

public String getDAGName()

Specified by:

getDAGName in interface TaskContext

getTaskVertexName

public String getTaskVertexName()

Specified by:

getTaskVertexName in interface TaskContext

getTaskVertexIndex

public int getTaskVertexIndex()

Specified by:

getTaskVertexIndex in interface TaskContext

getDagIdentifier

public int getDagIdentifier()

Specified by:

getDagIdentifier in interface TaskContext

getCounters

public TezCounters getCounters()

Specified by:

getCounters in interface TaskContext

getVertexParallelism

public int getVertexParallelism()

Specified by:

getVertexParallelism in interface TaskContext

getWorkDirs

public String[] getWorkDirs()

Specified by:

getWorkDirs in interface TaskContext

getUniqueIdentifier

public String getUniqueIdentifier()

Specified by:

getUniqueIdentifier in interface TaskContext

getObjectRegistry

public ObjectRegistry getObjectRegistry()

Specified by:

getObjectRegistry in interface TaskContext

notifyProgress

public final void notifyProgress()

Specified by:

notifyProgress in interface TaskContext

getServiceConsumerMetaData

public ByteBuffer getServiceConsumerMetaData(String serviceName)

Specified by:

getServiceConsumerMetaData in interface TaskContext

getServiceProviderMetaData

@Nullable
public ByteBuffer getServiceProviderMetaData(String serviceName)

Specified by:

getServiceProviderMetaData in interface TaskContext

requestInitialMemory

public void requestInitialMemory(long size,
                                 MemoryUpdateCallback callbackHandler)

Specified by:

requestInitialMemory in interface TaskContext

getTotalMemoryAvailableToTask

public long getTotalMemoryAvailableToTask()

Specified by:

getTotalMemoryAvailableToTask in interface TaskContext

signalFatalError

protected void signalFatalError(Throwable t,
                                String message,
                                EventMetaData sourceInfo)

signalFailure

protected void signalFailure(TaskFailureType taskFailureType,
                             Throwable t,
                             String message,
                             EventMetaData sourceInfo)

signalKillSelf

protected void signalKillSelf(Throwable t,
                              String message,
                              EventMetaData sourceInfo)

getExecutionContext

public ExecutionContext getExecutionContext()

Specified by:

getExecutionContext in interface TaskContext

createTezFrameworkExecutorService

public ExecutorService createTezFrameworkExecutorService(int parallelism,
                                                         String threadNameFormat)

Specified by:

createTezFrameworkExecutorService in interface TaskContext

close

public void close()
           throws IOException

Specified by:

close in interface Closeable

Specified by:

close in interface AutoCloseable

Throws:

IOException