From f73716d36a50a91fb9e072228b3e64937a7c4f65 Mon Sep 17 00:00:00 2001
From: Jonathan Strong <jonathan.strong@gmail.com>
Date: Tue, 17 Mar 2020 02:42:36 -0400
Subject: [PATCH] improved stdout perf

---
 src/commands/search.rs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/commands/search.rs b/src/commands/search.rs
index a338414..fa95e71 100644
--- a/src/commands/search.rs
+++ b/src/commands/search.rs
@@ -4,6 +4,7 @@ use std::str::FromStr;
 use std::convert::From;
 use std::path::Path;
 use std::path::PathBuf;
+use std::io::Write;
 use tantivy;
 use tantivy::query::QueryParser;
 use tantivy::schema::Field;
@@ -47,6 +48,9 @@ fn run_search(directory: &Path, query: &str) -> tantivy::Result<()> {
     let searcher = index.reader()?.searcher();
     let weight = query.weight(&searcher, false)?;
     let schema = index.schema();
+    let stdout = std::io::stdout();
+    let handle = stdout.lock();
+    let mut wtr = std::io::BufWriter::new(handle);
     for segment_reader in searcher.segment_readers() {
         let mut scorer = weight.scorer(segment_reader, 1.0)?;
         let store_reader = segment_reader.get_store_reader();
@@ -54,7 +58,8 @@ fn run_search(directory: &Path, query: &str) -> tantivy::Result<()> {
             let doc_id = scorer.doc();
             let doc = store_reader.get(doc_id)?;
             let named_doc = schema.to_named_doc(&doc);
-            println!("{}", serde_json::to_string(&named_doc).unwrap());
+            serde_json::to_writer(&mut wtr, &named_doc).unwrap();
+            wtr.write_all(b"\n")?;
         }
     }
     Ok(())