From de4f8e6f63cb4bc93b881b9da3d5c0529d424b8d Mon Sep 17 00:00:00 2001
From: Michael Snoyman <michael@snoyman.com>
Date: Mon, 5 Jan 2015 09:13:18 +0200
Subject: [PATCH] Start showing docs as soon as they are unpacked

---
 Data/Unpacking.hs | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/Data/Unpacking.hs b/Data/Unpacking.hs
index 8b7add2..4c2cc3b 100644
--- a/Data/Unpacking.hs
+++ b/Data/Unpacking.hs
@@ -111,7 +111,11 @@ unpackWorker dirs runDB store messageVar workChan = do
                 , msg
                 ]
         say "Beginning of processing"
-        eres <- tryAny $ unpacker dirs runDB store say ent
+
+        -- As soon as the raw unpack is complete, start serving docs
+        let onRawComplete = atomically $ writeTVar resVar USReady
+
+        eres <- tryAny $ unpacker dirs runDB store say onRawComplete ent
         atomically $ writeTVar resVar $ case eres of
             Left e -> USFailed $ tshow e
             Right () -> USReady
@@ -149,9 +153,10 @@ unpacker
     -> (forall a m. (MonadIO m, MonadBaseControl IO m) => SqlPersistT m a -> m a)
     -> BlobStore StoreKey
     -> (Text -> IO ())
+    -> IO () -- ^ onRawComplete
     -> Entity Stackage
     -> IO ()
-unpacker dirs runDB store say (Entity sid stackage@Stackage {..}) = do
+unpacker dirs runDB store say onRawComplete (Entity sid stackage@Stackage {..}) = do
     say "Removing old directories, if they exist"
     removeTreeIfExists $ dirRawIdent dirs stackageIdent
     removeTreeIfExists $ dirGzIdent dirs stackageIdent
@@ -159,6 +164,7 @@ unpacker dirs runDB store say (Entity sid stackage@Stackage {..}) = do
 
     let destdir = dirRawIdent dirs stackageIdent
     unpackRawDocsTo store stackageIdent say destdir
+    onRawComplete
 
     createTree $ dirHoogleIdent dirs stackageIdent