Flush file contents each frame and when closing the file

1 files changed, 5 insertions, 1 deletions

diff --git a/src/devices/stream.rs b/src/devices/stream.rs
index 7f69a7a..532df58 100644
--- a/src/devices/stream.rs
+++ b/src/devices/stream.rs
@@ -25,6 +25,10 @@ impl StreamDevice {
         }
     }
 
+    pub fn flush_local(&mut self) {
+        self.stdout.flush().unwrap();
+    }
+
     pub fn read_queue_len(&self) -> usize {
         self.stdin.buffer().len()
     }
@@ -44,6 +48,6 @@ impl StreamDevice {
 
 impl Drop for StreamDevice {
     fn drop(&mut self) {
-        self.stdout.flush().unwrap();
+        self.flush_local();
     }
 }