opensource/rocketmq
3
0
Fork 0
mirror of https://github.com/apache/rocketmq.git synced 2026-08-28 20:09:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[ISSUE #10103] Improve batch polling efficiency in pollIndexRecord method (#10104)

Browse Source
This commit is contained in:
yx9o
2026-03-17 16:22:28 +08:00
committed by GitHub
parent 2bc1783f54
commit ebf1595418
1 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
store/src/main/java/org/apache/rocketmq/store/index/rocksdb/IndexRocksDBStore.java
+11 -11
View File
@@ -321,18 +321,18 @@ public class IndexRocksDBStore implements CommitLogDispatchStore {
private void pollIndexRecord() {
try {
IndexRocksDBRecord firstReq = originIndexMsgQueue.poll(100, TimeUnit.MILLISECONDS);
if (null != firstReq) {
irs.add(firstReq);
while (true) {
IndexRocksDBRecord tmpReq = originIndexMsgQueue.poll(100, TimeUnit.MILLISECONDS);
if (null == tmpReq) {
break;
}
irs.add(tmpReq);
if (irs.size() >= BATCH_SIZE) {
break;
}
if (firstReq == null) {
return;
}
irs.add(firstReq);
originIndexMsgQueue.drainTo(irs, BATCH_SIZE - irs.size());
while (irs.size() < BATCH_SIZE) {
IndexRocksDBRecord tmpReq = originIndexMsgQueue.poll(100, TimeUnit.MILLISECONDS);
if (tmpReq == null) {
break;
}
irs.add(tmpReq);
originIndexMsgQueue.drainTo(irs, BATCH_SIZE - irs.size());
}
} catch (Exception e) {
logError.error("IndexRocksDBStore IndexBuildService error: {}", e.getMessage());