Add zipStream destroy call when response is aborted

microsoft · Oct 3, 2023 · e16671a · e16671a
1 parent 70c2754
commit e16671a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/Extensions/ArtifactEngine/Providers/webProvider.ts b/Extensions/ArtifactEngine/Providers/webProvider.ts
@@ -50,6 +50,7 @@ export class WebProvider implements IArtifactProvider {
  var downloadSize: number = 0;
  var contentType: string = artifactItem.contentType;
  var itemUrl: string = artifactItem.metadata['downloadUrl'];
+ var zipStream = null;
  itemUrl = itemUrl.replace(/([^:]\/)\/+/g, "$1");
  this.webClient.get(itemUrl, contentType ? { 'Accept': contentType } : undefined).then((res: HttpClientResponse) => {
  res.message.on('data', (chunk) => {
@@ -59,12 +60,17 @@ export class WebProvider implements IArtifactProvider {
  this.artifactItemStore.updateDownloadSize(artifactItem, downloadSize);
  });
  res.message.on('error', (error) => {
+ if (zipStream) {
+ zipStream.destroy(error);
+ Logger.logMessage(error);
+ }
  reject(error);
  });
 
  if (res.message.headers['content-encoding'] === 'gzip') {
  try {
- resolve(res.message.pipe(zlib.createUnzip()));
+ zipStream = zlib.createUnzip();
+ resolve(res.message.pipe(zipStream));
  }
  catch (err) {
  reject(err);