From 79a29953dd92b9c025a8b9915aeb56d1d78e5607 Mon Sep 17 00:00:00 2001
From: Nicolas Werner <nicolas.werner@hotmail.de>
Date: Sat, 4 Jul 2020 02:09:12 +0200
Subject: Persist event order

---
 src/Cache.cpp | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

(limited to 'src/Cache.cpp')

diff --git a/src/Cache.cpp b/src/Cache.cpp
index 27f4e694..2824960b 100644
--- a/src/Cache.cpp
+++ b/src/Cache.cpp
@@ -1957,9 +1957,21 @@ Cache::saveTimelineMessages(lmdb::txn &txn,
         auto db = getMessagesDb(txn, room_id);
         auto eventsDb = getEventsDb(txn, room_id);
 
+        auto orderDb = getEventOrderDb(txn, room_id);
+        if (res.limited)
+                lmdb::dbi_drop(txn, orderDb, false);
+
         using namespace mtx::events;
         using namespace mtx::events::state;
 
+        lmdb::val indexVal, val;
+        int64_t index = 0;
+        auto cursor = lmdb::cursor::open(txn, orderDb);
+        if (cursor.get(indexVal, val, MDB_LAST)) {
+                index = *indexVal.data<int64_t>();
+        }
+
+        bool first = true;
         for (const auto &e : res.events) {
                 auto event = mtx::accessors::serialize_event(e);
                 if (auto redaction =
@@ -1982,6 +1994,14 @@ Cache::saveTimelineMessages(lmdb::txn &txn,
                                       eventsDb,
                                       lmdb::val(event["event_id"].get<std::string>()),
                                       lmdb::val(event.dump()));
+
+                        ++index;
+
+                        lmdb::cursor_put(cursor.handle(),
+                                         lmdb::val(&index, sizeof(index)),
+                                         lmdb::val(first ? res.prev_batch : ""),
+                                         MDB_APPEND);
+                        first = false;
                 }
         }
 }
-- 
cgit 1.5.1