Mirrors/rust-analyzer
2
0
Fork 0
mirror of https://github.com/rust-lang/rust-analyzer synced 2025-01-13 13:48:50 +00:00
Code Issues Projects Releases Packages Wiki Activity

vscode: switched to stream.pipeline with .on(close) workaround

Browse source
This commit is contained in:
Veetaha 2020-02-12 21:40:35 +02:00
parent 36dc3edb7a
commit a3febc1c57
1 changed files with 16 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
editors/code/src/installation/download_file.ts
View file

@ -1,8 +1,12 @@
import fetch from "node-fetch";
import * as fs from "fs";
import * as stream from "stream";
import * as util from "util";
import { strict as assert } from "assert";
import { NestedError } from "ts-nested-error";
const pipeline = util.promisify(stream.pipeline);
class DownloadFileError extends NestedError {}
/**
@ -29,25 +33,19 @@ export async function downloadFile(
const totalBytes = Number(res.headers.get('content-length'));
assert(!Number.isNaN(totalBytes), "Sanity check of content-length protocol");
let readBytes = 0;
console.log("Downloading file of", totalBytes, "bytes size from", url, "to", destFilePath);
// Here reject() may be called 2 times. As per ECMAScript standard, 2-d call is ignored
// https://tc39.es/ecma262/#sec-promise-reject-functions
let readBytes = 0;
res.body.on("data", (chunk: Buffer) => {
readBytes += chunk.length;
onProgress(readBytes, totalBytes);
});
return new Promise<void>((resolve, reject) => res.body
.on("data", (chunk: Buffer) => {
readBytes += chunk.length;
onProgress(readBytes, totalBytes);
})
.on("error", err => reject(
new DownloadFileError(`Read-stream error, read bytes: ${readBytes}`, err)
))
.pipe(fs.createWriteStream(destFilePath, { mode: destFilePermissions }))
.on("error", err => reject(
new DownloadFileError(`Write-stream error, read bytes: ${readBytes}`, err)
))
.on("close", resolve)
);
const destFileStream = fs.createWriteStream(destFilePath, { mode: destFilePermissions });
await pipeline(res.body, destFileStream).catch(DownloadFileError.rethrow("Piping file error"));
return new Promise<void>(resolve => {
destFileStream.on("close", resolve); // details on workaround: https://github.com/rust-analyzer/rust-analyzer/pull/3092#discussion_r378191131
destFileStream.destroy();
});
}