Michael McGuinness
68 lines
1.8 KiB
Go
68 lines
1.8 KiB
Go
package queryservice
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
|
|
collectorupdate "queryorchestration/internal/job/collector/update"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/labstack/echo/v4"
|
|
"github.com/oapi-codegen/runtime/types"
|
|
)
|
|
|
|
func (s *Controllers) GetJobCollectorByJobId(ctx echo.Context, jobId types.UUID) error {
|
|
coll, err := s.svc.Collector.GetByJobID(ctx.Request().Context(), jobId)
|
|
if err != nil {
|
|
return echo.NewHTTPError(http.StatusBadRequest, fmt.Sprintf("Unable to get collector: %s", err))
|
|
}
|
|
|
|
fields := make([]JobCollectorField, len(coll.Fields))
|
|
index := 0
|
|
for name, id := range coll.Fields {
|
|
fields[index] = JobCollectorField{
|
|
Name: name,
|
|
QueryId: id,
|
|
}
|
|
index++
|
|
}
|
|
|
|
return ctx.JSON(http.StatusOK, JobCollector{
|
|
JobId: coll.JobID,
|
|
MinimumCleanerVersion: coll.MinCleanVersion,
|
|
MinimumTextVersion: coll.MinTextVersion,
|
|
ActiveVersion: coll.ActiveVersion,
|
|
LatestVersion: coll.LatestVersion,
|
|
Fields: fields,
|
|
})
|
|
}
|
|
|
|
func (s *Controllers) UpdateJobCollectorByJobId(ctx echo.Context, jobId types.UUID) error {
|
|
req := JobCollectorUpdate{}
|
|
if err := ctx.Bind(&req); err != nil {
|
|
return echo.NewHTTPError(http.StatusBadRequest, err)
|
|
}
|
|
|
|
var fields *map[string]uuid.UUID
|
|
if req.Fields != nil {
|
|
fs := map[string]uuid.UUID{}
|
|
for _, field := range *req.Fields {
|
|
fs[field.Name] = field.QueryId
|
|
}
|
|
fields = &fs
|
|
}
|
|
|
|
err := s.svc.CollectorUpdate.UpdateByJobId(ctx.Request().Context(), &collectorupdate.UpdateParams{
|
|
JobID: jobId,
|
|
ActiveVersion: req.ActiveVersion,
|
|
MinCleanVersion: req.MinimumCleanerVersion,
|
|
MinTextVersion: req.MinimumTextVersion,
|
|
Fields: fields,
|
|
})
|
|
if err != nil {
|
|
return echo.NewHTTPError(http.StatusBadRequest, fmt.Sprintf("Unable to update query: %s", err))
|
|
}
|
|
|
|
return ctx.NoContent(http.StatusOK)
|
|
}
|