Michael McGuinness
47 lines
1.0 KiB
Go
47 lines
1.0 KiB
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.Query
|
|
collectorQueries *[]query.Query
|
|
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.Query {
|
|
return *q.unsyncedQueue
|
|
}
|