perf: use stdout buf writer

1 files changed, 32 insertions, 18 deletions

diff --git a/fctdrive/src/main.rs b/fctdrive/src/main.rs
index 0f7c83c..2035680 100644
--- a/fctdrive/src/main.rs
+++ b/fctdrive/src/main.rs
@@ -849,11 +849,42 @@ fn cmd_rename(args: RenameArgs) {
     write_log_file(&ops);
 }
 
+type FsNodeWriter<'a> = std::io::BufWriter<std::io::StdoutLock<'a>>;
+
 fn cmd_view(args: ViewArgs) {
+    let stdout = std::io::stdout().lock();
+    let mut writer = BufWriter::new(stdout);
     let mut fs = Fs::default();
     let ops = read_log_file();
     ops.iter().for_each(|op| apply(&mut fs, op).unwrap());
-    write_node(&fs, fs.root, Default::default());
+    write_node(&mut writer, &fs, fs.root, Default::default());
+    writer.flush().unwrap();
+}
+
+fn write_node(writer: &mut FsNodeWriter, fs: &Fs, node_id: FsNodeId, path: DrivePath) {
+    let node = &fs.nodes[node_id];
+    match node.kind {
+        FsNodeKind::File => {
+            write!(
+                writer,
+                "{}\tfile\t{}\t{}\t{}",
+                path, node.lastmod, node.blob, node.author
+            )
+            .unwrap();
+        }
+        FsNodeKind::Directory => {
+            write!(
+                writer,
+                "{}\tdir\t{}\t-\t{}",
+                path, node.lastmod, node.author
+            )
+            .unwrap();
+            for (child_comp, child_id) in node.children.iter() {
+                let child_path = path.push(child_comp.clone());
+                write_node(writer, fs, *child_id, child_path);
+            }
+        }
+    }
 }
 
 #[derive(Debug, Clone)]
@@ -954,23 +985,6 @@ fn collect_all_file_paths(root: &Path) -> Vec<PathBuf> {
     files
 }
 
-fn write_node(fs: &Fs, node_id: FsNodeId, path: DrivePath) {
-    let node = &fs.nodes[node_id];
-    match node.kind {
-        FsNodeKind::File => println!(
-            "{}\tfile\t{}\t{}\t{}",
-            path, node.lastmod, node.blob, node.author
-        ),
-        FsNodeKind::Directory => {
-            println!("{}\tdir\t{}\t-\t{}", path, node.lastmod, node.author);
-            for (child_comp, child_id) in node.children.iter() {
-                let child_path = path.push(child_comp.clone());
-                write_node(fs, *child_id, child_path);
-            }
-        }
-    }
-}
-
 fn read_log_file() -> Vec<Operation> {
     let mut operations = Vec::default();
     if std::fs::exists("log.txt").unwrap() {