flibusta-apps/books_downloader
1
0
Fork 0
mirror of https://github.com/flibusta-apps/books_downloader.git synced 2025-12-08 09:30:43 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add Content-Length header

Browse Source
This commit is contained in:
Bulat Kurbanov
2023-07-30 01:38:24 +02:00
parent d99b8a439f
commit e7766c9270
7 changed files with 58 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/services/downloader/utils.rs
View File

@@ -6,9 +6,11 @@ use bytes::Buf;
use std::io::{Seek, SeekFrom, Write};
pub async fn response_to_tempfile(res: &mut Response) -> Option<SpooledTempFile> {
pub async fn response_to_tempfile(res: &mut Response) -> Option<(SpooledTempFile, usize)> {
let mut tmp_file = tempfile::spooled_tempfile(5 * 1024 * 1024);
let mut data_size: usize = 0;
{
loop {
let chunk = res.chunk().await;
@@ -23,6 +25,8 @@ pub async fn response_to_tempfile(res: &mut Response) -> Option<SpooledTempFile>
None => break,
};
data_size += data.len();
match tmp_file.write(data.chunk()) {
Ok(_) => (),
Err(_) => return None,
@@ -32,5 +36,5 @@ pub async fn response_to_tempfile(res: &mut Response) -> Option<SpooledTempFile>
tmp_file.seek(SeekFrom::Start(0)).unwrap();
}
Some(tmp_file)
Some((tmp_file, data_size))
}