From 8ce6763b1e656ce80b4dd038beef4b93abb23ac0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?France=CC=81=20Wilke?= <francewilke@gmail.com>
Date: Fri, 26 Nov 2021 14:10:26 +0200
Subject: [PATCH] Pass sqs message type through
---
sqs/sqs.go | 11 ++++++++---
1 file changed, 8 insertions(+), 3 deletions(-)
diff --git a/sqs/sqs.go b/sqs/sqs.go
index 6775da0..0e19dc7 100644
--- a/sqs/sqs.go
+++ b/sqs/sqs.go
@@ -49,7 +49,7 @@ func NewSQSMessenger(awsRegion, queueUrl string) (*Messenger, error) {
// SendSQSMessage sends a message to the queue associated with the messenger
// headers - string message attributes of the SQS message (see AWS SQS documentation)
// body - body of the SQS message (see AWS SQS documentation)
-func (m *Messenger) SendSQSMessage(headers map[string]string, body string, currentRequestID *string, headerKey string) (string, error) {
+func (m *Messenger) SendSQSMessage(headers map[string]string, body string, currentRequestID *string, sqsType string, headerKey string) (string, error) {
msgAttrs := make(map[string]*sqs.MessageAttributeValue)
for key, val := range headers {
@@ -67,6 +67,11 @@ func (m *Messenger) SendSQSMessage(headers map[string]string, body string, curre
}
}
+ msgAttrs["type"] = &sqs.MessageAttributeValue{
+ DataType: aws.String("String"),
+ StringValue: aws.String(sqsType),
+ }
+
res, err := m.service.SendMessage(&sqs.SendMessageInput{
MessageAttributes: msgAttrs,
MessageBody: aws.String(body),
@@ -80,7 +85,7 @@ func (m *Messenger) SendSQSMessage(headers map[string]string, body string, curre
return *res.MessageId, err
}
-func SendSQSMessage(msgr *Messenger, region string, envQueueURLName string, objectToSend interface{}, currentRequestID *string, headerKey string) error {
+func SendSQSMessage(msgr *Messenger, region string, envQueueURLName string, objectToSend interface{}, currentRequestID *string, sqsType string, headerKey string) error {
if msgr == nil {
var err error
msgr, err = NewSQSMessenger(region, os.Getenv(envQueueURLName))
@@ -96,7 +101,7 @@ func SendSQSMessage(msgr *Messenger, region string, envQueueURLName string, obje
}
headers := map[string]string{"Name": "dummy"}
- msgID, err := msgr.SendSQSMessage(headers, string(jsonBytes), currentRequestID, headerKey)
+ msgID, err := msgr.SendSQSMessage(headers, string(jsonBytes), currentRequestID, sqsType, headerKey)
if err != nil {
logs.ErrorWithMsg("Failed to send sqs event", err)
return err
--
GitLab