query-orchestration/internal/queryQueue/service.go

package queryQueue

import (
	"context"
	"queryorchestration/internal/collector"
	"queryorchestration/internal/database/repository"
	"queryorchestration/internal/query"
	"queryorchestration/internal/result"

	"github.com/google/uuid"
)

type Queue struct {
	unsyncedQueue    *[]query.QueryRow
	collectorQueries *[]query.QueryRow
	results          *[]result.Result
	collector        *collector.Collector
	db               *repository.Queries
	cleanVersion     int32
	textVersion      int32
	documentId       uuid.UUID
}

func New(ctx context.Context, db *repository.Queries, 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() []query.QueryRow {
	return *q.unsyncedQueue
}