LemmyNet/lemmy
2
0
Fork 0
mirror of https://github.com/LemmyNet/lemmy.git synced 2024-11-25 22:01:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

chore: simplify collect function

Browse source
This commit is contained in:
phiresky 2024-08-05 12:35:21 +02:00
parent b60e511493
commit 34631a9f13
1 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
crates/api_common/src/request.rs
View file

@ -106,14 +106,13 @@ async fn collect_bytes_until_limit(
) -> Result<Vec<u8>, LemmyError> { ) -> Result<Vec<u8>, LemmyError> {
let mut stream = response.bytes_stream(); let mut stream = response.bytes_stream();
let mut bytes = Vec::with_capacity(requested_bytes); let mut bytes = Vec::with_capacity(requested_bytes);
let mut total_bytes = 0;
while let Some(chunk) = stream.next().await { while let Some(chunk) = stream.next().await {
let chunk = chunk.map_err(LemmyError::from)?; let chunk = chunk.map_err(LemmyError::from)?;
total_bytes += chunk.len();
// we may go over the requested size here but the important part is we don't keep aggregating // we may go over the requested size here but the important part is we don't keep aggregating
// more chunks than needed // more chunks than needed
bytes.extend_from_slice(&chunk); bytes.extend_from_slice(&chunk);
if total_bytes >= requested_bytes { if bytes.len() >= requested_bytes {
bytes.truncate(requested_bytes);
break; break;
} }
} }