1
0
Fork 0
mirror of https://gitlab.com/famedly/conduit.git synced 2025-06-27 16:35:59 +00:00
conduit/src/service/rooms/lazy_loading/mod.rs

92 lines
2.5 KiB
Rust
Raw Normal View History

2022-01-04 14:30:13 +01:00
#[tracing::instrument(skip(self))]
pub 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);
2022-01-20 00:10:39 +01:00
key.extend_from_slice(device_id.as_bytes());
2022-01-04 14:30:13 +01:00
key.push(0xff);
2022-01-20 00:10:39 +01:00
key.extend_from_slice(room_id.as_bytes());
2022-01-04 14:30:13 +01:00
key.push(0xff);
2022-01-20 00:10:39 +01:00
key.extend_from_slice(ll_user.as_bytes());
2022-01-04 14:30:13 +01:00
Ok(self.lazyloadedids.get(&key)?.is_some())
}
#[tracing::instrument(skip(self))]
pub fn lazy_load_mark_sent(
&self,
user_id: &UserId,
device_id: &DeviceId,
room_id: &RoomId,
lazy_load: HashSet<Box<UserId>>,
2022-01-04 14:30:13 +01:00
count: u64,
) {
self.lazy_load_waiting.lock().unwrap().insert(
(
user_id.to_owned(),
device_id.to_owned(),
room_id.to_owned(),
count,
),
lazy_load,
);
}
#[tracing::instrument(skip(self))]
pub fn lazy_load_confirm_delivery(
&self,
user_id: &UserId,
device_id: &DeviceId,
room_id: &RoomId,
since: u64,
) -> Result<()> {
if let Some(user_ids) = self.lazy_load_waiting.lock().unwrap().remove(&(
user_id.to_owned(),
device_id.to_owned(),
room_id.to_owned(),
since,
)) {
let mut prefix = user_id.as_bytes().to_vec();
prefix.push(0xff);
2022-01-20 00:10:39 +01:00
prefix.extend_from_slice(device_id.as_bytes());
2022-01-04 14:30:13 +01:00
prefix.push(0xff);
2022-01-20 00:10:39 +01:00
prefix.extend_from_slice(room_id.as_bytes());
2022-01-04 14:30:13 +01:00
prefix.push(0xff);
for ll_id in user_ids {
let mut key = prefix.clone();
2022-01-20 00:10:39 +01:00
key.extend_from_slice(ll_id.as_bytes());
2022-01-04 14:30:13 +01:00
self.lazyloadedids.insert(&key, &[])?;
}
}
Ok(())
}
#[tracing::instrument(skip(self))]
pub fn lazy_load_reset(
&self,
2022-01-20 00:10:39 +01:00
user_id: &UserId,
device_id: &DeviceId,
room_id: &RoomId,
2022-01-04 14:30:13 +01:00
) -> Result<()> {
let mut prefix = user_id.as_bytes().to_vec();
prefix.push(0xff);
2022-01-20 00:10:39 +01:00
prefix.extend_from_slice(device_id.as_bytes());
2022-01-04 14:30:13 +01:00
prefix.push(0xff);
2022-01-20 00:10:39 +01:00
prefix.extend_from_slice(room_id.as_bytes());
2022-01-04 14:30:13 +01:00
prefix.push(0xff);
for (key, _) in self.lazyloadedids.scan_prefix(prefix) {
self.lazyloadedids.remove(&key)?;
}
Ok(())
}
2021-11-01 08:57:27 +00:00