Michael McGuinness
64 lines
1.4 KiB
Go
64 lines
1.4 KiB
Go
package documentsync
|
|
|
|
import (
|
|
"context"
|
|
"queryorchestration/internal/database"
|
|
"queryorchestration/internal/database/repository"
|
|
"queryorchestration/internal/job/collector"
|
|
queryqueue "queryorchestration/internal/query/queue"
|
|
"queryorchestration/internal/query/result"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
func (s *Service) Sync(ctx context.Context, doc *Document) error {
|
|
coll, err := s.svc.Collector.GetByJobID(ctx, doc.JobID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
results, err := s.getResults(ctx, doc.ID, coll)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
queue, err := queryqueue.New(ctx, &queryqueue.NewConfig{
|
|
DB: s.db,
|
|
Collector: coll,
|
|
Results: results,
|
|
DocumentID: doc.ID,
|
|
CleanVersion: doc.CleanVersion,
|
|
TextVersion: doc.TextVersion,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = queue.Execute(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *Service) getResults(ctx context.Context, id uuid.UUID, coll *collector.Collector) ([]*result.Result, error) {
|
|
docID := database.MustToDBUUID(id)
|
|
|
|
results, err := s.db.Queries.ListResultsByDocumentID(ctx, &repository.ListResultsByDocumentIDParams{
|
|
Documentid: docID,
|
|
Textversion: coll.MinTextVersion,
|
|
Cleanversion: coll.MinCleanVersion,
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
cleanResults := make([]*result.Result, len(results))
|
|
for index, dbResult := range results {
|
|
cleanResults[index] = result.Parse(dbResult)
|
|
}
|
|
|
|
return cleanResults, nil
|
|
}
|