org.apache.tez.runtime.api.impl

Class TezTaskContextImpl

java.lang.Object

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

All Implemented Interfaces:

Closeable, AutoCloseable, org.apache.tez.runtime.api.TaskContext

Direct Known Subclasses:

TezInputContextImpl, TezOutputContextImpl, TezProcessorContextImpl

public abstract class TezTaskContextImpl
extends Object
implements org.apache.tez.runtime.api.TaskContext, Closeable

Field Summary

Fields

Modifier and Type	Field and Description
protected org.apache.tez.dag.api.EntityDescriptor<?>	descriptor
protected MemoryDistributor	initialMemoryDistributor
protected LogicalIOProcessorRuntimeTask	runtimeTask
protected org.apache.tez.dag.records.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, org.apache.tez.dag.records.TezTaskAttemptID taskAttemptID, org.apache.tez.common.counters.TezCounters counters, LogicalIOProcessorRuntimeTask runtimeTask, TezUmbilical tezUmbilical, Map<String,ByteBuffer> serviceConsumerMetadata, Map<String,String> auxServiceEnv, MemoryDistributor memDist, org.apache.tez.dag.api.EntityDescriptor<?> descriptor, org.apache.tez.runtime.api.ObjectRegistry objectRegistry, org.apache.tez.runtime.api.ExecutionContext ExecutionContext, long memAvailable)

Method Summary

Methods

Modifier and Type	Method and Description
void	close()
org.apache.hadoop.yarn.api.records.ApplicationId	getApplicationId()
org.apache.tez.common.counters.TezCounters	getCounters()
int	getDAGAttemptNumber()
int	getDagIdentifier()
String	getDAGName()
org.apache.tez.runtime.api.ExecutionContext	getExecutionContext()
org.apache.tez.runtime.api.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	requestInitialMemory(long size, org.apache.tez.runtime.api.MemoryUpdateCallback callbackHandler)
protected void	signalFatalError(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, sendEvents

Field Detail

taskVertexName

protected final String taskVertexName

taskAttemptID

protected final org.apache.tez.dag.records.TezTaskAttemptID taskAttemptID

runtimeTask

protected final LogicalIOProcessorRuntimeTask runtimeTask

tezUmbilical

protected final TezUmbilical tezUmbilical

initialMemoryDistributor

protected volatile MemoryDistributor initialMemoryDistributor

descriptor

protected final org.apache.tez.dag.api.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,
                                            org.apache.tez.dag.records.TezTaskAttemptID taskAttemptID,
                                            org.apache.tez.common.counters.TezCounters counters,
                                            LogicalIOProcessorRuntimeTask runtimeTask,
                                            TezUmbilical tezUmbilical,
                                            Map<String,ByteBuffer> serviceConsumerMetadata,
                                            Map<String,String> auxServiceEnv,
                                            MemoryDistributor memDist,
                                            org.apache.tez.dag.api.EntityDescriptor<?> descriptor,
                                            org.apache.tez.runtime.api.ObjectRegistry objectRegistry,
                                            org.apache.tez.runtime.api.ExecutionContext ExecutionContext,
                                            long memAvailable)

Method Detail

getApplicationId

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

Specified by:

getApplicationId in interface org.apache.tez.runtime.api.TaskContext

getTaskIndex

public int getTaskIndex()

Specified by:

getTaskIndex in interface org.apache.tez.runtime.api.TaskContext

getDAGAttemptNumber

public int getDAGAttemptNumber()

Specified by:

getDAGAttemptNumber in interface org.apache.tez.runtime.api.TaskContext

getTaskAttemptNumber

public int getTaskAttemptNumber()

Specified by:

getTaskAttemptNumber in interface org.apache.tez.runtime.api.TaskContext

getDAGName

public String getDAGName()

Specified by:

getDAGName in interface org.apache.tez.runtime.api.TaskContext

getTaskVertexName

public String getTaskVertexName()

Specified by:

getTaskVertexName in interface org.apache.tez.runtime.api.TaskContext

getTaskVertexIndex

public int getTaskVertexIndex()

Specified by:

getTaskVertexIndex in interface org.apache.tez.runtime.api.TaskContext

getDagIdentifier

public int getDagIdentifier()

Specified by:

getDagIdentifier in interface org.apache.tez.runtime.api.TaskContext

getCounters

public org.apache.tez.common.counters.TezCounters getCounters()

Specified by:

getCounters in interface org.apache.tez.runtime.api.TaskContext

getVertexParallelism

public int getVertexParallelism()

Specified by:

getVertexParallelism in interface org.apache.tez.runtime.api.TaskContext

getWorkDirs

public String[] getWorkDirs()

Specified by:

getWorkDirs in interface org.apache.tez.runtime.api.TaskContext

getUniqueIdentifier

public String getUniqueIdentifier()

Specified by:

getUniqueIdentifier in interface org.apache.tez.runtime.api.TaskContext

getObjectRegistry

public org.apache.tez.runtime.api.ObjectRegistry getObjectRegistry()

Specified by:

getObjectRegistry in interface org.apache.tez.runtime.api.TaskContext

getServiceConsumerMetaData

public ByteBuffer getServiceConsumerMetaData(String serviceName)

Specified by:

getServiceConsumerMetaData in interface org.apache.tez.runtime.api.TaskContext

getServiceProviderMetaData

@Nullable
public ByteBuffer getServiceProviderMetaData(String serviceName)

Specified by:

getServiceProviderMetaData in interface org.apache.tez.runtime.api.TaskContext

requestInitialMemory

public void requestInitialMemory(long size,
                        org.apache.tez.runtime.api.MemoryUpdateCallback callbackHandler)

Specified by:

requestInitialMemory in interface org.apache.tez.runtime.api.TaskContext

getTotalMemoryAvailableToTask

public long getTotalMemoryAvailableToTask()

Specified by:

getTotalMemoryAvailableToTask in interface org.apache.tez.runtime.api.TaskContext

signalFatalError

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

getExecutionContext

public org.apache.tez.runtime.api.ExecutionContext getExecutionContext()

Specified by:

getExecutionContext in interface org.apache.tez.runtime.api.TaskContext

close

public void close()
           throws IOException

Specified by:

close in interface Closeable

Specified by:

close in interface AutoCloseable

Throws:

IOException