Michael McGuinness
47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
package queryqueue
|
|
|
|
import (
|
|
"context"
|
|
"queryorchestration/internal/collector"
|
|
"queryorchestration/internal/database"
|
|
queryprocessor "queryorchestration/internal/queryProcessor"
|
|
"queryorchestration/internal/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
|
|
}
|