sqs_poller.go

package main

import (
	"database/sql"
	"encoding/json"
	"log"
	"time"

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

type MessageFormat struct {
	CheckHit bool `json:"check_hit"`
}

func main() {
	sess, err := session.NewSession(&aws.Config{
		Region: aws.String("us-east-1"),
		// other config here
	})
	if err != nil {
		log.Println("Error creating aws session", err)
		return
	}

	DB, err := sql.Open("postgres", "...connection string...")
	if err != nil {
		log.Println("Error connecting to database", err)
		return
	}

	queueService := sqs.New(sess)

	for {
		time.Sleep(10 * time.Second)

		resp, err := queueService.ReceiveMessage(&sqs.ReceiveMessageInput{
			QueueUrl:            aws.String(""),
			MaxNumberOfMessages: aws.Int64(1),
		})
		if err != nil {
			log.Println("Error receiving message", err)
			continue
		}

		for _, msg := range resp.Messages {
			var message MessageFormat
			err := json.Unmarshal([]byte(*msg.Body), &message)
			if err != nil {
				log.Println("Error parsing JSON", err)
				continue
			}

			_, err = DB.Exec("UPDATE checks SET state = ?", message.CheckHit)
			if err != nil {
				log.Println("Error updating check state", err)
				continue
			}

			_, err = queueService.DeleteMessage(&sqs.DeleteMessageInput{
				QueueUrl:      aws.String(""),
				ReceiptHandle: msg.ReceiptHandle,
			})

			if err != nil {
				log.Println("Error deleting message", err)
				continue
			}
		}
	}
}