Add request compression option (microsoft#1203)

Co-authored-by: Grzegorz Gurgul (from Dev Box) <ggurgul@microsoft.com>
v-mohithgc · Nov 10, 2023 · 214ba0c · 214ba0c
1 parent 329bfca
commit 214ba0c
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/Extensions/ArtifactEngine/Providers/typed-rest-client/Interfaces.ts b/Extensions/ArtifactEngine/Providers/typed-rest-client/Interfaces.ts
@@ -24,7 +24,8 @@ export interface IRequestOptions {
     allowRedirects?: boolean, 
     maxRedirects?: number,
     maxSockets?: number,
-    keepAlive?: boolean
+    keepAlive?: boolean,
+    requestCompressionForDownloads?: boolean
 }
 
 export interface IProxyConfiguration {

diff --git a/Extensions/ArtifactEngine/Providers/webProvider.ts b/Extensions/ArtifactEngine/Providers/webProvider.ts
@@ -25,6 +25,7 @@ export class WebProvider implements IArtifactProvider {
         this.templateFile = templateFile;
         this.webClient = WebClientFactory.getClient([handler], requestOptions);
         this.variables = variables;
+        this.requestCompressionForDownloads = requestOptions ? requestOptions.requestCompressionForDownloads : false;
     }
 
     getRootItems(): Promise<ArtifactItem[]> {
@@ -52,7 +53,17 @@ export class WebProvider implements IArtifactProvider {
             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) => {
+
+            const additionalHeaders = {};
+            if (contentType) {
+                additionalHeaders['Accept'] = contentType
+            }
+
+            if (this.requestCompressionForDownloads) {
+                additionalHeaders['Accept-Encoding'] = "gzip"
+            }
+
+            this.webClient.get(itemUrl, additionalHeaders).then((res: HttpClientResponse) => {
                 res.message.on('data', (chunk) => {
                     downloadSize += chunk.length;
                 });
@@ -145,5 +156,6 @@ export class WebProvider implements IArtifactProvider {
     private rootItemsLocation: string;
     private templateFile: string;
     private variables: string;
+    private requestCompressionForDownloads: boolean;
     public webClient: WebClient;
 }