解决分段拉取消息时可能丢消息的问题

2025-03-05 19:13:29 +08:00
parent f897e640fc
commit 1df192f730
1 changed files with 4 additions and 0 deletions
@@ -900,11 +900,13 @@ public class MemoryMessagesStore implements IMessagesStore {
                session.setPullHistoryMsg(true);
            }

+            long previousCurrent;
            while (true) {
                Map.Entry<Long, Long> entry = maps.higherEntry(current);
                if (entry == null) {
                    break;
                }
+                previousCurrent = current;
                current = entry.getKey();
                long targetMessageId = entry.getValue();

@@ -933,6 +935,8 @@ public class MemoryMessagesStore implements IMessagesStore {
                        if (size >= 512 * 1024) { //3M
                            if(builder.getMessageCount() == 0){
                                builder.addMessage(bundle.getMessage());
+                            } else {
+                                current = previousCurrent;
                            }
                            break;
                        }