refactor(fargate): emit phase started and completed events from workers

artilleryio · Feb 6, 2024 · cc71c23 · cc71c23
1 parent 0d740cb
commit cc71c23
Showing 1 changed file with 48 additions and 0 deletions.
diff --git a/...ages/artillery/lib/platform/aws-ecs/legacy/plugins/artillery-plugin-sqs-reporter/index.js b/...ages/artillery/lib/platform/aws-ecs/legacy/plugins/artillery-plugin-sqs-reporter/index.js
@@ -75,6 +75,54 @@ function ArtillerySQSPlugin(script, events) {
     });
   });
 
+  events.on('phaseStarted', (phaseContext) => {
+    this.unsent++;
+    const body = JSON.stringify({
+      event: 'phaseStarted',
+      phase: phaseContext
+    });
+
+    const params = {
+      MessageBody: body,
+      QueueUrl: this.queueUrl,
+      MessageAttributes: this.messageAttributes,
+      MessageDeduplicationId: uuid(),
+      MessageGroupId: this.testId
+    };
+
+    this.sqs.sendMessage(params, (err, data) => {
+      if (err) {
+        console.error(err);
+      }
+
+      this.unsent--;
+    });
+  });
+
+  events.on('phaseCompleted', (phaseContext) => {
+    this.unsent++;
+    const body = JSON.stringify({
+      event: 'phaseCompleted',
+      phase: phaseContext
+    });
+
+    const params = {
+      MessageBody: body,
+      QueueUrl: this.queueUrl,
+      MessageAttributes: this.messageAttributes,
+      MessageDeduplicationId: uuid(),
+      MessageGroupId: this.testId
+    };
+
+    this.sqs.sendMessage(params, (err, data) => {
+      if (err) {
+        console.error(err);
+      }
+
+      this.unsent--;
+    });
+  });
+
   events.on('done', (_stats) => {
     this.unsent++;
     const body = JSON.stringify({