add remove and update function

ebisbe · Jul 11, 2024 · c557e39 · c557e39
1 parent 8e37a21
commit c557e39
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 12 deletions.
diff --git a/src/stepFunctions/itemStream.ts b/src/stepFunctions/itemStream.ts
@@ -74,16 +74,23 @@ export const itemStream = {
             Type: "Task",
             Resource: "arn:aws:states:::dynamodb:updateItem",
             Parameters: {
-              TableName: "MyDynamoDBTable",
+              TableName: getRef(WeddingTable),
               Key: {
-                Column: {
-                  S: "MyEntry",
+                pk: {
+                  "S.$": "States.Format('LIST#{}',$.NewImage.listId.S)",
+                },
+                sk: {
+                  "S.$": "States.Format('LIST#{}',$.NewImage.listId.S)",
                 },
               },
-              UpdateExpression: "SET MyKey = :myValueRef",
+              UpdateExpression:
+                "SET totalPrice = totalPrice - :oldPrice + :newPrice",
               ExpressionAttributeValues: {
-                ":myValueRef": {
-                  S: "MyValue",
+                ":newPrice": {
+                  "N.$": "$.NewImage.price.N",
+                },
+                ":oldPrice": {
+                  "N.$": "$.OldImage.price.N",
                 },
               },
             },
@@ -93,16 +100,23 @@ export const itemStream = {
             Type: "Task",
             Resource: "arn:aws:states:::dynamodb:updateItem",
             Parameters: {
-              TableName: "MyDynamoDBTable",
+              TableName: getRef(WeddingTable),
               Key: {
-                Column: {
-                  S: "MyEntry",
+                pk: {
+                  "S.$": "States.Format('LIST#{}',$.NewImage.listId.S)",
+                },
+                sk: {
+                  "S.$": "States.Format('LIST#{}',$.NewImage.listId.S)",
                 },
               },
-              UpdateExpression: "SET MyKey = :myValueRef",
+              UpdateExpression:
+                "SET totalPrice = totalPrice - :itemPrice, totalItems = totalItems - :one",
               ExpressionAttributeValues: {
-                ":myValueRef": {
-                  S: "MyValue",
+                ":itemPrice": {
+                  "N.$": "$.NewImage.price.N",
+                },
+                ":one": {
+                  N: "1",
                 },
               },
             },

diff --git a/src/types/extendedSlsTypes.ts b/src/types/extendedSlsTypes.ts
@@ -57,6 +57,11 @@ export function getJoin(delimiter: string, stringsList: string[]): string {
   return { "Fn::Join": [delimiter, stringsList] } as unknown as string;
 }
 
+/**
+ * Returns a CF reference as string
+ * @param name string
+ * @returns string
+ */
 export function getRef(name: string): string {
   return { Ref: name } as unknown as string;
 }