From 610871e5fcf7079daf131b59f9c7928f4695e197 Mon Sep 17 00:00:00 2001
From: Hatter Jiang <jht5945@gmail.com>
Date: Sun, 21 Jul 2019 13:24:18 +0800
Subject: [PATCH] add fn flush_stdout()

---
 src/lib.rs | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/lib.rs b/src/lib.rs
index 520f918..d9bd48a 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -52,9 +52,16 @@ pub fn print_message(mt: MessageType, message: &str) {
     }
 }
 
+pub fn flush_stdout() {
+    match io::stdout().flush() {
+        Err(err) => print_message(MessageType::ERROR, &format!("Flush stdout failed: {}", err)),
+        Ok(_) => (),
+    }
+}
+
 pub fn print_lastline(line: &str) {
     print!("\x1b[100D{}\x1b[K", line);
-    io::stdout().flush().unwrap();
+    flush_stdout();
 }
 
 pub fn get_display_size(size: i64) -> String {