query-orchestration/internal/query/queue/result.go

package queryqueue

import (
	"context"
	"fmt"
	"queryorchestration/internal/database"
	"queryorchestration/internal/database/repository"
	queryprocessor "queryorchestration/internal/query/processor"
	"queryorchestration/internal/query/result"
	contextfull "queryorchestration/internal/query/types/contextFull"
	jsonextractor "queryorchestration/internal/query/types/jsonExtractor"
)

func (q *Queue) setResult(ctx context.Context, qu *queryprocessor.Query, resultValues *[]result.Value) error {
	processor, err := q.getProcessor(qu.Type)
	if err != nil {
		return err
	}

	value, err := processor.Process(ctx, qu, resultValues)
	if err != nil {
		return err
	}

	id, err := result.Store(ctx, q.db.Queries, &result.ResultStore{
		QueryID:      qu.ID,
		DocumentID:   q.documentId,
		Value:        value,
		CleanVersion: q.cleanVersion,
		TextVersion:  q.textVersion,
		QueryVersion: qu.Version,
	})
	if err != nil {
		return err
	}

	q.results = append(q.results, &result.Result{
		ID:           id,
		QueryID:      qu.ID,
		QueryVersion: qu.Version,
	})

	return nil
}

func (q *Queue) getProcessor(queryType queryprocessor.Type) (queryprocessor.Processor, error) {
	switch queryType {
	case queryprocessor.TypeJsonExtractor:
		return jsonextractor.NewExtractor(q.db), nil
	case queryprocessor.TypeContextFull:
		return contextfull.NewExtractor(), nil
	default:
		return nil, fmt.Errorf("attempting to process invalid query type")
	}
}

func (q *Queue) getResultValue(res *repository.ListResultValuesByIDRow) (result.Value, error) {
	var queryType queryprocessor.Type
	for _, qu := range q.collectorQueries {
		if qu.ID == database.MustToUUID(res.Queryid) {
			queryType = qu.Type
		}
	}

	switch queryType {
	case queryprocessor.TypeJsonExtractor:
		return jsonextractor.NewResult(res.Value), nil
	case queryprocessor.TypeContextFull:
		return contextfull.NewResult(res.Value), nil
	default:
		return nil, fmt.Errorf("attempting to process invalid query type")
	}
}