37 lines
1.1 KiB
Go
37 lines
1.1 KiB
Go
package elasticsearch
|
|
|
|
import (
|
|
"fmt"
|
|
)
|
|
|
|
// isQueryWithError validates the query and returns an error if invalid
|
|
func isQueryWithError(query *Query) error {
|
|
if len(query.BucketAggs) == 0 {
|
|
// If no aggregations, only document and logs queries are valid
|
|
if len(query.Metrics) == 0 || (!isLogsQuery(query) && !isDocumentQuery(query)) {
|
|
return fmt.Errorf("invalid query, missing metrics and aggregations")
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// isLogsQuery checks if the query is a logs query
|
|
func isLogsQuery(query *Query) bool {
|
|
return len(query.Metrics) > 0 && query.Metrics[0].Type == logsType
|
|
}
|
|
|
|
// isDocumentQuery checks if the query is a document query (raw_data or raw_document)
|
|
func isDocumentQuery(query *Query) bool {
|
|
return isRawDataQuery(query) || isRawDocumentQuery(query)
|
|
}
|
|
|
|
// isRawDataQuery checks if the query is a raw_data query
|
|
func isRawDataQuery(query *Query) bool {
|
|
return len(query.Metrics) > 0 && query.Metrics[0].Type == rawDataType
|
|
}
|
|
|
|
// isRawDocumentQuery checks if the query is a raw_document query
|
|
func isRawDocumentQuery(query *Query) bool {
|
|
return len(query.Metrics) > 0 && query.Metrics[0].Type == rawDocumentType
|
|
}
|