org.apache.flink.runtime.io.network.partition.hybrid.tiered.tier.disk

Class DiskTierConsumerAgent

java.lang.Object

org.apache.flink.runtime.io.network.partition.hybrid.tiered.tier.disk.DiskTierConsumerAgent

All Implemented Interfaces:

TierConsumerAgent

public class DiskTierConsumerAgent
extends Object
implements TierConsumerAgent

The data client is used to fetch data from disk tier.

Constructor Summary

Constructors

Constructor and Description
DiskTierConsumerAgent(List<TieredStorageConsumerSpec> tieredStorageConsumerSpecs, TieredStorageNettyService nettyService)

Method Summary

Modifier and Type	Method and Description
void	close() Close the consumer agent.
Optional<Buffer>	getNextBuffer(TieredStoragePartitionId partitionId, TieredStorageSubpartitionId subpartitionId, int segmentId) Get buffer from the consumer agent.
int	peekNextBufferSubpartitionId(TieredStoragePartitionId partitionId, ResultSubpartitionIndexSet indexSet) Returns the index of the subpartition where the next buffer locates, or -1 if there is no buffer available or the subpartition index does not belong to the specified indexSet.
void	registerAvailabilityNotifier(AvailabilityNotifier notifier) Register the notifier to notify the availability of a subpartition.
void	start() Start the consumer agent.

Methods inherited from class java.lang.Object

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

Constructor Detail

DiskTierConsumerAgent

public DiskTierConsumerAgent(List<TieredStorageConsumerSpec> tieredStorageConsumerSpecs,
                             TieredStorageNettyService nettyService)

Method Detail

start

public void start()

Description copied from interface: TierConsumerAgent

Start the consumer agent.

Specified by:

start in interface TierConsumerAgent

registerAvailabilityNotifier

public void registerAvailabilityNotifier(AvailabilityNotifier notifier)

Description copied from interface: TierConsumerAgent

Register the notifier to notify the availability of a subpartition.

Specified by:

registerAvailabilityNotifier in interface TierConsumerAgent

Parameters:

notifier - to notify availability.

peekNextBufferSubpartitionId

public int peekNextBufferSubpartitionId(TieredStoragePartitionId partitionId,
                                        ResultSubpartitionIndexSet indexSet)
                                 throws IOException

Description copied from interface: TierConsumerAgent

Returns the index of the subpartition where the next buffer locates, or -1 if there is no buffer available or the subpartition index does not belong to the specified indexSet.

Specified by:

peekNextBufferSubpartitionId in interface TierConsumerAgent

Parameters:

partitionId - The index of the partition which the returned subpartition should belong to.

indexSet - The indexes of the subpartitions expected.

Throws:

IOException

getNextBuffer

public Optional<Buffer> getNextBuffer(TieredStoragePartitionId partitionId,
                                      TieredStorageSubpartitionId subpartitionId,
                                      int segmentId)

Description copied from interface: TierConsumerAgent

Get buffer from the consumer agent.

Specified by:

getNextBuffer in interface TierConsumerAgent

Parameters:

partitionId - the id of partition.

subpartitionId - the id of subpartition.

segmentId - the id of segment.

Returns:

buffer.

close

public void close()
           throws IOException

Description copied from interface: TierConsumerAgent

Close the consumer agent.

Specified by:

close in interface TierConsumerAgent

Throws:

IOException