Rebox future segmentations.

Signed-off-by: Jason Volk <jason@zemos.net>

src/service/rooms/auth_chain/mod.rs

@@ -7,7 +7,7 @@ use std::{
 	time::Instant,
 };
 
-use futures::{FutureExt, Stream, StreamExt, TryFutureExt, TryStreamExt};
+use futures::{FutureExt, Stream, StreamExt, TryFutureExt, TryStreamExt, pin_mut};
 use ruma::{EventId, OwnedEventId, RoomId};
 use tuwunel_core::{
 	Err, Result, at, debug, debug_error, implement, trace,
@@ -80,13 +80,13 @@ where
 	const BUCKET: Bucket<'_> = BTreeSet::new();
 
 	let started = Instant::now();
-	let mut starting_ids = self
+	let starting_ids = self
 		.services
 		.short
 		.multi_get_or_create_shorteventid(starting_events.clone())
-		.zip(starting_events.clone().stream())
-		.boxed();
+		.zip(starting_events.clone().stream());
 
+	pin_mut!(starting_ids);
 	let mut buckets = [BUCKET; NUM_BUCKETS];
 	while let Some((short, starting_event)) = starting_ids.next().await {
 		let bucket: usize = short.try_into()?;

src/service/rooms/event_handler/handle_incoming_pdu.rs

@@ -112,6 +112,7 @@ pub async fn handle_incoming_pdu<'a>(
 
 	let (incoming_pdu, val) = self
 		.handle_outlier_pdu(origin, create_event, event_id, room_id, value, false)
+		.boxed()
 		.await?;
 
 	// 8. if not timeline event: stop

src/service/rooms/event_handler/upgrade_outlier_pdu.rs

@@ -61,6 +61,7 @@ where
 			.await?
 	} else {
 		self.state_at_incoming_resolved(&incoming_pdu, room_id, &room_version_id)
+			.boxed()
 			.await?
 	};
 
@@ -215,6 +216,7 @@ where
 
 		let new_room_state = self
 			.resolve_state(room_id, &room_version_id, state_after)
+			.boxed()
 			.await?;
 
 		// Set the new room state to the resolved state

src/service/rooms/timeline/backfill.rs

@@ -118,14 +118,11 @@ pub async fn backfill_if_required(&self, room_id: &RoomId, from: PduCount) -> Re
 				},
 			)
 			.await;
+
 		match response {
 			| Ok(response) => {
 				for pdu in response.pdus {
-					if let Err(e) = self
-						.backfill_pdu(backfill_server, pdu)
-						.boxed()
-						.await
-					{
+					if let Err(e) = self.backfill_pdu(backfill_server, pdu).await {
 						debug_warn!("Failed to add backfilled pdu in room {room_id}: {e}");
 					}
 				}