Included inflight messages, take a look if you consider this as an enhancement

README.md

@@ -1,5 +1,5 @@
 # kube-sqs-autoscaler
-Kubernetes pod autoscaler based on queue size in AWS SQS. It periodically retrieves the number of messages in your queue and scales pods accordingly.
+Kubernetes pod autoscaler based on queue size in AWS SQS. It periodically retrieves the number of messages(Visible and inflight) in your queue and scales pods accordingly.
 
 ## Setting up
 Setting up kube-sqs-autoscaler requires two steps:
@@ -28,7 +28,7 @@ spec:
     spec:
       containers:
       - name: kube-sqs-autoscaler
-        image: wattpad/kube-sqs-autoscaler:v1.2.1
+        image: Build and add your tag here
         command:
           - /kube-sqs-autoscaler
           - --sqs-queue-url=https://sqs.your_aws_region.amazonaws.com/your_aws_account_number/your_queue_name  # required

main.go

@@ -6,8 +6,8 @@ import (
 
 	log "github.com/Sirupsen/logrus"
 
-	"github.com/Wattpad/kube-sqs-autoscaler/scale"
-	"github.com/Wattpad/kube-sqs-autoscaler/sqs"
+	"kube-sqs-autoscaler/scale"
+	"kube-sqs-autoscaler/sqs"
 )
 
 var (
@@ -33,7 +33,13 @@ func Run(p *scale.PodAutoScaler, sqs *sqs.SqsClient) {
 		select {
 		case <-time.After(pollInterval):
 			{
-				numMessages, err := sqs.NumMessages()
+				numViMessages, err := sqs.NumMessages()
+  		                log.Info("Total no. of visible messages: ",numViMessages)
+    		                numinMessages, err := sqs.NumInflightMessages()
+				log.Info("Total no. of inflight messages: ",numinMessages)
+				numMessages := numinMessages + numViMessages
+				log.Info("Total messages for processing: ",numMessages)
+
 				if err != nil {
 					log.Errorf("Failed to get SQS messages: %v", err)
 					continue

sqs/sqs.go

@@ -47,3 +47,26 @@ func (s *SqsClient) NumMessages() (int, error) {
 
 	return messages, nil
 }
+
+func (s *SqsClient) NumInflightMessages() (int, error) {
+	params := &sqs.GetQueueAttributesInput{
+		AttributeNames: []*string{aws.String("ApproximateNumberOfMessagesNotVisible")},
+		QueueUrl:       aws.String(s.QueueUrl),
+	}
+
+
+	out, err := s.Client.GetQueueAttributes(params)
+
+	if err != nil {
+		return 0, errors.Wrap(err, "Failed to get messages in SQS")
+	}
+
+
+	messages, err := strconv.Atoi(*out.Attributes["ApproximateNumberOfMessagesNotVisible"])
+
+	if err != nil {
+		return 0, errors.Wrap(err, "Failed to get number of messages in queue")
+	}
+
+    return messages, nil
+}