query-orchestration/api/queryService/jobcollector.go

package queryservice

import (
	"fmt"
	"net/http"
	"queryorchestration/internal/job/collector"

	"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.StatusNotFound, 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.Collector.UpdateByJobId(ctx.Request().Context(), &collector.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)
}