fix(sqs): add total_record_count in SQS trigger lambda (#464)

* fix(sqs): add total_record_count in SQS trigger lambda * fix(sqs): improve metadata info for 2 methods Co-authored-by: Daniele Frasca <[email protected]> Co-authored-by: Ran Ribenzaft <[email protected]>
epsagon · Mar 18, 2021 · 6aa1922 · 6aa1922
1 parent e6ffae6
commit 6aa1922
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/src/events/aws_sdk.js b/src/events/aws_sdk.js
@@ -203,13 +203,39 @@ const SQSEventCreator = {
      * @param {proto.event_pb.Event} event The event to update the data on
      */
     responseHandler(response, event) {
+        let errorMessages = '';
+        let errorMessagesCount = 0;
         switch (response.request.operation) {
         case 'sendMessage':
             eventInterface.addToMetadata(event, {
                 'Message ID': `${response.data.MessageId}`,
                 'MD5 Of Message Body': `${response.data.MD5OfMessageBody}`,
             });
             break;
+        case 'sendMessageBatch':
+            if (response.data.Failed && response.data.Failed > 0) {
+                errorMessages = JSON.stringify(response.data.Failed
+                    .map(item => item));
+                errorMessagesCount = response.data.FailedRecordCount;
+            }
+            eventInterface.addToMetadata(event, {
+                successful_record_count: `${response.data.Successful.length}`,
+                failed_record_count: `${errorMessagesCount}`,
+                sqs_error_messages: errorMessages,
+            });
+            break;
+        case 'deleteMessageBatch':
+            if (response.data.Failed && response.data.Failed > 0) {
+                errorMessages = JSON.stringify(response.data.Failed
+                    .map(item => item));
+                errorMessagesCount = response.data.FailedRecordCount;
+            }
+            eventInterface.addToMetadata(event, {
+                successful_record_count: `${response.data.Successful.length}`,
+                failed_record_count: `${errorMessagesCount}`,
+                sqs_error_messages: errorMessages,
+            });
+            break;
         case 'receiveMessage': {
             let messagesNumber = 0;
             if (('Messages' in response.data) && (response.data.Messages.length > 0)) {

diff --git a/src/triggers/aws_lambda.js b/src/triggers/aws_lambda.js
@@ -125,6 +125,7 @@ function createSQSTrigger(event, trigger) {
             }
             return record;
         }),
+        total_record_count: event.Records.length,
     });
     try {
         const messageBody = JSON.parse(sqsMessageBody);