Session 4 : Step Functions

.vscode/settings.json

@@ -0,0 +1,3 @@
+{  
+    "eslint.workingDirectories": [{ "mode": "auto"}]
+}

backend/.eslintrc.json

@@ -12,7 +12,8 @@
   },
   "rules": {
     "prettier/prettier": "error",
-    "@typescript-eslint/no-var-requires": "off"
-  },
+    "@typescript-eslint/no-var-requires": "off",
+    "@typescript-eslint/ban-ts-ignore": "off"
+},
   "plugins": ["prettier"]
 }

backend/serverless.ts

@@ -2,7 +2,7 @@ import * as AwsConfig from 'serverless/aws';
 
 import ApiGatewayErrors from './resources/apiGatewayErrors';
 import DojoServerlessTable from './resources/dynamodb';
-// import ApplicationEventBus from './resources/eventBridge';
+import ApplicationEventBus from './resources/eventBridge';
 
 const serverlessConfiguration: AwsConfig.Serverless = {
   service: 'dojo-serverless-backend',
@@ -30,7 +30,11 @@ const serverlessConfiguration: AwsConfig.Serverless = {
         ],
         Resource: { 'Fn::GetAtt': ['DojoServerlessTable', 'Arn'] },
       },
-      // { Effect: 'Allow', Action: ['events:PutEvents'], Resource: '*' },
+      {
+        Effect: 'Allow',
+        Action: ['events:PutEvents'],
+        Resource: '*',
+      },
     ],
     usagePlan: {
       quota: {
@@ -97,7 +101,11 @@ const serverlessConfiguration: AwsConfig.Serverless = {
     //  --- WEBSOCKET ---
     connectWebsocket: {
       handler: 'src/handlers/real-time/connect.main',
-      events: [{ websocket: { route: '$connect' } }],
+      events: [
+        {
+          websocket: { route: '$connect' },
+        },
+      ],
     },
     disconnectWebsocket: {
       handler: 'src/handlers/real-time/disconnect.main',
@@ -131,37 +139,84 @@ const serverlessConfiguration: AwsConfig.Serverless = {
               'arn:aws:events:#{AWS::Region}:#{AWS::AccountId}:event-bus/dojo-serverless',
             pattern: {
               source: ['dojo-serverless'],
-              'detail-type': ['LAZYNESS_DETECTED'],
+              'detail-type': ['LAZYNESS_DETECTED'], // Je comprend pas quand cet evenement va etre detecte (par qui il est envoyé)
             },
           },
         },
       ],
     },
+
+    spreadVirus: {
+      handler: 'src/handlers/stateMachine/spreadVirus.main',
+      events: [
+        {
+          schedule: {
+            rate: 'rate(1 minute)',
+          },
+        },
+      ],
+    },
+
+    chooseWaitTime: {
+      handler: 'src/handlers/stateMachine/chooseWaitTime.main',
+    },
+    initiateMultipleViruses: {
+      handler: 'src/handlers/stateMachine/initiateMultipleViruses.main',
+    },
   },
   stepFunctions: {
     stateMachines: {
-      wait10SecondsAndDoNothing: {
+      generateMultipleViruses: {
         events: [
           {
-            cloudwatchEvent: {
+            eventBridge: {
               eventBusName:
                 'arn:aws:events:#{AWS::Region}:#{AWS::AccountId}:event-bus/dojo-serverless',
               event: {
                 source: ['dojo-serverless'],
-                'detail-type': ['NOTHING_REQUESTED'],
+                'detail-type': ['VIRUS_CREATION_REQUESTED'],
               },
             },
           },
         ],
         definition: {
-          StartAt: 'Wait10Sec',
+          StartAt: 'InitiateMultipleViruses',
           States: {
-            Wait10Sec: {
-              Type: 'Wait',
-              Seconds: 10,
-              Next: 'DoNothing',
+            InitiateMultipleViruses: {
+              Type: 'Task',
+              Resource: {
+                'Fn::GetAtt': ['initiateMultipleViruses', 'Arn'],
+              },
+              Next: 'MapVirusCreation',
+            },
+            MapVirusCreation: {
+              Type: 'Map',
+              Iterator: {
+                StartAt: 'ChooseWaitTime',
+                States: {
+                  ChooseWaitTime: {
+                    Type: 'Task',
+                    Resource: {
+                      'Fn::GetAtt': ['chooseWaitTime', 'Arn'],
+                    },
+                    Next: 'WaitXSeconds',
+                  },
+                  WaitXSeconds: {
+                    Type: 'Wait',
+                    SecondsPath: '$.timeInSeconds',
+                    Next: 'CreateVirus',
+                  },
+                  CreateVirus: {
+                    Type: 'Task',
+                    Resource: {
+                      'Fn::GetAtt': ['createVirus', 'Arn'],
+                    },
+                    End: true,
+                  },
+                },
+              },
+              End: true,
             },
-            DoNothing: { Type: 'Succeed' },
           },
         },
       },
@@ -171,7 +226,7 @@ const serverlessConfiguration: AwsConfig.Serverless = {
     Resources: {
       ...ApiGatewayErrors,
       DojoServerlessTable,
-      // ApplicationEventBus,
+      ApplicationEventBus,
     },
   },
 };

backend/src/handlers/stateMachine/chooseWaitTime.ts

@@ -0,0 +1,3 @@
+export const main = async (): Promise<{ timeInSeconds: number }> => ({
+  timeInSeconds: Math.ceil(Math.random() * 60),
+});

backend/src/handlers/stateMachine/initiateMultipleViruses.ts

@@ -0,0 +1,11 @@
+const maxViruses = 4;
+const minViruses = 2;
+
+export const main = async (): Promise<{ virusTaskId: number }[]> => {
+  const numberOfViruses =
+    minViruses + Math.floor(Math.random() * (maxViruses - minViruses + 1));
+  const data = Array.from({ length: numberOfViruses }, (_, i) => ({
+    virusTaskId: i + 1,
+  }));
+  return data;
+};

backend/src/handlers/stateMachine/requestNothing.ts

@@ -2,7 +2,7 @@ import EventBridge from 'aws-sdk/clients/eventbridge';
 
 const eventBridge = new EventBridge();
 
-export const main = async () => {
+export const main = async (): Promise<void> => {
   await eventBridge
     .putEvents({
       Entries: [

backend/src/handlers/stateMachine/spreadVirus.ts

@@ -0,0 +1,18 @@
+import EventBridge from 'aws-sdk/clients/eventbridge';
+
+const eventBridge = new EventBridge();
+
+export const main = async (): Promise<void> => {
+  await eventBridge
+    .putEvents({
+      Entries: [
+        {
+          Source: 'dojo-serverless',
+          DetailType: 'VIRUS_CREATION_REQUESTED',
+          Detail: JSON.stringify({}),
+          EventBusName: 'dojo-serverless',
+        },
+      ],
+    })
+    .promise();
+};