aws
/
sns-sqs-test
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

now the rems stuff is labelled rems

master
Bel LaPointe 2023-02-06 08:03:05 -07:00
parent 9a916ada09
commit 9befefb89c
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
main.go
View File

@ -29,7 +29,7 @@ type Config struct {
AWSR string AWSR string
Topic string Topic string
Message string Message string
Wrap bool REMSWrap bool
MessageID string MessageID string
Ack bool Ack bool
Pub bool Pub bool
@ -85,12 +85,12 @@ func getConfig() (Config, error) {
flag.StringVar(&c.Topic, "topic", "rems-queue-prod-fra1-priority-low", "sns topic/sqs queue") flag.StringVar(&c.Topic, "topic", "rems-queue-prod-fra1-priority-low", "sns topic/sqs queue")
flag.StringVar(&c.Message, "message", `{"messageType": "NoOpMessage"}`, "published message") flag.StringVar(&c.Message, "message", `{"messageType": "NoOpMessage"}`, "published message")
flag.StringVar(&c.MessageID, "message-id", "", "sqs message id to ack") flag.StringVar(&c.MessageID, "message-id", "", "sqs message id to ack")
flag.BoolVar(&c.Wrap, "wrap", true, "rems wrap message") flag.BoolVar(&c.REMSWrap, "rems-wrap", true, "rems wrap message")
flag.BoolVar(&c.Ack, "ack", true, "ack message matching message-id") flag.BoolVar(&c.Ack, "ack", true, "ack message matching message-id")
flag.BoolVar(&c.Pub, "pub", false, "publish message") flag.BoolVar(&c.Pub, "pub", false, "publish message")
flag.Parse() flag.Parse()
if c.Wrap { if c.REMSWrap {
wrap.Ciphertext = cter{} wrap.Ciphertext = cter{}
var v interface{} var v interface{}
if err := json.Unmarshal([]byte(c.Message), &v); err != nil { if err := json.Unmarshal([]byte(c.Message), &v); err != nil {
@ -144,7 +144,9 @@ func (c Config) Consume(ctx context.Context) error {
} }
for _, message := range messages.Messages { for _, message := range messages.Messages {
var v interface{} var v interface{}
if c.REMSWrap {
wrap.Unwrap([]byte(*message.Body), &v, isoc{}) wrap.Unwrap([]byte(*message.Body), &v, isoc{})
}
log.Printf("message in queue: %s: %s: %+v", *message.MessageId, *message.Body, v) log.Printf("message in queue: %s: %s: %+v", *message.MessageId, *message.Body, v)
if *message.MessageId != c.MessageID { if *message.MessageId != c.MessageID {
if err := c.unack(ctx, message); err != nil { if err := c.unack(ctx, message); err != nil {