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

package queryqueue

import (
	"context"
	"queryorchestration/internal/database"
	"queryorchestration/internal/job/collector"
	queryprocessor "queryorchestration/internal/query/processor"
	"queryorchestration/internal/query/result"

	"github.com/google/uuid"
)

type Queue struct {
	unsyncedQueue    []*queryprocessor.Query
	collectorQueries []*queryprocessor.Query
	results          []*result.Result
	collector        *collector.Collector
	db               *database.Connection
	cleanVersion     int32
	textVersion      int32
	documentId       uuid.UUID
}

func New(ctx context.Context, db *database.Connection, coll *collector.Collector, results []*result.Result, docId uuid.UUID, cleanVersion int32, textVersion int32) (*Queue, error) {
	queue := Queue{
		db:           db,
		results:      results,
		collector:    coll,
		documentId:   docId,
		cleanVersion: cleanVersion,
		textVersion:  textVersion,
	}

	err := queue.getCollectorQueries(ctx)
	if err != nil {
		return nil, err
	}

	queue.getUnsyncedQueries()

	return &queue, nil
}

func (q *Queue) GetQueue() []*queryprocessor.Query {
	return q.unsyncedQueue
}