From 9b2bea53c6bc87be639964aaca87c09ab16df486 Mon Sep 17 00:00:00 2001
From: tv <tv@shackspace.de>
Date: Thu, 6 Nov 2014 16:10:43 +0100
Subject: Flush log after each request.

---
 src/Main.hs | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

(limited to 'src/Main.hs')

diff --git a/src/Main.hs b/src/Main.hs
index ef7710f..66b4df6 100644
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -56,6 +56,7 @@ import Network.Wai.Handler.Warp (run)
 import Network.Wai.Middleware.RequestLogger (logStdout)
 import System.FilePath
 import System.IO.Error
+import System.IO (hFlush, stdout)
 
 
 -- | The sum of all resources known by the application.
@@ -79,10 +80,21 @@ main =
 start :: Config -> IO ()
 start c = do
     putStrLn $ "run cgserver with " <> show c
+    hFlush stdout
     run (httpPort c)
-        $ logStdout
+        $ logger
         $ \req ->
             resourceHandler (requestResource c req) req
+  where
+    logger =
+        if flushLog c
+            then flush stdout . logStdout
+            else logStdout
+
+    -- This middleware flushes the given handle after each request.
+    flush h app req respond = app req $ \res -> do
+            hFlush h
+            respond res
 
 
 -- | Determine which request is requested.
-- 
cgit v1.2.3