query-orchestration/api/docTextRunner/runner.go

package doctextrunner

import (
	"context"
	"encoding/json"
	"log/slog"

	documenttext "queryorchestration/internal/document/text"

	"github.com/go-playground/validator/v10"
	"github.com/google/uuid"

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

const Name = "docTextRunner"

type Services struct {
	Text *documenttext.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 {
	ID uuid.UUID `json:"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.Text.Extract(ctx, body.ID)
	if err != nil {
		slog.Error("unable to process", "error", err)
		return false
	}

	return true
}