TudbuT/conduit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: don't allow more than 50 PDUs in a transaction

Browse source
This commit is contained in:
Timo Kösters 2020-11-03 21:20:35 +01:00
parent 0762196963
commit 16b22bb432
No known key found for this signature in database
GPG key ID: 24DA7517711A2BA4
1 changed files with 11 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
src/database/sending.rs
View file

@ -54,13 +54,14 @@ impl Sending {
))
})
.filter_map(|r| r.ok())
.filter(|pdu| !pdu.is_empty()) // Skip reservation key
.take(50)
// This should not contain more than 50 anyway
{
if !pdu.is_empty() {
current_transactions
.entry(server)
.or_insert_with(Vec::new)
.push(pdu);
}
current_transactions
.entry(server)
.or_insert_with(Vec::new)
.push(pdu);
}
for (server, pdus) in current_transactions {
@ -95,7 +96,9 @@ impl Sending {
.filter_map(|r| r.ok())
.map(|k| {
k.subslice(prefix.len(), k.len() - prefix.len())
}).collect::<Vec<_>>();
})
.take(50)
.collect::<Vec<_>>();
if !new_pdus.is_empty() {
for pdu_id in &new_pdus {
@ -108,6 +111,7 @@ impl Sending {
futures.push(Self::handle_event(server, new_pdus, &globals, &rooms));
} else {
servercurrentpdus.remove(&prefix).unwrap();
// servercurrentpdus with the prefix should be empty now
}
}
Err((_server, _e)) => {