org.apache.flink.api.common.io

Class LocatableInputSplitAssigner

java.lang.Object

org.apache.flink.api.common.io.LocatableInputSplitAssigner

All Implemented Interfaces:

InputSplitAssigner

@Public
public final class LocatableInputSplitAssigner
extends Object
implements InputSplitAssigner

The locatable input split assigner assigns to each host splits that are local, before assigning splits that are not local.

Constructor Summary

Constructors

Constructor and Description
LocatableInputSplitAssigner(Collection<LocatableInputSplit> splits)
LocatableInputSplitAssigner(LocatableInputSplit[] splits)

Method Summary

Modifier and Type	Method and Description
LocatableInputSplit	getNextInputSplit(String host, int taskId) Returns the next input split that shall be consumed.
int	getNumberOfLocalAssignments()
int	getNumberOfRemoteAssignments()
void	returnInputSplit(List<InputSplit> splits, int taskId) Return the splits to assigner if the task failed to process it.

Methods inherited from class java.lang.Object

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

Constructor Detail

LocatableInputSplitAssigner

public LocatableInputSplitAssigner(Collection<LocatableInputSplit> splits)

LocatableInputSplitAssigner

public LocatableInputSplitAssigner(LocatableInputSplit[] splits)

Method Detail

getNextInputSplit

public LocatableInputSplit getNextInputSplit(String host,
                                             int taskId)

Description copied from interface: InputSplitAssigner

Returns the next input split that shall be consumed. The consumer's host is passed as a parameter to allow localized assignments.

Specified by:

getNextInputSplit in interface InputSplitAssigner

Parameters:

host - The host address of split requesting task.

taskId - The id of the split requesting task.

Returns:

the next input split to be consumed, or null if no more splits remain.

returnInputSplit

public void returnInputSplit(List<InputSplit> splits,
                             int taskId)

Description copied from interface: InputSplitAssigner

Return the splits to assigner if the task failed to process it.

Specified by:

returnInputSplit in interface InputSplitAssigner

Parameters:

splits - The list of input splits to be returned.

taskId - The id of the task that failed to process the input splits.

getNumberOfLocalAssignments

public int getNumberOfLocalAssignments()

getNumberOfRemoteAssignments

public int getNumberOfRemoteAssignments()