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

parallelize current and incoming fork-state fetch

Browse source 
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk 2025-01-20 07:38:32 +00:00
parent 388730d6dd
commit ea25dc04b2
1 changed files with 18 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
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, ReadyExt, WidebandExt},
utils::stream::{automatic_width, IterStream, ReadyExt, TryWidebandExt, WidebandExt},
Result,
};
use futures::{FutureExt, StreamExt};
use futures::{FutureExt, StreamExt, TryStreamExt};
use ruma::{
state_res::{self, StateMap},
OwnedEventId, RoomId, RoomVersionId,
@ -40,20 +40,24 @@ pub async fn resolve_state(
.await?;
let fork_states = [current_state_ids, incoming_state];
let mut auth_chain_sets = Vec::with_capacity(fork_states.len());
for state in &fork_states {
let starting_events = state.values().map(Borrow::borrow);
let auth_chain_sets: Vec<HashSet<OwnedEventId>> = fork_states
.iter()
.try_stream()
.wide_and_then(|state| async move {
let starting_events = state.values().map(Borrow::borrow);
let auth_chain: HashSet<OwnedEventId> = self
.services
.auth_chain
.get_event_ids(room_id, starting_events)
.await?
.into_iter()
.collect();
let auth_chain = self
.services
.auth_chain
.get_event_ids(room_id, starting_events)
.await?
.into_iter()
.collect();
auth_chain_sets.push(auth_chain);
}
Ok(auth_chain)
})
.try_collect()
.await?;
debug!("Loading fork states");
let fork_states: Vec<StateMap<OwnedEventId>> = fork_states