hack out bufwriter capacity

Author: rebasedming

src/directory/directory.rs

@@ -296,6 +296,10 @@ pub trait Directory: DirectoryClone + fmt::Debug + Send + Sync + 'static {
     fn log(&self, message: &str) {
         log!(Level::Info, "{message}");
     }
+
+    fn bufwriter_capacity(&self) -> usize {
+        8192
+    }
 }
 
 /// DirectoryClone

src/directory/managed_directory.rs

@@ -313,13 +313,16 @@ impl Directory for ManagedDirectory {
     fn open_write(&self, path: &Path) -> result::Result<WritePtr, OpenWriteError> {
         self.register_file_as_managed(path)
             .map_err(|io_error| OpenWriteError::wrap_io_error(io_error, path.to_path_buf()))?;
-        Ok(io::BufWriter::new(Box::new(FooterProxy::new(
-            self.directory
-                .open_write(path)?
-                .into_inner()
-                .map_err(|_| ())
-                .expect("buffer should be empty"),
-        ))))
+        Ok(io::BufWriter::with_capacity(
+            self.bufwriter_capacity(),
+            Box::new(FooterProxy::new(
+                self.directory
+                    .open_write(path)?
+                    .into_inner()
+                    .map_err(|_| ())
+                    .expect("buffer should be empty"),
+            )),
+        ))
     }
 
     fn atomic_write(&self, path: &Path, data: &[u8]) -> io::Result<()> {
@@ -380,6 +383,10 @@ impl Directory for ManagedDirectory {
     fn log(&self, message: &str) {
         self.directory.log(message);
     }
+
+    fn bufwriter_capacity(&self) -> usize {
+        self.directory.bufwriter_capacity()
+    }
 }
 
 impl Clone for ManagedDirectory {

src/directory/mmap_directory.rs

@@ -443,7 +443,10 @@ impl Directory for MmapDirectory {
         // sync_directory() is called.
 
         let writer = SafeFileWriter::new(file);
-        Ok(BufWriter::new(Box::new(writer)))
+        Ok(BufWriter::with_capacity(
+            self.bufwriter_capacity(),
+            Box::new(writer),
+        ))
     }
 
     fn atomic_read(&self, path: &Path) -> Result<Vec<u8>, OpenReadError> {

src/directory/ram_directory.rs

@@ -206,7 +206,10 @@ impl Directory for RamDirectory {
         if exists {
             Err(OpenWriteError::FileAlreadyExists(path_buf))
         } else {
-            Ok(BufWriter::new(Box::new(vec_writer)))
+            Ok(BufWriter::with_capacity(
+                self.bufwriter_capacity(),
+                Box::new(vec_writer),
+            ))
         }
     }