fedi/lemmy
1
0
Fork 0
mirror of https://github.com/LemmyNet/lemmy.git synced 2024-11-09 09:52:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

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> {
let mut stream = response.bytes_stream();
let mut bytes = Vec::with_capacity(requested_bytes);
let mut total_bytes = 0;
while let Some(chunk) = stream.next().await {
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
// more chunks than needed
bytes.extend_from_slice(&chunk);
if total_bytes >= requested_bytes {
if bytes.len() >= requested_bytes {
bytes.truncate(requested_bytes);
break;
}
}