dataquery

type PrometheusQuery struct {
	connection.PrometheusConnection `json:",inline" yaml:",inline"`

	// Query is the PromQL query string
	Query string `json:"query" yaml:"query" template:"true"`

	// Range runs a PromQL range query when specified
	Range *PrometheusRange `json:"range,omitempty" yaml:"range,omitempty" template:"true"`

	// MatchLabels is a list of labels, when provided, are included in the result.
	// Example:
	// If a query produces {a:1, b:2, c:3, d:4} value=30.0
	// then, with matchLabels = [a,b]
	// The result will only have {a:1, b:2} value=30.0
	//
	// This helps in have a deterministic schema for the query result.
	//
	// Deprecated: use SelectLabels
	MatchLabels []string `json:"matchLabels,omitempty" yaml:"matchLabels,omitempty"`

	// SelectLabels is a list of labels, when provided, are included in the result.
	// Example:
	// If a query produces {a:1, b:2, c:3, d:4} value=30.0
	// then, with matchLabels = [a,b]
	// The result will only have {a:1, b:2} value=30.0
	//
	// This helps in have a deterministic schema for the query result.
	SelectLabels []string `json:"selectLabels,omitempty"`
}

type PrometheusRange struct {
	Start string `json:"start" yaml:"start" template:"true"`
	End   string `json:"end" yaml:"end" template:"true"`
	Step  string `json:"step" yaml:"step" template:"true"`
}

type Query struct {
	// Prometheus queries metrics from Prometheus
	Prometheus *PrometheusQuery `json:"prometheus,omitempty" yaml:"prometheus,omitempty"`

	// SQL runs arbitrary SQL queries against a configured SQL connection
	SQL *SQLQuery `json:"sql,omitempty" yaml:"sql,omitempty"`
}

type QueryResultRow map[string]any

type QueryResultSet struct {
	Name    string
	Results []QueryResultRow

	// Map column name to column type
	ColumnDefs map[string]models.ColumnType
}

type SQLQuery struct {
	connection.SQLConnection `json:",inline" yaml:",inline"`

	// Query is the SQL query string to execute against the configured connection.
	Query string `json:"query" yaml:"query"`
}