query-orchestration/internal/queryQueue/result.go

package queryqueue

import (
	"context"
	"fmt"
	contextfull "queryorchestration/internal/contextFull"
	jsonextractor "queryorchestration/internal/jsonExtractor"
	queryprocessor "queryorchestration/internal/queryProcessor"
	"queryorchestration/internal/result"
)

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")
	}
}