common

type AQLQuery struct {
	// Name of the main table.
	Table string `json:"table"`

	// Shards of the query
	// If empty then all shards of the table
	// owned by they host will be queried
	Shards []int `json:"shards"`

	// Foreign tables to be joined.
	Joins []Join `json:"joins,omitempty"`

	// Dimensions to group by on.
	Dimensions []Dimension `json:"dimensions,omitempty"`

	// Measures/metrics to report.
	Measures []Measure `json:"measures"`

	// Row level filters to apply for all measures. The filters are ANDed together.
	Filters       []string    `json:"rowFilters,omitempty"`
	FiltersParsed []expr.Expr `json:"-"`

	// Syntax sugar for specifying a time based range filter.
	TimeFilter TimeFilter `json:"timeFilter,omitempty"`

	// Additional supporting dimensions, these dimensions will not be grouped by,
	// but they may be referenced in Dimensions, Measures, SupportingDimensions and SupportingMeasures.
	SupportingDimensions []Dimension `json:"supportingDimensions,omitempty"`
	// Additional supporting measures, these measures will not be reported,
	// but they may be referenced in Measures and SupportingMeasures.
	SupportingMeasures []Measure `json:"supportingMeasures,omitempty"`

	// Timezone to use when converting timestamp to calendar time, specified as:
	//   - -8:00
	//   - GMT
	//   - America/Los_Angeles
	//   - timezone(city_id)
	//   - region_timezone(city_id)
	//   - mega_region_timezone(city_id)
	//   - sub_region_timezone(city_id)
	//   - country_timezone(city_id)
	Timezone string `json:"timezone,omitempty"`

	// This overrides "now" (in seconds)
	Now int64 `json:"now,omitempty"`

	// Limit is the max number of rows need to be return, and only used for non-aggregation
	Limit int `json:"limit,omitempty"`

	Sorts []SortField `json:"sorts,omitempty" yaml:"sorts"`

	// SQLQuery
	SQLQuery string `json:"sql,omitempty"`
}

type AQLQueryResult map[string]interface{}

type AQLRequest struct {
	Queries []AQLQuery `json:"queries"`
}

type AQLResponse struct {
	Results      []AQLQueryResult `json:"results"`
	Errors       []error          `json:"errors,omitempty"`
	QueryContext []string         `json:"context,omitempty"`
}

type AlignedTime struct {
	Time time.Time `json:"time"`
	// Values for unit: y, q, M, w, d, {12, 8, 6, 4, 3, 2}h, h, {30, 20, 15, 12, 10, 6, 5, 4, 3, 2}m, m
	Unit string `json:"unit"`
}

type DimCountsPerDimWidth [5]uint8

type Dimension struct {
	// Alias/name of the dimension, to be referenced by other dimensions and measures.
	Alias string `json:"alias,omitempty"`
	// The SQL expression for computing the dimension.
	// Expr can be empty when TimeBucketizer is specified, which implies the
	// designated time column from the main table is used as the expresssion.
	Expr       string    `json:"sqlExpression"`
	ExprParsed expr.Expr `json:"-"`

	// Decides how to bucketize a timestamp Dimension before grouping by.
	// See https://github.com/uber/aresdb/wiki/aql#time_bucketizer
	TimeBucketizer string `json:"timeBucketizer,omitempty"`

	TimeUnit string `json:"timeUnit,omitempty"`

	// Bucketizes numeric dimensions for integers and floating point numbers.
	NumericBucketizer NumericBucketizerDef `json:"numericBucketizer,omitempty"`
}

type HLL struct {
	SparseData       []HLLRegister // Unsorted registers.
	DenseData        []byte        // Rho by register index.
	NonZeroRegisters uint16
}

type HLLData struct {
	NumDimsPerDimWidth             DimCountsPerDimWidth
	ResultSize                     uint32
	PaddedRawDimValuesVectorLength uint32
	PaddedHLLVectorLength          int64

	DimIndexes []int
	DataTypes  []memCom.DataType
	// map from dimension index => enum cases. It will
	// only include columns used in dimensions.
	EnumDicts map[int][]string
}

type HLLDataWriter struct {
	HLLData
	Buffer []byte
}

type HLLQueryResults struct {
	// contains filtered or unexported fields
}

type HLLRegister struct {
	Index uint16 `json:"index"`
	Rho   byte   `json:"rho"`
}

type Join struct {
	// Name of the table to join against.
	Table string `json:"table"`

	// Alias for the table. Empty means the table name will be used as alias.
	Alias string `json:"alias"`

	// Condition expressions to be ANDed together for the join.
	Conditions       []string    `json:"conditions"`
	ConditionsParsed []expr.Expr `json:"-"`
}

type Measure struct {
	// Alias/name of the measure, to be referenced by other (derived) measures.
	Alias string `json:"alias,omitempty"`
	// The SQL expression for computing the measure.
	Expr       string    `json:"sqlExpression"`
	ExprParsed expr.Expr `json:"-"`

	// Row level filters to apply for this measure.
	// The filters are ANDed togther.
	Filters       []string    `json:"rowFilters,omitempty"`
	FiltersParsed []expr.Expr `json:"-"`
}

type NumericBucketizerDef struct {

	// Generates equal-width buckets. BucketWidth should be positive.
	// The generated buckets are:
	// ... [-2w, -w), [-w, 0), [0, w), [w, 2w) ...
	BucketWidth float64 `json:"bucketWidth,omitempty"`

	// Generates exponential/log buckets. LogBase should be positive.
	// The generated buckets are:
	// ... [pow(b, -2), pow(b, -1)), [pow(b, -1), 1), [1, pow(b, 1)), [pow(b, 1), pow(b, 2)) ...
	LogBase float64 `json:"logBase,omitempty"`

	// Generates a fixed number of buckets using the specified partitions.
	// The numbers should be in sorted order. The generated buckets are:
	// [-inf, p0), [p0, p1), [p1, p2), ... [pn-1, inf)
	ManualPartitions []float64 `json:"manualPartitions,omitempty"`
}

type SortField struct {
	// Name or alias of the field
	Name string `json:"name"`

	// Order the column, will be asc or desc
	Order string `json:"order"`
}

type TimeDimensionMeta struct {
	TimeBucketizer  string
	TimeUnit        string
	IsTimezoneTable bool
	TimeZone        *time.Location
	DSTSwitchTs     int64
	FromOffset      int
	ToOffset        int
}

type TimeFilter struct {
	// A table time column in the format of column, or table_alias.column.
	// When empty, it defaults to the designated time column of the main table.
	Column string `json:"column"`

	// The time specified in from and to are both inclusive.
	// See https://github.com/uber/aresdb/wiki/aql#time_filter
	From string `json:"from"`
	To   string `json:"to"`
}

type TimeSeriesBucketizer struct {
	Size int
	Unit string
}