studio/tuwunel
2
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Improve mutable state in sender closure captures.

Browse Source 
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk
2025-07-08 13:16:40 +00:00
parent c8d35cca57
commit aac49b09c7
1 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/service/sending/sender.rs
View File

@@ -11,8 +11,8 @@ use std::{
use base64::{Engine as _, engine::general_purpose::URL_SAFE_NO_PAD};
use futures::{
FutureExt, StreamExt,
future::{BoxFuture, OptionFuture},
join, pin_mut,
future::{BoxFuture, OptionFuture, join3},
pin_mut,
stream::FuturesUnordered,
};
use ruma::{
@@ -400,7 +400,8 @@ impl Service {
.then(|| self.select_edus_presence(server_name, batch, &max_edu_count))
.into();
let (device_changes, receipts, presence) = join!(device_changes, receipts, presence);
let (device_changes, receipts, presence) =
join3(device_changes, receipts, presence).await;
let mut events = device_changes;
events.extend(presence.into_iter().flatten());
@@ -486,7 +487,7 @@ impl Service {
since: (u64, u64),
max_edu_count: &AtomicU64,
) -> Option<EduBuf> {
let mut num = 0;
let num = AtomicUsize::new(0);
let receipts: BTreeMap<OwnedRoomId, ReceiptMap> = self
.services
.state_cache
@@ -494,7 +495,7 @@ impl Service {
.map(ToOwned::to_owned)
.broad_filter_map(async |room_id| {
let receipt_map = self
.select_edus_receipts_room(&room_id, since, max_edu_count, &mut num)
.select_edus_receipts_room(&room_id, since, max_edu_count, &num)
.await;
receipt_map
@@ -530,7 +531,7 @@ impl Service {
room_id: &RoomId,
since: (u64, u64),
max_edu_count: &AtomicU64,
num: &mut usize,
num: &AtomicUsize,
) -> ReceiptMap {
let receipts = self
.services
@@ -581,8 +582,8 @@ impl Service {
.insert(user_id.to_owned(), receipt_data)
.is_none()
{
*num = num.saturating_add(1);
if *num >= SELECT_RECEIPT_LIMIT {
let num = num.fetch_add(1, Ordering::Relaxed);
if num >= SELECT_RECEIPT_LIMIT {
break;
}
}