@middy/dynamodb 5.1.0 → 5.2.1

package/index.js

@@ -1,62 +1,85 @@
-import { canPrefetch, createPrefetchClient, createClient, getCache, getInternal, processCache, modifyCache } from '@middy/util';
-import { DynamoDBClient, GetItemCommand } from '@aws-sdk/client-dynamodb';
-import { marshall, unmarshall } from '@aws-sdk/util-dynamodb';
+import {
+  canPrefetch,
+  createPrefetchClient,
+  createClient,
+  getCache,
+  getInternal,
+  processCache,
+  modifyCache,
+  catchInvalidSignatureException
+} from '@middy/util'
+import { DynamoDBClient, GetItemCommand } from '@aws-sdk/client-dynamodb'
+import { marshall, unmarshall } from '@aws-sdk/util-dynamodb'
+
 const defaults = {
-    AwsClient: DynamoDBClient,
-    awsClientOptions: {},
-    awsClientAssumeRole: undefined,
-    awsClientCapture: undefined,
-    fetchData: {},
-    disablePrefetch: false,
-    cacheKey: 'dynamodb',
-    cacheKeyExpiry: {},
-    cacheExpiry: -1,
-    setToContext: false
-};
-const dynamodbMiddleware = (opts = {})=>{
-    const options = {
-        ...defaults,
-        ...opts
-    };
-    for(const internalKey in options.fetchData){
-        options.fetchData[internalKey].Key = marshall(options.fetchData[internalKey].Key);
+  AwsClient: DynamoDBClient,
+  awsClientOptions: {},
+  awsClientAssumeRole: undefined,
+  awsClientCapture: undefined,
+  fetchData: {},
+  disablePrefetch: false,
+  cacheKey: 'dynamodb',
+  cacheKeyExpiry: {},
+  cacheExpiry: -1,
+  setToContext: false
+}
+const dynamodbMiddleware = (opts = {}) => {
+  const options = {
+    ...defaults,
+    ...opts
+  }
+
+  // force marshall of Key during cold start
+  for (const internalKey in options.fetchData) {
+    options.fetchData[internalKey].Key = marshall(
+      options.fetchData[internalKey].Key
+    )
+  }
+
+  const fetch = (request, cachedValues = {}) => {
+    const values = {}
+    for (const internalKey in options.fetchData) {
+      if (cachedValues[internalKey]) continue
+      const inputParameters = options.fetchData[internalKey]
+      const command = new GetItemCommand(inputParameters)
+      values[internalKey] = client
+        .send(command)
+        .catch((e) => catchInvalidSignatureException(e, client, command))
+        .then((resp) => unmarshall(resp.Item))
+        .catch((e) => {
+          const value = getCache(options.cacheKey).value ?? {}
+          value[internalKey] = undefined
+          modifyCache(options.cacheKey, value)
+          throw e
+        })
     }
-    const fetch = (request, cachedValues = {})=>{
-        const values = {};
-        for(const internalKey in options.fetchData){
-            if (cachedValues[internalKey]) continue;
-            const inputParameters = options.fetchData[internalKey];
-            values[internalKey] = client.send(new GetItemCommand(inputParameters)).then((resp)=>unmarshall(resp.Item)).catch((e)=>{
-                const value = getCache(options.cacheKey).value ?? {};
-                value[internalKey] = undefined;
-                modifyCache(options.cacheKey, value);
-                throw e;
-            });
-        }
-        return values;
-    };
-    let client;
-    if (canPrefetch(options)) {
-        client = createPrefetchClient(options);
-        processCache(options, fetch);
+    return values
+  }
+
+  let client
+  if (canPrefetch(options)) {
+    client = createPrefetchClient(options)
+    processCache(options, fetch)
+  }
+  const dynamodbMiddlewareBefore = async (request) => {
+    if (!client) {
+      client = await createClient(options, request)
+    }
+    const { value } = processCache(options, fetch, request)
+    Object.assign(request.internal, value)
+    if (options.setToContext) {
+      const data = await getInternal(Object.keys(options.fetchData), request)
+      Object.assign(request.context, data)
     }
-    const dynamodbMiddlewareBefore = async (request)=>{
-        if (!client) {
-            client = await createClient(options, request);
-        }
-        const { value } = processCache(options, fetch, request);
-        Object.assign(request.internal, value);
-        if (options.setToContext) {
-            const data = await getInternal(Object.keys(options.fetchData), request);
-            Object.assign(request.context, data);
-        }
-    };
-    return {
-        before: dynamodbMiddlewareBefore
-    };
-};
-export function dynamoDbReq(req) {
-    return req;
+  }
+  return {
+    before: dynamodbMiddlewareBefore
+  }
+}
+
+// used for TS type inference (see index.d.ts)
+export function dynamoDbReq (req) {
+  return req
 }
-export default dynamodbMiddleware;
 
+export default dynamodbMiddleware

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@middy/dynamodb",
-  "version": "5.1.0",
+  "version": "5.2.1",
   "description": "DynamoDB middleware for the middy framework",
   "type": "module",
   "engines": {
@@ -58,14 +58,14 @@
     "url": "https://github.com/sponsors/willfarrell"
   },
   "dependencies": {
-    "@middy/util": "5.1.0"
+    "@middy/util": "5.2.1"
   },
   "devDependencies": {
     "@aws-sdk/client-dynamodb": "^3.245.0",
     "@aws-sdk/util-dynamodb": "^3.245.0",
-    "@middy/core": "5.1.0",
+    "@middy/core": "5.2.1",
     "@types/aws-lambda": "^8.10.101",
     "aws-xray-sdk": "^3.3.3"
   },
-  "gitHead": "bbdaf5843914921804ba085dd58117273febe6b5"
+  "gitHead": "4d55da221b9165b4b3e59a12632fd40a149a1e92"
 }