org.apache.tez.dag.api

Class VertexManagerPlugin

java.lang.Object

org.apache.tez.dag.api.VertexManagerPlugin

@InterfaceStability.Unstable
@InterfaceAudience.Public
public abstract class VertexManagerPlugin
extends Object

Interface to plugin user logic into the VertexManager to implement runtime scheduling optimizations and graph reconfiguration. The plugin will be notified of interesting events in the vertex execution life cycle and can respond to them by via the context object

Constructor Summary

Constructors

Constructor and Description
VertexManagerPlugin(VertexManagerPluginContext context) Crete an instance of the VertexManagerPlugin.

Method Summary

Methods

Modifier and Type	Method and Description
VertexManagerPluginContext	getContext() Return ahe VertexManagerPluginContext for this specific instance of the vertex manager.
abstract void	initialize() Initialize the plugin.
abstract void	onRootVertexInitialized(String inputName, InputDescriptor inputDescriptor, List<Event> events) Notification that the inputs of this vertex have initialized
abstract void	onSourceTaskCompleted(String srcVertexName, Integer taskId) Notification of a source vertex completion.
abstract void	onVertexManagerEventReceived(VertexManagerEvent vmEvent) Notification of an event directly sent to this vertex manager
abstract void	onVertexStarted(Map<String,List<Integer>> completions) Notification that the vertex is ready to start running tasks

Methods inherited from class java.lang.Object

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

Constructor Detail

VertexManagerPlugin

public VertexManagerPlugin(VertexManagerPluginContext context)

Crete an instance of the VertexManagerPlugin. Classes extending this to create a VertexManagerPlugin, must provide the same constructor so that Tez can create an instance of the class at runtime.

Parameters:

context - vertex manager plugin context which can be used to access the payload, vertex properties, etc

Method Detail

initialize

public abstract void initialize()

Initialize the plugin. Called when the vertex is initializing. This happens after all source vertices and inputs have initialized

onVertexStarted

public abstract void onVertexStarted(Map<String,List<Integer>> completions)

Notification that the vertex is ready to start running tasks

Parameters:

completions - Source vertices and all their tasks that have already completed

onSourceTaskCompleted

public abstract void onSourceTaskCompleted(String srcVertexName,
                         Integer taskId)

Notification of a source vertex completion.

Parameters:

srcVertexName -

taskId - Index of the task that completed

onVertexManagerEventReceived

public abstract void onVertexManagerEventReceived(VertexManagerEvent vmEvent)

Notification of an event directly sent to this vertex manager

Parameters:

vmEvent -

onRootVertexInitialized

public abstract void onRootVertexInitialized(String inputName,
                           InputDescriptor inputDescriptor,
                           List<Event> events)

Notification that the inputs of this vertex have initialized

Parameters:

inputName -

inputDescriptor -

events -

getContext

public final VertexManagerPluginContext getContext()

Return ahe VertexManagerPluginContext for this specific instance of the vertex manager.

Returns:

the VertexManagerPluginContext for the input