blythe/conduwuit
blythe/conduwuit
1
0
Fork 0
mirror of https://github.com/girlbossceo/conduwuit.git synced 2025-03-14 18:55:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

optimize statekey-from-short loopsite

Browse source 
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk 2025-01-14 19:17:45 +00:00
parent 3dae02b886
commit ac944496c1
1 changed files with 13 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
src/service/rooms/event_handler/resolve_state.rs
View file

@ -6,10 +6,10 @@ use std::{
use conduwuit::{
debug, err, implement,
utils::stream::{automatic_width, IterStream, WidebandExt},
utils::stream::{automatic_width, IterStream, ReadyExt, WidebandExt},
Result,
};
use futures::{FutureExt, StreamExt, TryFutureExt};
use futures::{FutureExt, StreamExt};
use ruma::{
state_res::{self, StateMap},
OwnedEventId, RoomId, RoomVersionId,
@ -59,17 +59,18 @@ pub async fn resolve_state(
let fork_states: Vec<StateMap<OwnedEventId>> = fork_states
.into_iter()
.stream()
.wide_then(|fork_state| {
fork_state
.into_iter()
.stream()
.wide_filter_map(|(k, id)| {
self.services
.short
.get_statekey_from_short(k)
.map_ok_or_else(|_| None, move |(ty, st_key)| Some(((ty, st_key), id)))
})
.wide_then(|fork_state| async move {
let shortstatekeys = fork_state.keys().copied().stream();
let event_ids = fork_state.values().cloned().stream().boxed();
self.services
.short
.multi_get_statekey_from_short(shortstatekeys)
.zip(event_ids)
.ready_filter_map(|(ty_sk, id)| Some((ty_sk.ok()?, id)))
.collect()
.await
})
.collect()
.await;