Add timeout into healthcheck

Author: minhduc140583

func.go

@@ -15,16 +15,16 @@ func GetQueueUrl(client *sqs.SQS, queueName string) (string, error) {
 	return *result.QueueUrl, err
 }
 
-func MapToAttributes(messageAttributes map[string]string) map[string]*sqs.MessageAttributeValue {
-	attributes := make(map[string]*sqs.MessageAttributeValue)
-	if messageAttributes != nil {
-		for k, v := range messageAttributes {
+func MapToAttributes(attributes map[string]string) map[string]*sqs.MessageAttributeValue {
+	attrs := make(map[string]*sqs.MessageAttributeValue)
+	if attributes != nil {
+		for k, v := range attributes {
 			x := sqs.MessageAttributeValue{
 				DataType:    aws.String("String"),
 				StringValue: aws.String(v),
 			}
-			attributes[k] = &x
+			attrs[k] = &x
 		}
 	}
-	return attributes
+	return attrs
 }

health_checker.go

@@ -3,15 +3,17 @@ package sqs
 import (
 	"context"
 	"github.com/aws/aws-sdk-go/service/sqs"
+	"time"
 )
 
 type HealthChecker struct {
 	Client    *sqs.SQS
 	QueueName *string
 	Service   string
+	Timeout   time.Duration
 }
 
-func NewHealthChecker(client *sqs.SQS, queueName string, options...string) *HealthChecker {
+func NewHealthChecker(client *sqs.SQS, queueName string, options ...string) *HealthChecker {
 	var name string
 	if len(options) > 0 && len(options[0]) > 0 {
 		name = options[0]
@@ -20,8 +22,14 @@ func NewHealthChecker(client *sqs.SQS, queueName string, options...string) *Heal
 	}
 	return NewSQSHealthChecker(client, name, queueName)
 }
-func NewSQSHealthChecker(client *sqs.SQS, name string, queueName string) *HealthChecker {
-	return &HealthChecker{Client: client, QueueName: &queueName, Service: name}
+func NewSQSHealthChecker(client *sqs.SQS, name string, queueName string, options ...time.Duration) *HealthChecker {
+	var timeout time.Duration
+	if len(options) >= 1 && options[0] > 0 {
+		timeout = options[0]
+	} else {
+		timeout = 4 * time.Second
+	}
+	return &HealthChecker{Client: client, QueueName: &queueName, Service: name, Timeout: timeout}
 }
 
 func (h *HealthChecker) Name() string {

sender.go

@@ -24,12 +24,12 @@ func NewSender(client *sqs.SQS, queueURL string, delaySeconds *int64) *Sender {
 	return &Sender{Client: client, QueueURL: &queueURL, DelaySeconds: delaySeconds}
 }
 
-func (p *Sender) Send(ctx context.Context, data []byte, messageAttributes map[string]string) (string, error) {
-	attributes := MapToAttributes(messageAttributes)
+func (p *Sender) Send(ctx context.Context, data []byte, attributes map[string]string) (string, error) {
+	attrs := MapToAttributes(attributes)
 	s := string(data)
 	result, err := p.Client.SendMessage(&sqs.SendMessageInput{
 		DelaySeconds:      p.DelaySeconds,
-		MessageAttributes: attributes,
+		MessageAttributes: attrs,
 		MessageBody:       aws.String(s),
 		QueueUrl:          p.QueueURL,
 	})