dissolve key_value/*

Signed-off-by: Jason Volk <jason@zemos.net>

src/service/rooms/lazy_loading/data.rs

@@ -1,6 +1,6 @@
 use ruma::{DeviceId, RoomId, UserId};
 
-use crate::Result;
+use crate::{KeyValueDatabase, Result};
 
 pub trait Data: Send + Sync {
 	fn lazy_load_was_sent_before(
@@ -14,3 +14,53 @@ pub trait Data: Send + Sync {
 
 	fn lazy_load_reset(&self, user_id: &UserId, device_id: &DeviceId, room_id: &RoomId) -> Result<()>;
 }
+
+impl Data for KeyValueDatabase {
+	fn lazy_load_was_sent_before(
+		&self, user_id: &UserId, device_id: &DeviceId, room_id: &RoomId, ll_user: &UserId,
+	) -> Result<bool> {
+		let mut key = user_id.as_bytes().to_vec();
+		key.push(0xFF);
+		key.extend_from_slice(device_id.as_bytes());
+		key.push(0xFF);
+		key.extend_from_slice(room_id.as_bytes());
+		key.push(0xFF);
+		key.extend_from_slice(ll_user.as_bytes());
+		Ok(self.lazyloadedids.get(&key)?.is_some())
+	}
+
+	fn lazy_load_confirm_delivery(
+		&self, user_id: &UserId, device_id: &DeviceId, room_id: &RoomId,
+		confirmed_user_ids: &mut dyn Iterator<Item = &UserId>,
+	) -> Result<()> {
+		let mut prefix = user_id.as_bytes().to_vec();
+		prefix.push(0xFF);
+		prefix.extend_from_slice(device_id.as_bytes());
+		prefix.push(0xFF);
+		prefix.extend_from_slice(room_id.as_bytes());
+		prefix.push(0xFF);
+
+		for ll_id in confirmed_user_ids {
+			let mut key = prefix.clone();
+			key.extend_from_slice(ll_id.as_bytes());
+			self.lazyloadedids.insert(&key, &[])?;
+		}
+
+		Ok(())
+	}
+
+	fn lazy_load_reset(&self, user_id: &UserId, device_id: &DeviceId, room_id: &RoomId) -> Result<()> {
+		let mut prefix = user_id.as_bytes().to_vec();
+		prefix.push(0xFF);
+		prefix.extend_from_slice(device_id.as_bytes());
+		prefix.push(0xFF);
+		prefix.extend_from_slice(room_id.as_bytes());
+		prefix.push(0xFF);
+
+		for (key, _) in self.lazyloadedids.scan_prefix(prefix) {
+			self.lazyloadedids.remove(&key)?;
+		}
+
+		Ok(())
+	}
+}

src/service/rooms/lazy_loading/mod.rs

@@ -4,7 +4,7 @@ use std::{
 	sync::Arc,
 };
 
-pub use data::Data;
+use data::Data;
 use ruma::{DeviceId, OwnedDeviceId, OwnedRoomId, OwnedUserId, RoomId, UserId};
 use tokio::sync::Mutex;