Michael McGuinness
81e7223560
Text Extraction * bases * go * splitting * structure * movetoasync * movetoasync * settinguptrigger * reorder * storevent * standardisepollingvalidation * unittests * fixlint * fixlint * awscfg * generatesample * followthrough * tests * clena * store * externalidcleanup * clientid * local * baseunittests * putobjecttests * tests
36 lines
838 B
Go
36 lines
838 B
Go
package collector
|
|
|
|
import (
|
|
"context"
|
|
|
|
resultprocessor "queryorchestration/internal/query/result/processor"
|
|
)
|
|
|
|
func (s *Service) GetByClientID(ctx context.Context, clientID string) (*Collector, error) {
|
|
dbColl, err := s.cfg.GetDBQueries().GetCollectorByClientID(ctx, clientID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return parseDBCollector(dbColl)
|
|
}
|
|
|
|
func (s *Service) ListQueries(ctx context.Context, id string) ([]*resultprocessor.Query, error) {
|
|
queries, err := s.cfg.GetDBQueries().ListCollectorQueries(ctx, id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
cleanQueries := make([]*resultprocessor.Query, len(queries))
|
|
for index, dbQuery := range queries {
|
|
cleanQuery, err := resultprocessor.ParseDBCollectorQuery(dbQuery)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
cleanQueries[index] = cleanQuery
|
|
}
|
|
|
|
return cleanQueries, nil
|
|
}
|