Create Queues in AWS SQS using Go

development golang aws sqs

In order to work with AWS SQS using Go, we will use github.com/aws/aws-sdk-go-v2. In this gist, we will use the aws-sdk-go-v2 package to create queues in AWS SQS. The complete source code for the snippet is available here https://github.com/abvarun226/blog-source-code/tree/master/create-queues-in-aws-sqs-using-go

In order to test the snippet, we will use localstack to run SQS emulator in a docker container on the laptop. The container exposes SQS endpoint on port 4566. The snippet will then create a new queue first-queue in SQS.

In order to start the localstack container, you will need to first install localstack:


$ pip install localstack

Localstack provides a fully functional local AWS cloud stack. It is a cloud service emulator that runs in a single container on a laptop or a development machine.

In order to start the AWS services, run:


$ localstack start -d

You can check the status of the services by running:


$ localstack status services


package main

import (
	"context"
	"flag"
	"log"

	"github.com/aws/aws-sdk-go-v2/aws"
	"github.com/aws/aws-sdk-go-v2/config"
	"github.com/aws/aws-sdk-go-v2/service/sqs"
)

type SQSQueueAPI interface {
	CreateQueue(ctx context.Context,
		params *sqs.CreateQueueInput,
		optFns ...func(*sqs.Options)) (*sqs.CreateQueueOutput, error)
}

func CreateQueue(c context.Context, api SQSQueueAPI, input *sqs.CreateQueueInput) (*sqs.CreateQueueOutput, error) {
	return api.CreateQueue(c, input)
}

// creates an sqs client.
func client(ctx context.Context, awsURL, region string) *sqs.Client {
	// customResolver is required here since we use localstack and need to point the aws url to localhost.
	customResolver := aws.EndpointResolverWithOptionsFunc(func(service, region string, options ...interface{}) (aws.Endpoint, error) {
		return aws.Endpoint{
			PartitionID:   "aws",
			URL:           awsURL,
			SigningRegion: region,
		}, nil

	})

	// load the default aws config along with custom resolver.
	cfg, err := config.LoadDefaultConfig(ctx, config.WithEndpointResolverWithOptions(customResolver))
	if err != nil {
		log.Fatalf("configuration error: %v", err)
	}

	return sqs.NewFromConfig(cfg)
}

// create a queue with the given name and attribute.
func createQueue(ctx context.Context, c *sqs.Client, queue *string, attr map[string]string) {
	input := &sqs.CreateQueueInput{
		QueueName:  queue,
		Attributes: attr,
	}

	result, err := CreateQueue(ctx, c, input)
	if err != nil {
		log.Printf("error creating the queue: %v", err)
		return
	}

	log.Printf("Queue URL: %s", *result.QueueUrl)
}

func main() {
	ctx := context.TODO()

	// name of the queue as a command line option.
	queue := flag.String("q", "", "name of the queue")
	flag.Parse()

	// queue cannot be empty string.
	if *queue == "" {
		log.Println("-q argument is required. Specify a name for the queue")
		return
	}

	awsURL := "http://127.0.0.1:4566"
	awsRegion := "us-west-2"

	// create aws client.
	c := client(ctx, awsURL, awsRegion)

	// queue attributes.
	queueAttributes := map[string]string{
		"DelaySeconds":           "60",
		"MessageRetentionPeriod": "86400",
	}

	// create a queue with the given name.
	createQueue(ctx, c, queue, queueAttributes)
}


Output:

export AWS_SECRET_ACCESS_KEY=foobar
➜  export AWS_ACCESS_KEY_ID=foobar

➜  go build -o sqs .

➜  ./sqs -q first-queue
2022/09/26 22:46:50 Queue URL: http://127.0.0.1:4566/000000000000/first-queue

Get new posts by email

Comments

comments powered by Disqus