feat: init db wrapper

adobe · Oct 19, 2023 · 95ff60d · 95ff60d
1 parent 057be2d
commit 95ff60d
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 11 deletions.
diff --git a/src/db-wrapper.js b/src/db-wrapper.js
@@ -13,9 +13,10 @@ import createDynamoDBService from './db.js';
 import serviceWrap from './service-wrap.js';
 
 function wrapper(func) {
-  return (params) => serviceWrap(
+  return (request, context) => serviceWrap(
     func,
-    params,
+    request,
+    context,
     '__ow_dynamodb',
     createDynamoDBService,
   );

diff --git a/src/db.js b/src/db.js
@@ -16,8 +16,8 @@ import { log } from './util.js';
 const TABLE_SITES = 'spacecat-site';
 const TABLE_AUDITS = 'spacecat-audit-index';
 
-function DB(params) {
-  const client = new DynamoDBClient({ region: params.region });
+function DB() {
+  const client = new DynamoDBClient({ region: process.env.REGION });
   const docClient = DynamoDBDocumentClient.from(client);
 
   /**
@@ -136,8 +136,8 @@ function DB(params) {
   };
 }
 
-const createDynamoDBService = (params) => Object.freeze({
-  getInstance: () => DB(params),
+const createDynamoDBService = () => Object.freeze({
+  getInstance: () => DB(),
 });
 
 export default { createDynamoDBService };
diff --git a/src/service-wrap.js b/src/service-wrap.js
@@ -11,19 +11,20 @@
  */
 const serviceWrap = (
   lambdaFn,
-  lambdaParams,
+  lambdaRequest,
+  lambdaContext,
   paramName,
   factoryFn,
 ) => {
-  if (!lambdaParams[paramName]) {
-    const service = factoryFn(lambdaParams);
+  if (!lambdaContext[paramName]) {
+    const service = factoryFn(lambdaContext);
 
     // pass params by reference and not value so later modifications
     // of the params are accessible to the wrap
     // eslint-disable-next-line no-param-reassign
-    lambdaParams[paramName] = service.getInstance(lambdaParams);
+    lambdaContext[paramName] = service.getInstance(lambdaContext);
   }
-  return lambdaFn(lambdaParams);
+  return lambdaFn(lambdaContext);
 };
 
 module.exports = serviceWrap;