Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ public Map<Consumer, List<EntryAndMetadata>> assign(final List<EntryAndMetadata>
if (metadata == null || !metadata.hasUuid() || !metadata.hasChunkId() || !metadata.hasNumChunksFromMsg()) {
consumerToEntries.computeIfAbsent(consumer, __ -> new ArrayList<>()).add(entryAndMetadata);
} else {
final Consumer consumerForUuid = getConsumerForUuid(metadata, consumer, availablePermits);
final Consumer consumerForUuid = getConsumerForUuid(metadata, consumer);
if (consumerForUuid == null) {
unassignedMessageProcessor.accept(entryAndMetadata);
continue;
Expand Down Expand Up @@ -123,9 +123,7 @@ private Consumer getConsumer(final int numConsumers) {
return null;
}

private Consumer getConsumerForUuid(final MessageMetadata metadata,
final Consumer defaultConsumer,
final int currentAvailablePermits) {
private Consumer getConsumerForUuid(final MessageMetadata metadata, final Consumer defaultConsumer) {
final String uuid = metadata.getUuid();
Consumer consumer = uuidToConsumer.get(uuid);
if (consumer == null) {
Expand All @@ -144,7 +142,9 @@ private Consumer getConsumerForUuid(final MessageMetadata metadata,
// The last chunk is received, we should remove the cache
uuidToConsumer.remove(uuid);
}
consumerToPermits.put(consumer, currentAvailablePermits - 1);
// Decrement target consumer's permits, not the loop's local availablePermits — on a cache
// redirect those track different consumers.
consumerToPermits.put(consumer, permits - 1);
return consumer;
}
}
Loading