Michael McGuinness
82 lines
2.1 KiB
Go
82 lines
2.1 KiB
Go
package queue
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log/slog"
|
|
awsc "queryorchestration/internal/serviceconfig/aws"
|
|
"time"
|
|
|
|
"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"
|
|
"github.com/aws/aws-sdk-go-v2/service/sqs/types"
|
|
)
|
|
|
|
type QueueConfig struct {
|
|
AWSEndpointUrlSQS string `env:"AWS_ENDPOINT_URL_SQS"`
|
|
QueueClient SQSClient
|
|
}
|
|
|
|
type ConfigProvider interface {
|
|
awsc.ConfigProvider
|
|
SetQueueClient(context.Context) error
|
|
GetQueueClient() SQSClient
|
|
PingQueueByURL(context.Context, string) error
|
|
SendToQueue(context.Context, *SendParams) error
|
|
ReceiveFromQueue(context.Context, *ReceiveParams) (*sqs.ReceiveMessageOutput, error)
|
|
DeleteFromQueue(context.Context, *DeleteParams) error
|
|
SetSQSEndpoint(string)
|
|
GetSQSEndpoint() string
|
|
}
|
|
|
|
func (c *QueueConfig) GetQueueClient() SQSClient {
|
|
return c.QueueClient
|
|
}
|
|
|
|
func (c *QueueConfig) GetSQSEndpoint() string {
|
|
return c.AWSEndpointUrlSQS
|
|
}
|
|
|
|
func (c *QueueConfig) PingQueueByURL(ctx context.Context, url string) error {
|
|
timeout := time.After(30 * time.Second)
|
|
tick := time.NewTicker(2 * time.Second)
|
|
defer tick.Stop()
|
|
|
|
input := &sqs.GetQueueAttributesInput{
|
|
QueueUrl: aws.String(url),
|
|
AttributeNames: []types.QueueAttributeName{types.QueueAttributeNameApproximateNumberOfMessages},
|
|
}
|
|
|
|
for {
|
|
select {
|
|
case <-timeout:
|
|
return fmt.Errorf("unable to ping queue by url: %s", url)
|
|
case <-tick.C:
|
|
r, err := c.QueueClient.GetQueueAttributes(ctx, input)
|
|
if err == nil {
|
|
slog.Info("Successful queue ping", "url", url, "approx_msgs", r.Attributes[string(types.QueueAttributeNameApproximateNumberOfMessages)])
|
|
return nil
|
|
}
|
|
slog.Debug("Attempted queue ping", "url", url, "address", c.GetSQSEndpoint())
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
}
|
|
}
|
|
}
|
|
|
|
func (c *QueueConfig) SetQueueClient(ctx context.Context) error {
|
|
cfg, err := config.LoadDefaultConfig(ctx)
|
|
if err != nil {
|
|
return fmt.Errorf("unable to load SDK config: %v", err)
|
|
}
|
|
|
|
c.QueueClient = sqs.NewFromConfig(cfg)
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c *QueueConfig) SetSQSEndpoint(e string) {
|
|
c.AWSEndpointUrlSQS = e
|
|
}
|