fixed and improved region detection within queue url (#403)

* fixed and improved region detection within queue url

* add debug log for queue-url region detectio n

Author: bwagner5

pkg/config/config.go

@@ -20,6 +20,7 @@ import (
 	"strconv"
 	"strings"
 
+	"github.com/aws/aws-sdk-go/aws/endpoints"
 	"github.com/aws/aws-sdk-go/aws/session"
 	"github.com/rs/zerolog/log"
 )
@@ -193,7 +194,8 @@ func ParseCliArgs() (config Config, err error) {
 		if config.AWSRegion != "" {
 			sess.Config.Region = &config.AWSRegion
 		} else if *sess.Config.Region == "" && config.QueueURL != "" {
-			config.AWSRegion = strings.Split(config.QueueURL, ".")[1]
+			config.AWSRegion = getRegionFromQueueURL(config.QueueURL)
+			log.Debug().Str("Retrieved AWS region from queue-url: \"%s\"", config.AWSRegion)
 			sess.Config.Region = &config.AWSRegion
 		} else {
 			config.AWSRegion = *sess.Config.Region
@@ -397,3 +399,14 @@ func isConfigProvided(cliArgName string, envVarName string) bool {
 	})
 	return cliArgProvided
 }
+
+func getRegionFromQueueURL(queueURL string) string {
+	for _, partition := range endpoints.DefaultPartitions() {
+		for regionID := range partition.Regions() {
+			if strings.Contains(queueURL, regionID) {
+				return regionID
+			}
+		}
+	}
+	return ""
+}