feat(batch-get.request): method to set ReturnConsumedCapacity

Author: mumenthalers

src/dynamo/batchget/batch-get.request.spec.ts

@@ -35,6 +35,11 @@ describe('batch get', () => {
         },
       ])
     })
+
+    it('returnConsumedCapacity', () => {
+      request.returnConsumedCapacity('TOTAL')
+      expect(request.params.ReturnConsumedCapacity).toBe('TOTAL')
+    })
   })
 
   describe('forModel', () => {

src/dynamo/batchget/batch-get.request.ts

@@ -24,6 +24,11 @@ export class BatchGetRequest {
     }
   }
 
+  returnConsumedCapacity(level: DynamoDB.ReturnConsumedCapacity): BatchGetRequest {
+    this.params.ReturnConsumedCapacity = level
+    return this
+  }
+
   /**
    * @param {ModelConstructor<T>} modelClazz
    * @param {Partial<T>[]} keys a partial of T that contains Partition key and SortKey (if necessary). Throws if missing.