11package checkawssqsqueuesize
22
33import (
4+ "context"
45 "errors"
56 "fmt"
67 "os"
78 "strconv"
89
9- "github.com/aws/aws-sdk-go/aws"
10- "github.com/aws/aws-sdk-go/aws/credentials"
11- "github.com/aws/aws-sdk-go/aws/session"
12- "github.com/aws/aws-sdk-go/service/sqs"
10+ "github.com/aws/aws-sdk-go-v2/aws"
11+ "github.com/aws/aws-sdk-go-v2/config"
12+ "github.com/aws/aws-sdk-go-v2/credentials"
13+ "github.com/aws/aws-sdk-go-v2/service/sqs"
14+ "github.com/aws/aws-sdk-go-v2/service/sqs/types"
1315 "github.com/jessevdk/go-flags"
1416 "github.com/mackerelio/checkers"
1517)
@@ -32,51 +34,54 @@ var opts struct {
3234
3335const sqsAttributeOfQueueSize = "ApproximateNumberOfMessages"
3436
35- func createService (region , awsAccessKeyID , awsSecretAccessKey string ) (* sqs.SQS , error ) {
36- sess , err := session .NewSession ()
37- if err != nil {
38- return nil , err
39- }
40-
41- config := aws .NewConfig ()
37+ func createService (ctx context.Context , region , awsAccessKeyID , awsSecretAccessKey string ) (* sqs.Client , error ) {
38+ var opts []func (* config.LoadOptions ) error
4239 if awsAccessKeyID != "" && awsSecretAccessKey != "" {
43- config = config .WithCredentials (credentials .NewStaticCredentials (awsAccessKeyID , awsSecretAccessKey , "" ))
40+ opts = append ( opts , config .WithCredentialsProvider (credentials .NewStaticCredentialsProvider (awsAccessKeyID , awsSecretAccessKey , "" ) ))
4441 }
4542 if region != "" {
46- config = config .WithRegion (region )
43+ opts = append (opts , config .WithRegion (region ))
44+ }
45+
46+ cfg , err := config .LoadDefaultConfig (ctx , opts ... )
47+ if err != nil {
48+ return nil , err
4749 }
48- return sqs .New (sess , config ), nil
50+
51+ return sqs .NewFromConfig (cfg ), nil
4952}
5053
51- func getSqsQueueSize (region , awsAccessKeyID , awsSecretAccessKey , queueName string ) (int , error ) {
52- sqsClient , err := createService (region , awsAccessKeyID , awsSecretAccessKey )
54+ func getSqsQueueSize (ctx context. Context , region , awsAccessKeyID , awsSecretAccessKey , queueName string ) (int , error ) {
55+ sqsClient , err := createService (ctx , region , awsAccessKeyID , awsSecretAccessKey )
5356 if err != nil {
5457 return - 1 , err
5558 }
5659
5760 // Get queue url
58- q , err := sqsClient .GetQueueUrl (& sqs.GetQueueUrlInput {
61+ q , err := sqsClient .GetQueueUrl (ctx , & sqs.GetQueueUrlInput {
5962 QueueName : aws .String (queueName ),
6063 })
6164 if err != nil {
6265 return - 1 , err
6366 }
6467
6568 // Get queue attribute
66- attr , err := sqsClient .GetQueueAttributes (& sqs.GetQueueAttributesInput {
67- QueueUrl : q .QueueUrl ,
68- AttributeNames : []* string {aws .String (sqsAttributeOfQueueSize )},
69+ attr , err := sqsClient .GetQueueAttributes (ctx , & sqs.GetQueueAttributesInput {
70+ QueueUrl : q .QueueUrl ,
71+ AttributeNames : []types.QueueAttributeName {
72+ types .QueueAttributeNameApproximateNumberOfMessages ,
73+ },
6974 })
7075 if err != nil {
7176 return - 1 , err
7277 }
7378
7479 // Queue size
7580 sizeStr , ok := attr .Attributes [sqsAttributeOfQueueSize ]
76- if ! ok || sizeStr == nil {
81+ if ! ok {
7782 return - 1 , errors .New ("attribute not found" )
7883 }
79- size , err := strconv .Atoi (* sizeStr )
84+ size , err := strconv .Atoi (sizeStr )
8085 if err != nil {
8186 return - 1 , err
8287 }
@@ -85,12 +90,14 @@ func getSqsQueueSize(region, awsAccessKeyID, awsSecretAccessKey, queueName strin
8590}
8691
8792func run (args []string ) * checkers.Checker {
93+ ctx := context .Background ()
94+
8895 _ , err := flags .ParseArgs (& opts , args )
8996 if err != nil {
9097 os .Exit (1 )
9198 }
9299
93- size , err := getSqsQueueSize (opts .Region , opts .AccessKeyID , opts .SecretAccessKey , opts .QueueName )
100+ size , err := getSqsQueueSize (ctx , opts .Region , opts .AccessKeyID , opts .SecretAccessKey , opts .QueueName )
94101 if err != nil {
95102 return checkers .NewChecker (checkers .UNKNOWN , err .Error ())
96103 }
0 commit comments