edu.snu.coral.runtime.executor.datatransfer

Class InputReader

java.lang.Object

edu.snu.coral.runtime.executor.datatransfer.DataTransfer

edu.snu.coral.runtime.executor.datatransfer.InputReader

public final class InputReader
extends DataTransfer

Represents the input data transfer to a task. TODO #492: Modularize the data communication pattern.

Constructor Summary

Constructors

Constructor and Description
InputReader(int dstTaskIndex, IRVertex srcVertex, RuntimeEdge runtimeEdge, BlockManagerWorker blockManagerWorker)

Method Summary

Modifier and Type	Method and Description
static Iterator	combineFutures(List<CompletableFuture<Iterator>> futures) Combine the given list of futures.
RuntimeEdge	getRuntimeEdge()
CompletableFuture<Object>	getSideInput()
int	getSourceParallelism() Get the parallelism of the source task.
String	getSrcIrVertexId()
boolean	isSideInputReader()
List<CompletableFuture<Iterator>>	read() Reads input data depending on the communication pattern of the srcVertex.

Methods inherited from class edu.snu.coral.runtime.executor.datatransfer.DataTransfer

getId

Methods inherited from class java.lang.Object

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

Constructor Detail

InputReader

public InputReader(int dstTaskIndex,
                   @Nullable
                   IRVertex srcVertex,
                   RuntimeEdge runtimeEdge,
                   BlockManagerWorker blockManagerWorker)

Method Detail

read

public List<CompletableFuture<Iterator>> read()

Reads input data depending on the communication pattern of the srcVertex.

Returns:

the read data.

getRuntimeEdge

public RuntimeEdge getRuntimeEdge()

getSrcIrVertexId

public String getSrcIrVertexId()

isSideInputReader

public boolean isSideInputReader()

getSideInput

public CompletableFuture<Object> getSideInput()

getSourceParallelism

public int getSourceParallelism()

Get the parallelism of the source task.

Returns:

the parallelism of the source task.

combineFutures

public static Iterator combineFutures(List<CompletableFuture<Iterator>> futures)
                               throws ExecutionException,
                                      InterruptedException

Combine the given list of futures.

Parameters:

futures - to combine.

Returns:

the combined iterable of elements.

Throws:

ExecutionException - when fail to get results from futures.

InterruptedException - when interrupted during getting results from futures.