magmaus3/continuwuity
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

optimize for pdu_exists; remove a yield thing

Browse source 
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk 2024-10-22 06:37:09 +00:00 • committed by strawberry
parent ca57dc7928
commit b8260e0104
1 changed files with 2 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/service/rooms/event_handler/mod.rs
View file

@ -359,7 +359,7 @@ impl Service {
}; };
// Skip the PDU if it is redacted and we already have it as an outlier event // Skip the PDU if it is redacted and we already have it as an outlier event
if self.services.timeline.get_pdu_json(event_id).await.is_ok() { if self.services.timeline.pdu_exists(event_id).await {
return Err!(Request(InvalidParam("Event was redacted and we already knew about it"))); return Err!(Request(InvalidParam("Event was redacted and we already knew about it")));
} }
@ -1123,7 +1123,6 @@ impl Service {
let mut todo_auth_events = vec![Arc::clone(id)]; let mut todo_auth_events = vec![Arc::clone(id)];
let mut events_in_reverse_order = Vec::with_capacity(todo_auth_events.len()); let mut events_in_reverse_order = Vec::with_capacity(todo_auth_events.len());
let mut events_all = HashSet::with_capacity(todo_auth_events.len()); let mut events_all = HashSet::with_capacity(todo_auth_events.len());
let mut i: u64 = 0;
while let Some(next_id) = todo_auth_events.pop() { while let Some(next_id) = todo_auth_events.pop() {
if let Some((time, tries)) = self if let Some((time, tries)) = self
.services .services
@ -1146,12 +1145,7 @@ impl Service {
continue; continue;
} }
i = i.saturating_add(1); if self.services.timeline.pdu_exists(&next_id).await {
if i % 100 == 0 {
tokio::task::yield_now().await;
}
if self.services.timeline.get_pdu(&next_id).await.is_ok() {
trace!("Found {next_id} in db"); trace!("Found {next_id} in db");
continue; continue;
} }