query-orchestration/api/queryAPI/collector.go

package queryapi

import (
	"fmt"
	"net/http"

	collectorset "queryorchestration/internal/collector/set"

	"github.com/google/uuid"
	"github.com/labstack/echo/v4"
)

func (s *Controllers) GetCollectorByClientId(ctx echo.Context, clientId ClientID) error {
	coll, err := s.svc.Collector.GetByClientID(ctx.Request().Context(), clientId)
	if err != nil {
		return echo.NewHTTPError(http.StatusBadRequest, fmt.Sprintf("Unable to get collector: %s", err))
	}

	fields := make([]CollectorField, len(coll.Fields))
	index := 0
	for name, id := range coll.Fields {
		fields[index] = CollectorField{
			Name:    name,
			QueryId: id,
		}
		index++
	}

	return ctx.JSON(http.StatusOK, Collector{
		ClientId:              clientId,
		MinimumCleanerVersion: coll.MinCleanVersion,
		MinimumTextVersion:    coll.MinTextVersion,
		ActiveVersion:         coll.ActiveVersion,
		LatestVersion:         coll.LatestVersion,
		Fields:                fields,
	})
}

func (s *Controllers) SetCollectorByClientId(ctx echo.Context, clientId ClientID) error {
	req := CollectorSet{}
	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.CollectorSet.SetByClientId(ctx.Request().Context(), clientId, &collectorset.SetParams{
		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)
}