From 489df8f0dd22043507299d02f934fe0213ca29e6 Mon Sep 17 00:00:00 2001
From: Peter Wu <peter@lekensteyn.nl>
Date: Wed, 7 May 2014 12:23:23 +0200
Subject: Add flush command

---
 src/io/DataWriter.java | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

(limited to 'src/io/DataWriter.java')

diff --git a/src/io/DataWriter.java b/src/io/DataWriter.java
index ec7a30e..4f4c022 100644
--- a/src/io/DataWriter.java
+++ b/src/io/DataWriter.java
@@ -5,6 +5,7 @@ import java.io.Closeable;
 import java.io.FileInputStream;
 import java.io.FileNotFoundException;
 import java.io.FileOutputStream;
+import java.io.Flushable;
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.InputStreamReader;
@@ -24,7 +25,7 @@ import provider.ResultListener;
  *
  * @author Maurice Laveaux
  */
-public class DataWriter implements ResultListener, Closeable {
+public class DataWriter implements ResultListener, Closeable, Flushable {
 
     /**
      * The writer for the tweet stream.
@@ -148,6 +149,11 @@ public class DataWriter implements ResultListener, Closeable {
         return Logger.getLogger(getClass().getName());
     }
 
+    @Override
+    public void flush() throws IOException {
+        m_tweet.flush();
+    }
+
     abstract class Store {
 
         private final String filename;
@@ -204,6 +210,12 @@ public class DataWriter implements ResultListener, Closeable {
          * @throws IOException if no input file can be retrieved.
          */
         abstract public InputStream getInputStream() throws IOException;
+
+        private void flush() throws IOException {
+            if (os != null) {
+                os.flush();
+            }
+        }
     }
 
     class SimpleFileStore extends Store {
-- 
cgit v1.2.1