feat: init db wrapper

adobe · Oct 19, 2023 · c15a214 · c15a214
1 parent fabbd71
commit c15a214
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 13 deletions.
diff --git a/src/db-wrapper.js b/src/db-wrapper.js
@@ -13,15 +13,21 @@ import createDynamoDBService from './db.js';
 import serviceWrap from './service-wrap.js';
 
 export default function dynamoDBWrapper(func) {
-  const region = process.env.AWS_REGION;
-  if (!region) {
-    throw Error('Please define region in secrets');
-  }
-  return (request, context) => serviceWrap(
-    func,
-    request,
-    { ...context, region },
-    '__ow_dynamodb',
-    createDynamoDBService,
-  );
+  return async (...args) => {
+    let params = {};
+    let context;
+    [params, context = {}] = args;
+    const region = process.env.AWS_REGION;
+    if (!region) {
+      throw Error('Please define region in secrets');
+    }
+    context = { ...context, region };
+    serviceWrap(
+      params,
+      context,
+      '__ow_dynamodb',
+      createDynamoDBService,
+    );
+    return func(...args);
+  };
 }
diff --git a/src/service-wrap.js b/src/service-wrap.js
@@ -10,7 +10,6 @@
  * governing permissions and limitations under the License.
  */
 export default function serviceWrap(
-  lambdaFn,
   lambdaRequest,
   lambdaContext,
   paramName,
@@ -22,5 +21,4 @@ export default function serviceWrap(
     // eslint-disable-next-line no-param-reassign
     lambdaContext[paramName] = factoryFn(lambdaContext);
   }
-  return lambdaFn(lambdaContext);
 }