Michael McGuinness
64 lines
1.3 KiB
Go
64 lines
1.3 KiB
Go
package queryrunner
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"log/slog"
|
|
|
|
resultset "queryorchestration/internal/query/result/set"
|
|
|
|
"github.com/go-playground/validator/v10"
|
|
"github.com/google/uuid"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/service/sqs/types"
|
|
)
|
|
|
|
const Name = "queryRunner"
|
|
|
|
type Services struct {
|
|
ResultSet *resultset.Service
|
|
}
|
|
|
|
type Runner struct {
|
|
validator *validator.Validate
|
|
svc *Services
|
|
}
|
|
|
|
func New(validator *validator.Validate, svc *Services) Runner {
|
|
return Runner{
|
|
validator: validator,
|
|
svc: svc,
|
|
}
|
|
}
|
|
|
|
type Body struct {
|
|
DocumentID uuid.UUID `json:"document_id" validate:"required,uuid"`
|
|
QueryID uuid.UUID `json:"query_id" validate:"required,uuid"`
|
|
}
|
|
|
|
func (s *Runner) Process(ctx context.Context, req *types.Message) bool {
|
|
var body Body
|
|
err := json.Unmarshal([]byte(*req.Body), &body)
|
|
if err != nil {
|
|
slog.Error("parsing body", "messageId", *req.MessageId, "body", *req.Body)
|
|
return true
|
|
}
|
|
|
|
err = s.validator.Struct(body)
|
|
if err != nil {
|
|
slog.Error("invalid body", "messageId", *req.MessageId, "error", err)
|
|
return true
|
|
}
|
|
|
|
err = s.svc.ResultSet.Set(ctx, &resultset.Set{
|
|
DocumentID: body.DocumentID,
|
|
QueryID: body.QueryID,
|
|
})
|
|
if err != nil {
|
|
slog.Error("unable to process", "error", err)
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|