@middy/dynamodb 5.0.0-alpha.1 → 5.0.0-alpha.2

package/index.js

@@ -1,82 +1,64 @@
-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 } 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
-  }
-
-  // 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]
-      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
-        })
+    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);
     }
-    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 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 {
-    before: dynamodbMiddlewareBefore
-  }
-}
-
+    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
+    };
+};
 // used for TS type inference (see index.d.ts)
-export function dynamoDbReq (req) {
-  return req
+export function dynamoDbReq(req) {
+    return req;
 }
+export default dynamodbMiddleware;
 
-export default dynamodbMiddleware

package/package.json

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