[compat][server][client][test] Global RT DIV improvement (part 2): Chunking support for DIV message #1257
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
…unking support for DIV message
This change mainly focuses on adding chunking support for DIV messages when they are produced to Kafka topics,
as the size of DIV message can be large. We leverage today's chunking mechanism for regular records and extend
it to support DIV with the following modifications:
1. All the DIV messages are of type {@link MessageType#CONTROL_MESSAGE_DIV} in its KafkaKey and their corresponding
KafkaMessageEnvelope uses Put as the payload.
2. Inside the Put payload, the actual message is stored in the putValue field and the schemaId can have the following cases:
- If the DIV message is non-chunked, the schemaId is set to GLOBAL_DIV_STATE.
- If the DIV message is chunk message, the schemaId is set to CHUNK.
- If the DIV message is a chunk manifest message, the schemaId is set to CHUNKED_VALUE_MANIFEST.
3. ChunkAssembler is adapted, on the receiver side, to buffer, assemble, and deserialize DIV messages (chunked/non-chunked).
lusong64
reviewed
Oct 23, 2024
...ts/da-vinci-client/src/main/java/com/linkedin/davinci/kafka/consumer/StoreIngestionTask.java
Outdated
Show resolved
Hide resolved
lusong64
reviewed
Oct 23, 2024
...ts/da-vinci-client/src/main/java/com/linkedin/davinci/kafka/consumer/StoreIngestionTask.java
Outdated
Show resolved
Hide resolved
lusong64
reviewed
Oct 23, 2024
lusong64
reviewed
Oct 23, 2024
...a-vinci-client/src/test/java/com/linkedin/davinci/kafka/consumer/StoreIngestionTaskTest.java
Outdated
Show resolved
Hide resolved
sixpluszero
changed the title
sixpluszero
reviewed
Nov 7, 2024
| @@ -0,0 +1,85 @@ | |||
| { | |||
There was a problem hiding this comment.
let's start with v1 and follow the folder naming convention
| * Note that if the passed-in record is a regular record (not chunked), we will return the record after | ||
| * deserializing it without buffering it in memory. | ||
| */ | ||
| public <T> T bufferAndAssembleRecord( |
There was a problem hiding this comment.
I see there already a method that's doing almost exactly the same as this one, can we try to reuse that method and just pass in avro protocol's deserializer instead?
| @@ -1083,6 +1089,13 @@ private int handleSingleMessage( | |||
| record.getTopicPartition().getPartitionNumber(), | |||
| partitionConsumptionStateMap.get(topicPartition.getPartitionNumber())); | |||
| } | |||
| } else if (record.getKey().isDivControlMessage()) { | |||
| // This is a control message from the DIV topic, process it and return early. | |||
There was a problem hiding this comment.
I am a bit confused about this "DIV topic" part. Do you mean follower see a DIV control message from local VT?
| "Leader for replica: %s received a div control message in remote version topic. Skipping the message.", | ||
| partitionConsumptionState.getReplicaId()); | ||
| if (!REDUNDANT_LOGGING_FILTER.isRedundantException(msg)) { | ||
| LOGGER.info(msg); |
There was a problem hiding this comment.
I think we can just skip it without logging as it is part of native replication from VT not from RT
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This change mainly focuses on adding chunking support for DIV messages when they are produced to Kafka topics, as the size of DIV message can be large. We leverage today's chunking mechanism for regular records and extend it to support DIV with the following modifications:
How was this PR tested?
Does this PR introduce any user-facing changes?