Fix consumer crash when no message received

amazon-archives · May 22, 2019 · 7437d80 · 7437d80
1 parent 1302cd6
commit 7437d80
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/samples/sqs/consumer/main.go b/samples/sqs/consumer/main.go
@@ -61,12 +61,14 @@ func main() {
 		} else {
 			fmt.Println("message:", msg)
 		}
-		_, err = svc.DeleteMessageRequest(&sqs.DeleteMessageInput{
-			QueueUrl:      q.QueueUrl,
-			ReceiptHandle: msg.Messages[0].ReceiptHandle,
-		}).Send()
-		if err != nil {
-			fmt.Println("error deleting message from queue:", err)
+		if len(msg.Messages) > 0 {
+			_, err = svc.DeleteMessageRequest(&sqs.DeleteMessageInput{
+				QueueUrl:      q.QueueUrl,
+				ReceiptHandle: msg.Messages[0].ReceiptHandle,
+			}).Send()
+			if err != nil {
+				fmt.Println("error deleting message from queue:", err)
+			}
 		}
 		// Implement some delay here to simulate processing time
 		time.Sleep(time.Duration(1000) * time.Millisecond)