@eresearchqut/ddb-repository 1.2.0 → 1.3.0

package/CHANGELOG.md

@@ -1,3 +1,10 @@
+# [1.3.0](https://github.com/eresearchqut/ddb-repository/compare/v1.2.0...v1.3.0) (2025-11-20)
+
+
+### Features
+
+* Add test for the tracking of the consumed capacity ([4a1b23c](https://github.com/eresearchqut/ddb-repository/commit/4a1b23cadadbbb02d78507ca9d34715219fc952c))
+
 # [1.2.0](https://github.com/eresearchqut/ddb-repository/compare/v1.1.0...v1.2.0) (2025-11-19)
 
 

package/dist/DynamoDbRepository.js

@@ -27,7 +27,7 @@ var __asyncValues = (this && this.__asyncValues) || function (o) {
     function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.DynamoDbRepository = exports.mapFilterExpressions = exports.mapFilterExpression = exports.FilterOperator = void 0;
+exports.DynamoDbRepository = exports.consumedCapacityMiddleware = exports.mapFilterExpressions = exports.mapFilterExpression = exports.FilterOperator = void 0;
 const client_dynamodb_1 = require("@aws-sdk/client-dynamodb");
 const util_dynamodb_1 = require("@aws-sdk/util-dynamodb");
 const lodash_1 = require("lodash");
@@ -79,6 +79,21 @@ const paginate = (array, pageSize) => {
         return acc;
     }, []);
 };
+const consumedCapacityMiddleware = (consumedCapacityMiddlewareConfig) => (next, context) => (args) => __awaiter(void 0, void 0, void 0, function* () {
+    try {
+        const { input } = args;
+        const { ReturnConsumedCapacity } = input;
+        const response = yield next(args);
+        const { output } = response;
+        const consumedCapacity = (0, lodash_1.get)(output, "ConsumedCapacity");
+        yield consumedCapacityMiddlewareConfig.onConsumedCapacity({ ReturnConsumedCapacity, ConsumedCapacity: consumedCapacity });
+        return response;
+    }
+    catch (error) {
+        throw error;
+    }
+});
+exports.consumedCapacityMiddleware = consumedCapacityMiddleware;
 class DynamoDbRepository {
     constructor(dynamoDBClient, tableName, hashKey, rangKey, returnConsumedCapacity = client_dynamodb_1.ReturnConsumedCapacity.TOTAL) {
         this.dynamoDBClient = dynamoDBClient;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@eresearchqut/ddb-repository",
-  "version": "1.2.0",
+  "version": "1.3.0",
   "description": "",
   "type": "module",
   "main": "dist/index.js",
@@ -23,6 +23,7 @@
     "@semantic-release/git": "^10.0.1",
     "@semantic-release/github": "^12.0.2",
     "@semantic-release/npm": "^13.1.2",
+    "@smithy/types": "^4.9.0",
     "@testcontainers/localstack": "^10.15.0",
     "@types/jest": "^29.5.0",
     "@types/lodash": "^4.17.20",

package/src/DynamoDbRepository.ts

@@ -1,6 +1,7 @@
 import {
     BatchGetItemCommand,
     BatchGetItemCommandInput,
+    ConsumedCapacity,
     DeleteItemCommand,
     DynamoDBClient,
     GetItemCommand,
@@ -11,7 +12,14 @@ import {
     UpdateItemCommand,
 } from "@aws-sdk/client-dynamodb";
 import {marshall, unmarshall} from "@aws-sdk/util-dynamodb";
-import {replace, uniqWith, isEqual, pickBy} from "lodash";
+import {replace, uniqWith, isEqual, pickBy, get} from "lodash";
+import {
+    HandlerExecutionContext,
+    InitializeHandler,
+    InitializeHandlerArguments,
+    InitializeHandlerOutput,
+    MetadataBearer
+} from "@smithy/types";
 
 const expressionAttributeKey = (key: string) => replace(key, /-/g, "_");
 
@@ -119,6 +127,35 @@ const paginate = <T>(array: Array<T>, pageSize: number) => {
     }, [] as Array<Array<T>>);
 }
 
+export interface ConsumedCapacityDetail {
+    ReturnConsumedCapacity: ReturnConsumedCapacity | undefined
+    ConsumedCapacity: ConsumedCapacity | undefined
+}
+
+export interface ConsumedCapacityMiddlewareConfig {
+    onConsumedCapacity: (consumedCapacity: ConsumedCapacityDetail) => Promise<unknown>;
+}
+
+export const consumedCapacityMiddleware =
+    (consumedCapacityMiddlewareConfig: ConsumedCapacityMiddlewareConfig) =>
+        <Output extends MetadataBearer = MetadataBearer>(
+            next: InitializeHandler<any, Output>,
+            context: HandlerExecutionContext
+        ): InitializeHandler<any, Output> =>
+            async (args: InitializeHandlerArguments<any>): Promise<InitializeHandlerOutput<Output>> => {
+                try {
+                    const {input} = args;
+                    const {ReturnConsumedCapacity} = input;
+                    const response = await next(args);
+                    const {output} = response;
+                    const consumedCapacity = get(output, "ConsumedCapacity") as ConsumedCapacity | undefined;
+                    await consumedCapacityMiddlewareConfig.onConsumedCapacity({ReturnConsumedCapacity, ConsumedCapacity: consumedCapacity});
+                    return response;
+                } catch (error) {
+                    throw error;
+                }
+            };
+
 export class DynamoDbRepository<K, T> {
 
     constructor(

package/test/repository.test.ts

@@ -1,7 +1,7 @@
 
 import { DynamoDBClient, CreateTableCommand, DescribeTableCommand } from "@aws-sdk/client-dynamodb";
 import { LocalstackContainer, StartedLocalStackContainer } from "@testcontainers/localstack";
-import {DynamoDbRepository, FilterOperator} from "../src";
+import {ConsumedCapacityDetail, consumedCapacityMiddleware, DynamoDbRepository, FilterOperator} from "../src";
 
 describe('DynamoDbRepository Integration Tests', () => {
     let container: StartedLocalStackContainer;
@@ -12,6 +12,11 @@ describe('DynamoDbRepository Integration Tests', () => {
     const tableName = 'test-table';
     const compositeTableName = 'test-composite-table';
     const gsiTableName = 'test-gsi-table';
+    const consumedCapacityRegister = new Array<ConsumedCapacityDetail>() ;
+
+    const sumConsumedCapacity = () =>
+        consumedCapacityRegister.reduce((total, value) =>
+            total + (value.ConsumedCapacity?.CapacityUnits || 0), 0);
 
     beforeAll(async () => {
         // Start LocalStack container with DynamoDB
@@ -28,7 +33,10 @@ describe('DynamoDbRepository Integration Tests', () => {
             },
         });
 
-        // Create the test table with simple key
+        dynamoDBClient.middlewareStack
+            .add(consumedCapacityMiddleware({onConsumedCapacity: async (consumedCapacity) => consumedCapacityRegister.push(consumedCapacity)}));
+
+        // Create the test table with a simple key
         await dynamoDBClient.send(
             new CreateTableCommand({
                 TableName: tableName,
@@ -42,7 +50,7 @@ describe('DynamoDbRepository Integration Tests', () => {
             })
         );
 
-        // Create the test table with composite key (partition key + sort key)
+        // Create the test table with a composite key (partition key + sort key)
         await dynamoDBClient.send(
             new CreateTableCommand({
                 TableName: compositeTableName,
@@ -108,6 +116,10 @@ describe('DynamoDbRepository Integration Tests', () => {
         gsiRepository = new DynamoDbRepository(dynamoDBClient, gsiTableName, "userId", "itemId");
     });
 
+    afterEach(async () => {
+        consumedCapacityRegister.splice(0, consumedCapacityRegister.length);
+    });
+
     afterAll(async () => {
         // Clean up
         if (dynamoDBClient) {
@@ -122,6 +134,7 @@ describe('DynamoDbRepository Integration Tests', () => {
         it('should return undefined when item does not exist', async () => {
             const result = await repository.getItem({ id: 'non-existent' });
             expect(result).toBeUndefined();
+            expect(sumConsumedCapacity()).toEqual(0.5);
         });
 
         it('should return the item when it exists', async () => {
@@ -132,6 +145,7 @@ describe('DynamoDbRepository Integration Tests', () => {
             const result = await repository.getItem(key);
 
             expect(result).toEqual(record);
+            expect(sumConsumedCapacity()).toEqual(2);
         });
     });
 
@@ -143,6 +157,7 @@ describe('DynamoDbRepository Integration Tests', () => {
             const result = await repository.putItem(key, record);
 
             expect(result).toEqual(record);
+            expect(sumConsumedCapacity()).toEqual(1.5);
         });
     });
 
@@ -155,6 +170,7 @@ describe('DynamoDbRepository Integration Tests', () => {
             await repository.deleteItem(key);
             const afterDelete = await repository.getItem(key);
             expect(afterDelete).toBeUndefined();
+            expect(sumConsumedCapacity()).toEqual(2);
         });
     });