query-orchestration/api/queryService/parse.go

package queryservice

import (
	"errors"
	"queryorchestration/internal/query"
	queryprocessor "queryorchestration/internal/query/processor"
)

func parseQueries(queries []*query.Query) ([]Query, error) {
	outQueries := make([]Query, len(queries))
	for index, query := range queries {
		qt, err := parseQuery(query)
		if err != nil {
			return nil, err
		}

		outQueries[index] = *qt
	}

	return outQueries, nil
}

func parseQuery(query *query.Query) (*Query, error) {
	requiredQueries := make([]string, len(query.RequiredQueryIDs))
	for index, id := range query.RequiredQueryIDs {
		requiredQueries[index] = id.String()
	}

	qt, err := parseQueryType(query.Type)
	if err != nil {
		return nil, err
	}

	q := &Query{
		Id:            query.ID.String(),
		Type:          qt,
		ActiveVersion: query.ActiveVersion,
		LatestVersion: query.LatestVersion,
	}

	if query.Config != "" {
		q.Config = &query.Config
	}

	if len(query.RequiredQueryIDs) > 0 {
		q.RequiredQueries = requiredQueries
	}

	return q, nil
}

func parseQueryType(qType queryprocessor.Type) (QueryType, error) {
	switch qType {
	case queryprocessor.TypeJsonExtractor:
		return JSONEXTRACTOR, nil
	case queryprocessor.TypeContextFull:
		return CONTEXTFULL, nil
	default:
		return "", errors.New("invalid query type")
	}
}

func parseSpecQueryType(qType QueryType) (queryprocessor.Type, error) {
	switch qType {
	case JSONEXTRACTOR:
		return queryprocessor.TypeJsonExtractor, nil
	case CONTEXTFULL:
		return queryprocessor.TypeContextFull, nil
	default:
		return queryprocessor.Type(-1), errors.New("invalid query type")
	}
}