Function GetQueueAttributes

pkg/dequeuer/queue_attributes.go:37–74

(client *awslib.Client, queueURL string)

}

func GetQueueAttributes(client *awslib.Client, queueURL string) (QueueAttributes, error) {
	result, err := client.SQS().GetQueueAttributes(
		&sqs.GetQueueAttributesInput{
			QueueUrl:       aws.String(queueURL),
			AttributeNames: aws.StringSlice([]string{"All"}),
		},
	)
	if err != nil {
		return QueueAttributes{}, errors.WithStack(err)
	}

	attributes := aws.StringValueMap(result.Attributes)

	var visibleCount int
	var notVisibleCount int
	var hasRedrivePolicy bool
	if val, found := attributes["ApproximateNumberOfMessages"]; found {
		count, ok := s.ParseInt(val)
		if ok {
			visibleCount = count
		}
	}

	if val, found := attributes["ApproximateNumberOfMessagesNotVisible"]; found {
		count, ok := s.ParseInt(val)
		if ok {
			notVisibleCount = count
		}
	}

	_, hasRedrivePolicy = attributes["RedrivePolicy"]

	return QueueAttributes{
		VisibleMessages:   visibleCount,
		InvisibleMessages: notVisibleCount,
		HasRedrivePolicy:  hasRedrivePolicy,
	}, nil
}