MINOR: Optimize map lookup efficiency with getOrDefault (#20229)

Optimized `getRemainingRecords()` method by replacing inefficient
`containsKey() + get()` pattern with `getOrDefault()` to reduce map
lookups from 2 to 1 per partition.

Reviewers: Chia-Ping Tsai <chia7712@gmail.com>
This commit is contained in:
Now 2025-09-20 12:52:29 +09:00 committed by GitHub
parent 57e9f98e15
commit c49ab6b4ae
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 5 deletions

View File

@ -216,11 +216,7 @@ public class ExactlyOnceMessageProcessor extends Thread implements ConsumerRebal
}
return consumer.assignment().stream().mapToLong(partition -> {
long currentPosition = consumer.position(partition);
if (fullEndOffsets.containsKey(partition)) {
return fullEndOffsets.get(partition) - currentPosition;
} else {
return 0;
}
return fullEndOffsets.getOrDefault(partition, currentPosition) - currentPosition;
}).sum();
}