Skip to content

Commit 8292d47

Browse files
committedApr 13, 2025
Use configured buffer size for all buffered readers and writers
1 parent ceb5f7b commit 8292d47

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed
 

‎src/task_cleanup.rs

+8-5
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ use crate::{
2424
};
2525

2626
async fn body_to_file(body: &mut Incoming, file: tokio::fs::File) -> Result<(), ProxyCacheError> {
27-
let mut writer = BufWriter::new(file);
27+
let mut writer = BufWriter::with_capacity(global_config().buffer_size, file);
2828

2929
while let Some(next) = body.frame().await {
3030
let frame = next?;
@@ -69,7 +69,8 @@ enum PackageFormat {
6969
}
7070

7171
impl PackageFormat {
72-
fn extension(&self) -> &'static str {
72+
#[must_use]
73+
const fn extension(&self) -> &'static str {
7374
match self {
7475
Self::Raw => "",
7576
Self::Gz => ".gz",
@@ -89,7 +90,9 @@ impl PackageFormat {
8990
err
9091
})?;
9192

92-
let file_reader = tokio::io::BufReader::new(file);
93+
let buffer_size = global_config().buffer_size;
94+
95+
let file_reader = tokio::io::BufReader::with_capacity(buffer_size, file);
9396

9497
macro_rules! process_lines {
9598
($e: expr) => {
@@ -122,14 +125,14 @@ impl PackageFormat {
122125
Self::Gz => {
123126
let decoder = async_compression::tokio::bufread::GzipDecoder::new(file_reader);
124127

125-
let reader = tokio::io::BufReader::new(decoder);
128+
let reader = tokio::io::BufReader::with_capacity(buffer_size, decoder);
126129

127130
process_lines!(reader.lines());
128131
}
129132
Self::Xz => {
130133
let decoder = async_compression::tokio::bufread::XzDecoder::new(file_reader);
131134

132-
let reader = tokio::io::BufReader::new(decoder);
135+
let reader = tokio::io::BufReader::with_capacity(buffer_size, decoder);
133136

134137
process_lines!(reader.lines());
135138
}

0 commit comments

Comments
 (0)
Failed to load comments.