-
Notifications
You must be signed in to change notification settings - Fork 22
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Issue 89: Add Batch Reader Support #92
base: master
Are you sure you want to change the base?
Issue 89: Add Batch Reader Support #92
Conversation
Signed-off-by: David Maddison <david.maddison@dell.com>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good. I have a few comments.
return batchReaders ? getBatchConsumers() : getStreamingConsumers(); | ||
} | ||
|
||
public List<ReaderWorker> getBatchConsumers() { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
A key difference between the batch and streaming readers in this benchmark is that there can be multiple benchmark processes reading together through a reader group. Adding more processes will increase the parallelism and data will not be read multiple times.
The batch reader does not perform this coordination so if one attempts to run multiple reader benchmark processes, each process will read the entire stream.
This behavior is reasonable but it should be documented.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
That's a good point and it was a subtly in the code that I missed, i.e. that the ReaderGroup name is devised from the steam name itself and therefore there is implicit coordination between multiple distributed processes.
if (recreate) {
rdGrpName = streamName + startTime;
} else {
rdGrpName = streamName + "RdGrp";
}
Obviously, as you point out, this is a bit harder to achieve with the BatchClient as the coordination is basically left up to the processor itself. I'll document this current restriction and find a way of distributing the processing in another PR.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
If you need to distribute the processing (in another PR), consider accepting parameters for the process index and the number of processes. Then each process can sort the available segment ranges and take a deterministic subset.
* | ||
* @return A list of lists, each list representing the segments assigned to that consumer | ||
*/ | ||
private List<List<SegmentRange>> assignSegmentsToConsumers(List<SegmentRange> segmentRanges, int consumers) { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Consider using com.google.common.collect.Lists.partition
instead of this.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Originally I did use Google Lists.partition
but the problem with that API is that partitions the list based on a specific partition size and NOT (as I'd assumed) by the number of required partitions.
This lead to getting into calculating the required size of partitions along with handling the cases where the number of SegmentRanges wasn't completely divisible by the number of consumers. In the end it felt a lot simpler to partition the list as done here.
…ed work Signed-off-by: David Maddison <david.maddison@dell.com>
Signed-off-by: David Maddison <david.maddison@dell.com>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks good, I left some comments.
@@ -100,6 +100,7 @@ public void EventsReaderRW() throws IOException { | |||
i++; | |||
} | |||
} | |||
} catch(WorkerCompleteException ignore) { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Don't we need to log anything here?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
No, because the Exception only exists to indicate to the logic that the reader has completed, it's not an actual exception or an error condition.
The PravegaBatchReaderWorker
(which is the only worker that throws this exception) will log that it competed everything, so another log would just be redundant.
) | ||
.collect(Collectors.toList()); | ||
} else { | ||
readers = null; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Is this case possible? If the user does not specify consumers
, we could take as default the number of segments or just 1
. But if the user sets consumerCount <= 0
, the tool should throw an error even before reaching this point telling the user that the input is incorrect, right?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This logic follows the same pattern as the streaming consumers. The problem is that the test doesn't know if consumers are required, it only knows that because consumers
> 0. If consumers
was defaulted to 1 then a test would always have a consumer, which is not desirable.
Signed-off-by: David Maddison <david.maddison@dell.com>
@maddisondavid can you please have a look to the conflicts so we can merge this one? |
@maddisondavid any updates on this one? |
Change log description
Adds support for using the BatchReader instead of the StreamingReader within consumers
Purpose of the change
Fixes #89
What the code does
When reading historic data from a stream data can be read using either the Streaming Reader or the Batch (bounded) Reader. This adds a new
-batchreaders
option that signifies that the stream should be consumed using the BatchClient. In this mode all the segments from with a Stream bound are supplied by the Pravega Client and the consumers are free to consume them in any order.The existing consumer has been renamed
PravegaStreamingReaderWorker
and a newPravegaBatchReaderWorker
has been introduced. Each reader is assigned a certain number of the Batch segments with an attempt to give each consumer an equal number of the segments. If the segment count is lower than the number of specified consumers then the consumer count is reduced so that there is one consumer per Batch segment.How to verify
When running with the
-batchreaders
option the test will log the consumer to segment assignmentsAs the consumers progress through thier assigned segments ranges they will also log progress:
Signed-off-by: David Maddison david.maddison@dell.com