edu.snu.coral.runtime.executor.datatransfer

Class DataTransferFactory

java.lang.Object

edu.snu.coral.runtime.executor.datatransfer.DataTransferFactory

public final class DataTransferFactory
extends Object

A factory that produces InputReader and OutputWriter.

Constructor Summary

Constructors

Constructor and Description
DataTransferFactory(int hashRangeMultiplier, BlockManagerWorker blockManagerWorker)

Method Summary

Modifier and Type	Method and Description
InputReader	createLocalReader(int dstTaskIdx, RuntimeEdge runtimeEdge) Creates a local InputReader between two task in a single task group.
OutputWriter	createLocalWriter(Task srcTask, int srcTaskIdx, RuntimeEdge<?> runtimeEdge) Creates an local OutputWriter between two task in a single task group.
InputReader	createReader(int dstTaskIdx, IRVertex srcIRVertex, RuntimeEdge runtimeEdge) Creates an InputReader between two stages.
OutputWriter	createWriter(Task srcTask, int srcTaskIdx, IRVertex dstIRVertex, RuntimeEdge<?> runtimeEdge) Creates an OutputWriter between two stages.

Methods inherited from class java.lang.Object

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

Constructor Detail

DataTransferFactory

@Inject
public DataTransferFactory(int hashRangeMultiplier,
                                   BlockManagerWorker blockManagerWorker)

Method Detail

createWriter

public OutputWriter createWriter(Task srcTask,
                                 int srcTaskIdx,
                                 @Nullable
                                 IRVertex dstIRVertex,
                                 RuntimeEdge<?> runtimeEdge)

Creates an OutputWriter between two stages.

Parameters:

srcTask - the Task that outputs the data to be written.

srcTaskIdx - the index of the source task.

dstIRVertex - the IRVertex that will take the output data as its input.

runtimeEdge - that connects the srcTask to the tasks belonging to dstIRVertex.

Returns:

the OutputWriter created.

createLocalWriter

public OutputWriter createLocalWriter(Task srcTask,
                                      int srcTaskIdx,
                                      RuntimeEdge<?> runtimeEdge)

Creates an local OutputWriter between two task in a single task group.

Parameters:

srcTask - the Task that outputs the data to be written.

srcTaskIdx - the index of the source task.

runtimeEdge - that connects the srcTask to the tasks belonging to dstIRVertex.

Returns:

the OutputWriter created.

createReader

public InputReader createReader(int dstTaskIdx,
                                @Nullable
                                IRVertex srcIRVertex,
                                RuntimeEdge runtimeEdge)

Creates an InputReader between two stages.

Parameters:

dstTaskIdx - the index of the destination task.

srcIRVertex - the IRVertex that output the data to be read.

runtimeEdge - that connects the tasks belonging to srcIRVertex to dstTask.

Returns:

the InputReader created.

createLocalReader

public InputReader createLocalReader(int dstTaskIdx,
                                     RuntimeEdge runtimeEdge)

Creates a local InputReader between two task in a single task group.

Parameters:

dstTaskIdx - the index of the destination task.

runtimeEdge - that connects the tasks belonging to srcRuntimeVertex to dstTask.

Returns:

the InputReader created.