org.apache.tez.dag.library.vertexmanager
Class ShuffleVertexManager.CustomShuffleEdgeManager

java.lang.Object
  org.apache.tez.dag.library.vertexmanager.ShuffleVertexManager.CustomShuffleEdgeManager

All Implemented Interfaces:

org.apache.tez.dag.api.EdgeManager

Enclosing class:

ShuffleVertexManager

public static class ShuffleVertexManager.CustomShuffleEdgeManager
extends Object
implements org.apache.tez.dag.api.EdgeManager

Constructor Summary

ShuffleVertexManager.CustomShuffleEdgeManager()

Method Summary

int	getNumDestinationConsumerTasks(int sourceTaskIndex, int numDestTasks)
int	getNumDestinationTaskPhysicalInputs(int numSourceTasks, int destinationTaskIndex)
int	getNumSourceTaskPhysicalOutputs(int numDestinationTasks, int sourceTaskIndex)
void	initialize(org.apache.tez.dag.api.EdgeManagerContext edgeManagerContext)
void	routeDataMovementEventToDestination(org.apache.tez.runtime.api.events.DataMovementEvent event, int sourceTaskIndex, int numDestinationTasks, Map<Integer,List<Integer>> inputIndicesToTaskIndices)
int	routeInputErrorEventToSource(org.apache.tez.runtime.api.events.InputReadErrorEvent event, int destinationTaskIndex)
void	routeInputSourceTaskFailedEventToDestination(int sourceTaskIndex, int numDestinationTasks, Map<Integer,List<Integer>> inputIndicesToTaskIndices)

Methods inherited from class java.lang.Object

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

Constructor Detail

ShuffleVertexManager.CustomShuffleEdgeManager

public ShuffleVertexManager.CustomShuffleEdgeManager()

Method Detail

initialize

public void initialize(org.apache.tez.dag.api.EdgeManagerContext edgeManagerContext)

Specified by:

initialize in interface org.apache.tez.dag.api.EdgeManager

getNumDestinationTaskPhysicalInputs

public int getNumDestinationTaskPhysicalInputs(int numSourceTasks,
                                               int destinationTaskIndex)

Specified by:

getNumDestinationTaskPhysicalInputs in interface org.apache.tez.dag.api.EdgeManager

getNumSourceTaskPhysicalOutputs

public int getNumSourceTaskPhysicalOutputs(int numDestinationTasks,
                                           int sourceTaskIndex)

Specified by:

getNumSourceTaskPhysicalOutputs in interface org.apache.tez.dag.api.EdgeManager

routeDataMovementEventToDestination

public void routeDataMovementEventToDestination(org.apache.tez.runtime.api.events.DataMovementEvent event,
                                                int sourceTaskIndex,
                                                int numDestinationTasks,
                                                Map<Integer,List<Integer>> inputIndicesToTaskIndices)

Specified by:

routeDataMovementEventToDestination in interface org.apache.tez.dag.api.EdgeManager

routeInputSourceTaskFailedEventToDestination

public void routeInputSourceTaskFailedEventToDestination(int sourceTaskIndex,
                                                         int numDestinationTasks,
                                                         Map<Integer,List<Integer>> inputIndicesToTaskIndices)

Specified by:

routeInputSourceTaskFailedEventToDestination in interface org.apache.tez.dag.api.EdgeManager

routeInputErrorEventToSource

public int routeInputErrorEventToSource(org.apache.tez.runtime.api.events.InputReadErrorEvent event,
                                        int destinationTaskIndex)

Specified by:

routeInputErrorEventToSource in interface org.apache.tez.dag.api.EdgeManager

getNumDestinationConsumerTasks

public int getNumDestinationConsumerTasks(int sourceTaskIndex,
                                          int numDestTasks)

Specified by:

getNumDestinationConsumerTasks in interface org.apache.tez.dag.api.EdgeManager