use BufReader/BufWriter

Author: besaleli

encoderfile-runtime/src/main.rs

@@ -2,7 +2,7 @@ use clap::Parser;
 use parking_lot::Mutex;
 use std::{
     fs::File,
-    io::{Read, Seek},
+    io::{BufReader, Read, Seek},
     sync::Arc,
 };
 
@@ -22,7 +22,8 @@ use encoderfile_core::{
 async fn main() -> Result<()> {
     // open current executable
     let path = std::env::current_exe()?;
-    let mut file = File::open(path)?;
+    let file = File::open(path)?;
+    let mut file = BufReader::new(file);
     // load encoderfile
     let mut loader = load_assets(&mut file)?;
 
@@ -73,7 +74,7 @@ async fn entrypoint<'a, R: Read + Seek>(loader: &mut EncoderfileLoader<'a, R>) -
     }
 }
 
-fn load_assets<'a>(file: &'a mut File) -> Result<EncoderfileLoader<'a, std::fs::File>> {
+fn load_assets<'a, R: Read + Seek>(file: &'a mut R) -> Result<EncoderfileLoader<'a, R>> {
     let encoderfile = EncoderfileCodec::read(file)?;
     let loader = EncoderfileLoader::new(encoderfile, file);
 

encoderfile/src/cli.rs

@@ -7,7 +7,12 @@ use encoderfile_core::{
     },
     generated::manifest::Backend,
 };
-use std::{borrow::Cow, fs::File, io::Seek, path::PathBuf};
+use std::{
+    borrow::Cow,
+    fs::File,
+    io::{BufWriter, Seek, Write},
+    path::PathBuf,
+};
 
 use clap_derive::{Args, Parser, Subcommand};
 
@@ -120,7 +125,8 @@ impl BuildArgs {
         };
 
         // initialize final binary
-        let mut out = File::create(config.encoderfile.output_path())?;
+        let out = File::create(config.encoderfile.output_path())?;
+        let mut out = BufWriter::new(out);
         let mut base = File::open(base_path)?;
 
         // copy base binary to out
@@ -145,6 +151,8 @@ impl BuildArgs {
             &mut out,
         )?;
 
+        out.flush()?;
+
         Ok(())
     }
 }