From e50f2864ded296d0143f10782f1b60cf41189514 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Timo=20K=C3=B6sters?= <timo@koesters.xyz>
Date: Tue, 23 Mar 2021 22:31:19 +0100
Subject: [PATCH] improvement: save state for send_join pdu

---
 src/client_server/membership.rs | 8 ++++++++
 src/database/rooms.rs           | 1 -
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/client_server/membership.rs b/src/client_server/membership.rs
index 71be6ac7..36bbced7 100644
--- a/src/client_server/membership.rs
+++ b/src/client_server/membership.rs
@@ -650,6 +650,10 @@ async fn join_room_by_id_helper(
             db.rooms.add_pdu_outlier(&pdu)?;
         }
 
+        // We append to state before appending the pdu, so we don't have a moment in time with the
+        // pdu without it's state. This is okay because append_pdu can't fail.
+        let statehashid = db.rooms.append_to_state(&pdu, &db.globals)?;
+
         db.rooms.append_pdu(
             &pdu,
             utils::to_canonical_object(&pdu).expect("Pdu is valid canonical object"),
@@ -658,6 +662,10 @@ async fn join_room_by_id_helper(
             &[pdu.event_id.clone()],
             db,
         )?;
+
+        // We set the room state after inserting the pdu, so that we never have a moment in time
+        // where events in the current room state do not exist
+        db.rooms.set_room_state(&room_id, statehashid)?;
     } else {
         let event = member::MemberEventContent {
             membership: member::MembershipState::Join,
diff --git a/src/database/rooms.rs b/src/database/rooms.rs
index 3bf72d0d..91f468f1 100644
--- a/src/database/rooms.rs
+++ b/src/database/rooms.rs
@@ -1115,7 +1115,6 @@ impl Rooms {
         // pdu without it's state. This is okay because append_pdu can't fail.
         let statehashid = self.append_to_state(&pdu, &db.globals)?;
 
-        // remove the
         self.append_pdu(
             &pdu,
             pdu_json,