feat: Update save-data function

Author: fabidick22

src/handlers/save-data/app.js

@@ -1,19 +1,37 @@
 
-let response;
+const dynamodb = require('aws-sdk/clients/dynamodb');
+const docClient = new dynamodb.DocumentClient();
+const tableName = process.env.DATA_TABLE;
+const crypto = require("crypto");
+
 
 exports.lambdaHandler = async (event, context) => {
-    try {
-        response = {
-            'statusCode': 200,
-            'body': JSON.stringify({
-                message: 'save data',
-                // location: ret.data.trim()
-            })
-        }
-    } catch (err) {
-        console.log(err);
-        return err;
+    if (event.httpMethod !== 'POST') {
+        throw new Error(`postMethod only accepts POST method, you tried: ${event.httpMethod} method.`);
     }
+    console.info('table name:', tableName);
+
+    const body = JSON.parse(event.body)
+    var params = {
+        TableName: tableName,
+        Item: {
+            id: crypto.randomBytes(16).toString("hex"),
+            message: body.message,
+        }
+    };
+
+    const result = await docClient.put(params).promise();
+    console.log(result);
+    const response = {
+        statusCode: 200,
+        headers: {
+            "Access-Control-Allow-Origin": "*",
+            "Access-Control-Allow-Methods": "OPTIONS,POST"
+        },
+        body: JSON.stringify({message: 'saved data'})
+    };
+
+    console.info(`response from: ${event.path} statusCode: ${response.statusCode} body: ${response.body}`);
 
     return response
 };

template.yaml

@@ -48,6 +48,12 @@ Resources:
       FunctionName: !Sub "${AWS::StackName}-saveMyDataFunction"
       CodeUri: src/handlers/save-data/
       Handler: app.lambdaHandler
+      Policies:
+        - DynamoDBCrudPolicy:
+            TableName: !Ref MySimpleTableDB
+      Environment:
+        Variables:
+          DATA_TABLE: !Ref MySimpleTableDB
       Events:
         HelloWorld:
           Type: HttpApi
@@ -56,6 +62,17 @@ Resources:
             Path: /data
             Method: post
 
+  MySimpleTableDB:
+    Type: AWS::Serverless::SimpleTable
+    Properties:
+      TableName: !Sub "my-data-${Environment}"
+      PrimaryKey:
+        Name: id
+        Type: String
+      ProvisionedThroughput:
+        ReadCapacityUnits: 2
+        WriteCapacityUnits: 2
+
 Outputs:
   DataApi:
     Description: "API Gateway endpoint URL for Prod stage for Hello World function"