Package org.apache.flink.runtime.io.network.partition.hybrid

Class HsSubpartitionConsumer

java.lang.Object

org.apache.flink.runtime.io.network.partition.hybrid.HsSubpartitionConsumer

All Implemented Interfaces:

HsSubpartitionConsumerInternalOperations, ResultSubpartitionView

public class HsSubpartitionConsumer
extends Object
implements ResultSubpartitionView, HsSubpartitionConsumerInternalOperations

The read view of HsResultPartition, data can be read from memory or disk.

Nested Class Summary

Nested classes/interfaces inherited from interface org.apache.flink.runtime.io.network.partition.ResultSubpartitionView

ResultSubpartitionView.AvailabilityWithBacklog

Constructor Summary

Constructors

Constructor	Description
HsSubpartitionConsumer(BufferAvailabilityListener availabilityListener)

Method Summary

Modifier and Type	Method	Description
void	acknowledgeAllDataProcessed()
ResultSubpartitionView.AvailabilityWithBacklog	getAvailabilityAndBacklog(boolean isCreditAvailable)	Get the availability and backlog of the view.
int	getConsumingOffset(boolean withLock)	Get the latest consuming offset of the subpartition.
Throwable	getFailureCause()	ResultSubpartitionView can decide whether the failure cause should be reported to consumer as failure (primary failure) or ProducerFailedException (secondary failure).
ResultSubpartition.BufferAndBacklog	getNextBuffer()	Returns the next Buffer instance of this queue iterator.
int	getNumberOfQueuedBuffers()
boolean	isReleased()
void	notifyDataAvailable()	Callback for new data become available.
void	notifyNewBufferSize(int newBufferSize)
int	peekNextBufferSubpartitionId()	Returns the index of the subpartition where the next buffer locates, or -1 if there is no buffer available and the subpartition to be consumed is not determined.
void	releaseAllResources()
void	resumeConsumption()
int	unsynchronizedGetNumberOfQueuedBuffers()

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.apache.flink.runtime.io.network.partition.ResultSubpartitionView

notifyPriorityEvent, notifyRequiredSegmentId

Constructor Detail

HsSubpartitionConsumer

public HsSubpartitionConsumer(BufferAvailabilityListener availabilityListener)

Method Detail

getNextBuffer

@Nullable
public ResultSubpartition.BufferAndBacklog getNextBuffer()

Description copied from interface: ResultSubpartitionView

Returns the next Buffer instance of this queue iterator.

If there is currently no instance available, it will return null. This might happen for example when a pipelined queue producer is slower than the consumer or a spilled queue needs to read in more data.

Important: The consumer has to make sure that each buffer instance will eventually be recycled with Buffer.recycleBuffer() after it has been consumed.

Specified by:

getNextBuffer in interface ResultSubpartitionView

notifyDataAvailable

public void notifyDataAvailable()

Description copied from interface: HsSubpartitionConsumerInternalOperations

Callback for new data become available.

Specified by:

notifyDataAvailable in interface HsSubpartitionConsumerInternalOperations

Specified by:

notifyDataAvailable in interface ResultSubpartitionView

getAvailabilityAndBacklog

public ResultSubpartitionView.AvailabilityWithBacklog getAvailabilityAndBacklog(boolean isCreditAvailable)

Description copied from interface: ResultSubpartitionView

Get the availability and backlog of the view. The availability represents if the view is ready to get buffer from it. The backlog represents the number of available data buffers.

Specified by:

getAvailabilityAndBacklog in interface ResultSubpartitionView

Parameters:

isCreditAvailable - the availability of credits for this ResultSubpartitionView.

Returns:

availability and backlog.

releaseAllResources

public void releaseAllResources()
                         throws IOException

Specified by:

releaseAllResources in interface ResultSubpartitionView

Throws:

IOException

isReleased

public boolean isReleased()

Specified by:

isReleased in interface ResultSubpartitionView

getConsumingOffset

public int getConsumingOffset(boolean withLock)

Description copied from interface: HsSubpartitionConsumerInternalOperations

Get the latest consuming offset of the subpartition.

Specified by:

getConsumingOffset in interface HsSubpartitionConsumerInternalOperations

Parameters:

withLock - If true, read the consuming offset outside the guarding of lock. This is sometimes desired to avoid lock contention, if the caller does not depend on any other states to change atomically with the consuming offset.

Returns:

latest consuming offset.

getFailureCause

public Throwable getFailureCause()

Description copied from interface: ResultSubpartitionView

ResultSubpartitionView can decide whether the failure cause should be reported to consumer as failure (primary failure) or ProducerFailedException (secondary failure). Secondary failure can be reported only if producer (upstream task) is guaranteed to failover.

BEWARE: Incorrectly reporting failure cause as primary failure, can hide the root cause of the failure from the user.

Specified by:

getFailureCause in interface ResultSubpartitionView

resumeConsumption

public void resumeConsumption()

Specified by:

resumeConsumption in interface ResultSubpartitionView

acknowledgeAllDataProcessed

public void acknowledgeAllDataProcessed()

Specified by:

acknowledgeAllDataProcessed in interface ResultSubpartitionView

unsynchronizedGetNumberOfQueuedBuffers

public int unsynchronizedGetNumberOfQueuedBuffers()

Specified by:

unsynchronizedGetNumberOfQueuedBuffers in interface ResultSubpartitionView

getNumberOfQueuedBuffers

public int getNumberOfQueuedBuffers()

Specified by:

getNumberOfQueuedBuffers in interface ResultSubpartitionView

notifyNewBufferSize

public void notifyNewBufferSize(int newBufferSize)

Specified by:

notifyNewBufferSize in interface ResultSubpartitionView

peekNextBufferSubpartitionId

public int peekNextBufferSubpartitionId()

Description copied from interface: ResultSubpartitionView

Returns the index of the subpartition where the next buffer locates, or -1 if there is no buffer available and the subpartition to be consumed is not determined.

Specified by:

peekNextBufferSubpartitionId in interface ResultSubpartitionView