employ bufreader in zstd binary

KillingSpark · Oct 23, 2024 · c497e42 · c497e42
1 parent ade1b3d
commit c497e42
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/bin/zstd.rs b/src/bin/zstd.rs
@@ -1,5 +1,6 @@
 extern crate ruzstd;
 use std::fs::File;
+use std::io::BufReader;
 use std::io::Read;
 use std::io::Seek;
 use std::io::SeekFrom;
@@ -133,12 +134,13 @@ fn main() {
 
     if flags.is_empty() {
         for path in file_paths {
-            let mut file = std::fs::File::open(&path).unwrap();
+            let file = std::fs::File::open(&path).unwrap();
+            let input_len = file.metadata().unwrap().len() as usize;
+            let file = BufReader::new(file);
             let mut output = Vec::new();
             let mut encoder =
-                FrameCompressor::new(&mut file, &mut output, CompressionLevel::Fastest);
+                FrameCompressor::new(file, &mut output, CompressionLevel::Fastest);
             encoder.compress();
-            let input_len = file.metadata().unwrap().len() as usize;
             println!(
                 "Compressed {path:} from {} to {} ({}%)",
                 input_len,