Merge branch 'master' into update-doc-build-step

shiftcode · Oct 26, 2018 · cc94618 · cc94618
2 parents 3a99899 + e334f72
commit cc94618
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/dynamo/request/batchwritesingletable/batch-write-single-table.request.ts b/src/dynamo/request/batchwritesingletable/batch-write-single-table.request.ts
@@ -64,13 +64,15 @@ export class BatchWriteSingleTableRequest<T> {
 
     return this.dynamoRx.batchWriteItem(batchWriteItemInput).pipe(
       tap((batchWriteManyResponse: BatchWriteItemOutput) => {
-        if (batchWriteManyResponse.UnprocessedItems) {
+        if (batchWriteManyResponse.UnprocessedItems && batchWriteManyResponse.UnprocessedItems[this.tableName]) {
           this.itemsToProcess.unshift(...batchWriteManyResponse.UnprocessedItems[this.tableName])
         }
       }),
       map((batchWriteManyResponse: BatchWriteItemOutput) => ({
         remainingItems: this.itemsToProcess.length,
-        capacityExceeded: !!batchWriteManyResponse.UnprocessedItems,
+        capacityExceeded: !!(
+          batchWriteManyResponse.UnprocessedItems && batchWriteManyResponse.UnprocessedItems[this.tableName]
+        ),
         consumedCapacity: batchWriteManyResponse.ConsumedCapacity,
       }))
     )