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

Class TieredStorageProducerClient

java.lang.Object

org.apache.flink.runtime.io.network.partition.hybrid.tiered.storage.TieredStorageProducerClient

public class TieredStorageProducerClient
extends Object

Client of the Tiered Storage used by the producer.

Constructor Summary

Constructors

Constructor	Description
TieredStorageProducerClient(int numSubpartitions, boolean isBroadcastOnly, BufferAccumulator bufferAccumulator, BufferCompressor bufferCompressor, List<TierProducerAgent> tierProducerAgents)

Method Summary

Modifier and Type	Method	Description
void	close()
void	setMetricStatisticsUpdater(Consumer<TieredStorageProducerMetricUpdate> metricStatisticsUpdater)
void	write(ByteBuffer record, TieredStorageSubpartitionId subpartitionId, Buffer.DataType dataType, boolean isBroadcast)	Write records to the producer client.

Methods inherited from class java.lang.Object

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

Constructor Detail

TieredStorageProducerClient

public TieredStorageProducerClient(int numSubpartitions,
                                   boolean isBroadcastOnly,
                                   BufferAccumulator bufferAccumulator,
                                   @Nullable
                                   BufferCompressor bufferCompressor,
                                   List<TierProducerAgent> tierProducerAgents)

Method Detail

write

public void write(ByteBuffer record,
                  TieredStorageSubpartitionId subpartitionId,
                  Buffer.DataType dataType,
                  boolean isBroadcast)
           throws IOException

Write records to the producer client. The BufferAccumulator will accumulate the records into buffers.

Note that isBroadcast indicates whether the record is broadcast, while isBroadcastOnly indicates whether the result partition is broadcast-only. When the result partition is not broadcast-only and the record is a broadcast record, the record will be written to all the subpartitions.

Parameters:

record - the written record data

subpartitionId - the subpartition identifier

dataType - the data type of the record

isBroadcast - whether the record is a broadcast record

Throws:

IOException

setMetricStatisticsUpdater

public void setMetricStatisticsUpdater(Consumer<TieredStorageProducerMetricUpdate> metricStatisticsUpdater)

close

public void close()