From 1442c64420345b7e6dff7e19cb628348d3d96c6a Mon Sep 17 00:00:00 2001
From: Nyaaori <+@nyaaori.cat>
Date: Mon, 15 Aug 2022 18:46:50 +0200
Subject: [PATCH] refactor: restore src/service/rooms/state/data.rs

---
 src/service/rooms/state/data.rs | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)
 create mode 100644 src/service/rooms/state/data.rs

diff --git a/src/service/rooms/state/data.rs b/src/service/rooms/state/data.rs
new file mode 100644
index 00000000..4b42ca8e
--- /dev/null
+++ b/src/service/rooms/state/data.rs
@@ -0,0 +1,16 @@
+pub trait Data {
+    fn get_room_shortstatehash(room_id: &RoomId);
+}
+
+    /// Returns the last state hash key added to the db for the given room.
+    #[tracing::instrument(skip(self))]
+    pub fn current_shortstatehash(&self, room_id: &RoomId) -> Result<Option<u64>> {
+        self.roomid_shortstatehash
+            .get(room_id.as_bytes())?
+            .map_or(Ok(None), |bytes| {
+                Ok(Some(utils::u64_from_bytes(&bytes).map_err(|_| {
+                    Error::bad_database("Invalid shortstatehash in roomid_shortstatehash")
+                })?))
+            })
+    }
+