chore(middleware-content-length): set transfer-encoding if content-le…

…ngth is not set (#3403)
aws · Mar 9, 2022 · 7e4abf7 · 7e4abf7
1 parent a4beeba
commit 7e4abf7
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/packages/middleware-content-length/src/index.ts b/packages/middleware-content-length/src/index.ts
@@ -11,6 +11,7 @@ import {
 } from "@aws-sdk/types";
 
 const CONTENT_LENGTH_HEADER = "content-length";
+const TRANSFER_ENCODING_HEADER = "transfer-encoding";
 
 export function contentLengthMiddleware(bodyLengthChecker: BodyLengthCalculator): BuildMiddleware<any, any> {
   return <Output extends MetadataBearer>(next: BuildHandler<any, Output>): BuildHandler<any, Output> =>
@@ -24,12 +25,17 @@ export function contentLengthMiddleware(bodyLengthChecker: BodyLengthCalculator)
             .map((str) => str.toLowerCase())
             .indexOf(CONTENT_LENGTH_HEADER) === -1
         ) {
-          const length = bodyLengthChecker(body);
-          if (length !== undefined) {
+          try {
+            const length = bodyLengthChecker(body);
             request.headers = {
               ...request.headers,
               [CONTENT_LENGTH_HEADER]: String(length),
             };
+          } catch (error) {
+            request.headers = {
+              ...request.headers,
+              [TRANSFER_ENCODING_HEADER]: "chunked",
+            };
           }
         }
       }