studio/tuwunel
2
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Add missing reverse query for shortroomid => room_id.

Browse Source 
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk
2025-09-29 04:19:00 +00:00
parent 1058fbe9a7
commit a803b84b27
1 changed files with 20 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/service/rooms/short/mod.rs
View File

@@ -1,10 +1,15 @@
use std::{borrow::Borrow, fmt::Debug, mem::size_of_val, sync::Arc};
use futures::{Stream, StreamExt};
use ruma::{EventId, RoomId, events::StateEventType};
use futures::{FutureExt, Stream, StreamExt};
use ruma::{EventId, OwnedRoomId, RoomId, events::StateEventType};
use serde::Deserialize;
pub use tuwunel_core::matrix::pdu::{ShortEventId, ShortId, ShortRoomId, ShortStateKey};
use tuwunel_core::{Err, Result, err, implement, matrix::StateKey, utils, utils::IterStream};
use tuwunel_core::{
Err, Result, err, implement,
matrix::StateKey,
utils,
utils::{IterStream, stream::ReadyExt},
};
use tuwunel_database::{Deserialized, Get, Map, Qry};
pub struct Service {
@@ -238,6 +243,18 @@ pub async fn get_shortroomid(&self, room_id: &RoomId) -> Result<ShortRoomId> {
.deserialized()
}
#[implement(Service)]
pub async fn get_roomid_from_short(&self, shortroomid_: ShortRoomId) -> Result<OwnedRoomId> {
self.db
.roomid_shortroomid
.stream()
.ready_filter_map(Result::ok)
.ready_find(|&(_, shortroomid)| shortroomid == shortroomid_)
.map(|found| found.map(|(room_id, _): (&RoomId, ShortRoomId)| room_id.to_owned()))
.await
.ok_or_else(|| err!(Database("Failed to find RoomId from {shortroomid_:?}")))
}
#[implement(Service)]
pub async fn get_or_create_shortroomid(&self, room_id: &RoomId) -> ShortRoomId {
self.db