Package com.linkedin.davinci.kafka.consumer

Class LeaderProducerCallback

java.lang.Object

com.linkedin.davinci.kafka.consumer.LeaderProducerCallback

All Implemented Interfaces:

PubSubProducerCallback, ChunkAwareCallback

Direct Known Subclasses:

ActiveActiveProducerCallback

public class LeaderProducerCallback
extends Object
implements ChunkAwareCallback

Field Summary

Fields

Modifier and Type	Field	Description
protected static final ChunkedValueManifestSerializer	CHUNKED_VALUE_MANIFEST_SERIALIZER
protected ChunkedValueManifest	chunkedRmdManifest
protected static final ByteBuffer	EMPTY_BYTE_BUFFER
protected final LeaderFollowerStoreIngestionTask	ingestionTask
protected final LeaderProducedRecordContext	leaderProducedRecordContext
protected ChunkedValueManifest	oldRmdManifest
protected ChunkedValueManifest	oldValueManifest

Constructor Summary

Constructors

Constructor	Description
LeaderProducerCallback(LeaderFollowerStoreIngestionTask ingestionTask, DefaultPubSubMessage sourceConsumerRecord, PartitionConsumptionState partitionConsumptionState, LeaderProducedRecordContext leaderProducedRecordContext, int partition, String kafkaUrl, long beforeProcessingRecordTimestampNs)

Method Summary

Modifier and Type	Method	Description
LeaderFollowerStoreIngestionTask	getIngestionTask()
PartitionConsumptionState	getPartitionConsumptionState()
DefaultPubSubMessage	getSourceConsumerRecord()
protected Put	instantiateManifestPut()
protected Put	instantiateRmdChunkPut()
protected Put	instantiateValueChunkPut()
void	onCompletion(PubSubProduceResult produceResult, Exception e)	exception will be null if request was completed without an error.
void	setChunkingInfo(byte[] key, ByteBuffer[] valueChunks, ChunkedValueManifest chunkedValueManifest, ByteBuffer[] rmdChunks, ChunkedValueManifest chunkedRmdManifest, ChunkedValueManifest oldValueManifest, ChunkedValueManifest oldRmdManifest)	For all PUT operations, the is guaranteed to be passed via this function, whether chunking is enabled or not, and whether the value is chunked or not.
void	setOnCompletionFunction(Consumer<PubSubProduceResult> onCompletionFunction)

Methods inherited from class java.lang.Object

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

Methods inherited from interface com.linkedin.venice.pubsub.api.PubSubProducerCallback

setInternalCallback

Field Details

CHUNKED_VALUE_MANIFEST_SERIALIZER

protected static final ChunkedValueManifestSerializer CHUNKED_VALUE_MANIFEST_SERIALIZER

EMPTY_BYTE_BUFFER

protected static final ByteBuffer EMPTY_BYTE_BUFFER

ingestionTask

protected final LeaderFollowerStoreIngestionTask ingestionTask

leaderProducedRecordContext

protected final LeaderProducedRecordContext leaderProducedRecordContext

chunkedRmdManifest

protected ChunkedValueManifest chunkedRmdManifest

oldValueManifest

protected ChunkedValueManifest oldValueManifest

oldRmdManifest

protected ChunkedValueManifest oldRmdManifest

Constructor Details

LeaderProducerCallback

public LeaderProducerCallback(LeaderFollowerStoreIngestionTask ingestionTask,
 DefaultPubSubMessage sourceConsumerRecord,
 PartitionConsumptionState partitionConsumptionState,
 LeaderProducedRecordContext leaderProducedRecordContext,
 int partition,
 String kafkaUrl,
 long beforeProcessingRecordTimestampNs)

Method Details

onCompletion

public void onCompletion(PubSubProduceResult produceResult,
 Exception e)

Description copied from interface: PubSubProducerCallback

exception will be null if request was completed without an error.

Specified by:

onCompletion in interface PubSubProducerCallback

setChunkingInfo

public void setChunkingInfo(byte[] key,
 ByteBuffer[] valueChunks,
 ChunkedValueManifest chunkedValueManifest,
 ByteBuffer[] rmdChunks,
 ChunkedValueManifest chunkedRmdManifest,
 ChunkedValueManifest oldValueManifest,
 ChunkedValueManifest oldRmdManifest)

Description copied from interface: ChunkAwareCallback

For all PUT operations, the is guaranteed to be passed via this function, whether chunking is enabled or not, and whether the value is chunked or not. The other two parameters are null if the value is not chunked.

Specified by:

setChunkingInfo in interface ChunkAwareCallback

Parameters:

key - A byte[] corresponding to the top-level key written to Kafka, potentially including a chunking suffix

valueChunks - An array of ByteBuffer where the backing array has sufficient headroom to prepend Venice's header

chunkedValueManifest - The ChunkedValueManifest of the new chunked value

rmdChunks - An array of ByteBuffer where the backing array has sufficient headroom to prepend Venice's header

chunkedRmdManifest - The ChunkedValueManifest of the new chunked RMD

oldValueManifest - The ChunkedValueManifest of the previous chunked value

oldRmdManifest - The ChunkedValueManifest of the previous chunked RMD

instantiateValueChunkPut

protected Put instantiateValueChunkPut()

instantiateRmdChunkPut

protected Put instantiateRmdChunkPut()

instantiateManifestPut

protected Put instantiateManifestPut()

setOnCompletionFunction

public void setOnCompletionFunction(Consumer<PubSubProduceResult> onCompletionFunction)

getPartitionConsumptionState

public PartitionConsumptionState getPartitionConsumptionState()

getSourceConsumerRecord

public DefaultPubSubMessage getSourceConsumerRecord()

getIngestionTask

public LeaderFollowerStoreIngestionTask getIngestionTask()