fix: xz decompression issues

bazel-contrib · Nov 25, 2024 · d7616f1 · d7616f1
1 parent 0b382f4
commit d7616f1
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/src/domain/release-archive.ts b/src/domain/release-archive.ts
@@ -99,6 +99,11 @@ export class ReleaseArchive {
         cwd: extractDir
       });
       await decompressXz(reader, writer);
+      await new Promise(resolve => {
+        writer.on('finish', resolve);
+        writer.end();
+      });
+
       return;
     }
 

diff --git a/src/infrastructure/xzdec/xzdec.ts b/src/infrastructure/xzdec/xzdec.ts
@@ -126,7 +126,14 @@ export async function decompress(r: stream.Readable, w: stream.Writable) {
           }
           const outputLen = peekU32(mem, outputLenPtr);
           if (outputLen > 0) {
-            w.write(Buffer.from(mem.buffer, outputPtr, outputLen));
+            await new Promise((resolve) => {
+              if (!w.write(Buffer.from(mem.buffer, outputPtr, outputLen))) {
+                w.once('drain', resolve)
+              }
+              else {
+                resolve(null);
+              }
+            });
           }
         }
       }