query-orchestration/internal/query/create.go

package query

import (
	"context"
	"fmt"
	contextfull "queryorchestration/internal/contextFull"
	"queryorchestration/internal/database"
	"queryorchestration/internal/database/repository"
	jsonextractor "queryorchestration/internal/jsonExtractor"
	queryprocessor "queryorchestration/internal/queryProcessor"

	"github.com/google/uuid"
	"github.com/jackc/pgx/v5/pgtype"
)

func (s *Service) Create(ctx context.Context, entity *queryprocessor.Create) (uuid.UUID, error) {
	validator, err := s.getCreator(entity.Type)
	if err != nil {
		return uuid.Nil, err
	}

	err = validator.Validate(ctx, entity)
	if err != nil {
		return uuid.Nil, err
	}

	id, err := s.submitCreate(ctx, entity)
	if err != nil {
		return uuid.Nil, err
	}

	return id, err
}

func (s *Service) submitCreate(ctx context.Context, entity *queryprocessor.Create) (uuid.UUID, error) {
	query, err := parseCreateQuery(entity)
	if err != nil {
		return uuid.Nil, err
	}

	tx, err := s.db.Pool.Begin(ctx)
	if err != nil {
		return uuid.Nil, err
	}
	defer func() {
		_ = tx.Rollback(ctx)
	}()

	qtx := s.db.Queries.WithTx(tx)

	dbID, err := qtx.CreateQuery(ctx, query.Type)
	if err != nil {
		return uuid.Nil, err
	}

	for _, reqQuery := range query.RequiredQueryIDs {
		err = qtx.CreateRequiredQuery(ctx, &repository.CreateRequiredQueryParams{
			Queryid:         dbID,
			Requiredqueryid: reqQuery,
			Addedversion:    1,
		})
		if err != nil {
			return uuid.Nil, err
		}
	}

	if query.Config != nil && string(query.Config) != "" {
		err = qtx.CreateQueryConfig(ctx, &repository.CreateQueryConfigParams{
			Queryid:      dbID,
			Config:       query.Config,
			Addedversion: 1,
		})
		if err != nil {
			return uuid.Nil, err
		}
	}

	err = tx.Commit(ctx)
	if err != nil {
		return uuid.Nil, err
	}

	id := database.MustToUUID(dbID)

	return id, nil

}

func (s *Service) getCreator(qType queryprocessor.Type) (queryprocessor.Creator, error) {
	switch qType {
	case queryprocessor.TypeJsonExtractor:
		return jsonextractor.NewCreator(s.db), nil
	case queryprocessor.TypeContextFull:
		return contextfull.NewCreator(s.db), nil
	default:
		return nil, fmt.Errorf("attempting to process invalid query type")
	}
}

type createQuery struct {
	Type             repository.Querytype
	RequiredQueryIDs []pgtype.UUID
	Config           []byte
}

func parseCreateQuery(q *queryprocessor.Create) (*createQuery, error) {
	t, err := queryprocessor.ToDBQueryType(q.Type)
	if err != nil {
		return nil, err
	}

	reqIDs := database.MustToDBUUIDArray(q.RequiredQueryIDs)

	return &createQuery{
		Type:             t,
		RequiredQueryIDs: reqIDs,
		Config:           []byte(q.Config),
	}, nil
}