TW-2013: Work around the Dax issue with parsing dax response (Blackbi…

…rdHQ#4)
rchl · Apr 27, 2022 · 8ad2d19 · 8ad2d19
1 parent 4ec7b3a
commit 8ad2d19
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/CustomAssemblers.js b/src/CustomAssemblers.js
@@ -18,6 +18,7 @@ const Assembler = require('./Assembler');
 const Util = require('./Util');
 const DaxClientError = require('./DaxClientError');
 const DaxErrorCode = require('./DaxErrorCode');
+const CborDecoder = require('./CborDecoder');
 
 exports.Custom_defineKeySchema_N742646399_1_Assembler = class Custom_defineKeySchema_N742646399_1_Assembler extends Assembler {
   _assembleResult() {
@@ -118,9 +119,16 @@ exports.Custom_batchGetItem_N697851100_1_Assembler = class Custom_batchGetItem_N
       }
     });
 
-    let consumedCapacities = this.dec.buildArray(() => Assembler._decodeConsumedCapacityData(this.dec.decodeCbor()));
-    if(this._request.ReturnConsumedCapacity && this._request.ReturnConsumedCapacity !== 'NONE') {
-      result.ConsumedCapacity = verifyBatchConsumedCapacity(consumedCapacities, Object.getOwnPropertyNames(this._request.RequestItems));
+    try {
+      let consumedCapacities = this.dec.buildArray(() => Assembler._decodeConsumedCapacityData(this.dec.decodeCbor()));
+      if(this._request.ReturnConsumedCapacity && this._request.ReturnConsumedCapacity !== 'NONE') {
+        result.ConsumedCapacity = verifyBatchConsumedCapacity(consumedCapacities, Object.getOwnPropertyNames(this._request.RequestItems));
+      }
+    } catch (error) {
+      // Ignore "Not enough data" error (TW-2013) until upstream provides a proper fix.
+      if (!(error instanceof CborDecoder.NeedMoreData)) {
+        throw error
+      }
     }
 
     return result;