updated dbcontext

Author: DHAMODHARABALAJI R

order-manager/src/handlers/createOrder.js

@@ -1,17 +1,14 @@
 // const { setCache } = require("../utils/cache");
 // const { sendToQueue } = require("../utils/sqs");
 const { v4: uuidv4 } = require("uuid");
-const { DynamoDBClient } = require("@aws-sdk/client-dynamodb");
-const { PutCommand, DynamoDBDocumentClient } = require("@aws-sdk/lib-dynamodb");
-const client = new DynamoDBClient({});
-const docClient = DynamoDBDocumentClient.from(client);
-const ordersTable = "OrdersTable";
+const dbContext = require("../utils/dbContext");
+
 
 
 exports.handler = async (event) => {
   try {
 
-    const { id, items } = event;
+    const { items } = event;
 
     if (!items || !items.length) {
       throw 'Items not configured in the payload'
@@ -22,12 +19,8 @@ exports.handler = async (event) => {
       items
     };
 
-    const command = new PutCommand({
-      TableName: ordersTable,
-      Item: order,
-    });
+    const response = await dbContext.Orders().Put(order);
 
-    const response = await docClient.send(command);
     console.log('order-created:', order, response);
 
     // Payload Sample

order-manager/src/handlers/getMyOrders.js

@@ -1,16 +1,8 @@
-const { DynamoDBClient } = require("@aws-sdk/client-dynamodb");
-const { ScanCommand, DynamoDBDocumentClient } = require("@aws-sdk/lib-dynamodb");
-const client = new DynamoDBClient({});
-const docClient = DynamoDBDocumentClient.from(client);
-const ordersTable = "OrdersTable";
+const dbContext = require("../utils/dbContext");
 
 exports.handler = async (event) => {
 
-  const body = await dynamo.send(
-    new ScanCommand({
-      TableName: ordersTable,
-    })
-  );
+  const body = await dbContext.Orders().Scan();
 
   console.log('items: ', body.Items)
 

order-manager/src/utils/dbContext.js

@@ -0,0 +1,50 @@
+const { DynamoDBClient } = require("@aws-sdk/client-dynamodb");
+const { GetCommand, ScanCommand, PutCommand, DynamoDBDocumentClient } = require("@aws-sdk/lib-dynamodb");
+const client = new DynamoDBClient({});
+const docClient = DynamoDBDocumentClient.from(client);
+const ordersTable = "OrdersTable";
+
+// console.log("🔒 HOSTING_ENV:", process.env.HOSTING_ENV);
+// console.log("🔒 AWS Region:", process.env.AWS_REGION);
+
+const dbActions = (tableName) => {
+
+    const actions = {
+        Get: async (whereClause) => {
+            const command = new GetCommand({
+                TableName: tableName,
+                Key: whereClause
+            });
+
+            const response = await docClient.send(command);
+            return response.Item;
+        },
+        Put: async (obj) => {
+            const command = new PutCommand({
+                TableName: tableName,
+                Item: obj,
+            });
+
+            const response = await docClient.send(command);
+            return response;
+        },
+        Scan: async () => {
+            const body = await docClient.send(
+                new ScanCommand({
+                    TableName: tableName,
+                })
+            );
+            return body.Items;
+        },
+    };
+    return (actions)
+
+}
+
+const dbContext = {
+
+    Orders: () => dbActions(ordersTable)
+
+};
+
+module.exports = dbContext;