Flush data when download ends

1 files changed, 8 insertions, 7 deletions

diff --git a/src/libstore/filetransfer.cc b/src/libstore/filetransfer.cc
index 2818043e8..a9a5d1389 100644
--- a/src/libstore/filetransfer.cc
+++ b/src/libstore/filetransfer.cc
@@ -828,17 +828,18 @@ void FileTransfer::download(FileTransferRequest && request, Sink & sink)
         {
             auto state(_state->lock());
 
-            if (state->quit) {
-                if (state->exc) std::rethrow_exception(state->exc);
-                return;
-            }
+            if (state->data.empty()) {
+                if (state->quit) {
+                    if (state->exc) std::rethrow_exception(state->exc);
+                    return;
+                }
 
-            state.wait(state->avail);
+                state.wait(state->avail);
 
-            if (state->data.empty()) continue;
+                if (state->data.empty()) continue;
+            }
 
             chunk = std::move(state->data);
-
             state->request.notify_one();
         }