dolysis/conduwuit
3
0
Fork 0
mirror of https://github.com/girlbossceo/conduwuit.git synced 2024-11-30 13:43:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

improvement: more efficient /claim

Browse source
This commit is contained in:
Timo Kösters 2022-10-12 10:42:59 +02:00
parent cd835fc7a8
commit 0290f1f355
No known key found for this signature in database
GPG key ID: 24DA7517711A2BA4
1 changed files with 26 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
src/api/client_server/keys.rs
View file

@ -440,13 +440,16 @@ pub(crate) async fn claim_keys_helper(
let mut failures = BTreeMap::new();
for (server, vec) in get_over_federation {
let mut futures: FuturesUnordered<_> = get_over_federation
.into_iter()
.map(|(server, vec)| async move {
let mut one_time_keys_input_fed = BTreeMap::new();
for (user_id, keys) in vec {
one_time_keys_input_fed.insert(user_id.clone(), keys.clone());
}
// Ignore failures
if let Ok(keys) = services()
(
server,
services()
.sending
.send_federation_request(
server,
@ -454,13 +457,20 @@ pub(crate) async fn claim_keys_helper(
one_time_keys: one_time_keys_input_fed,
},
)
.await
{
.await,
)
}).collect();
while let Some((server, response)) = futures.next().await {
match response {
Ok(keys) => {
one_time_keys.extend(keys.one_time_keys);
} else {
}
Err(_e) => {
failures.insert(server.to_string(), json!({}));
}
}
}
Ok(claim_keys::v3::Response {
failures,