so close!

but I forgot to ensure the output stream gets closed. :(

1 files changed, 7 insertions, 5 deletions

diff --git a/src/loader/loader.nim b/src/loader/loader.nim
index 712db932..d859ae34 100644
--- a/src/loader/loader.nim
+++ b/src/loader/loader.nim
@@ -210,14 +210,11 @@ proc loadStream(ctx: LoaderContext, handle: LoaderHandle, request: Request,
     if S_ISREG(stats.st_mode):
       # stdin is a regular file, so we can't select on it.
       let originalUrl = if handle.cached: originalUrl else: nil
-      let output = handle.output
-      output.ostream.setBlocking(false)
+      handle.output.ostream.setBlocking(false)
       ctx.addFd0(handle, originalUrl)
-      var nn = 0
       while true:
         let buffer = newLoaderBuffer()
         let n = ps.recvData(buffer)
-        nn += n
         if n == 0:
           break
         var unregWrite: seq[OutputHandle] = @[]
@@ -238,7 +235,12 @@ proc loadStream(ctx: LoaderContext, handle: LoaderHandle, request: Request,
         if n < buffer.cap:
           break
       for output in handle.outputs:
-        output.parent = nil
+        if output.registered:
+          output.parent = nil
+          output.istreamAtEnd = true
+        else:
+          output.ostream.close()
+          output.ostream = nil
       handle.outputs.setLen(0)
       handle.istream.close()
       handle.istream = nil