From fab23737527824733d1664a84d5513d448f6a069 Mon Sep 17 00:00:00 2001 From: Tobias Guggenmos Date: Mon, 3 Feb 2020 17:15:07 +0100 Subject: [PATCH] Add everything the parser needs to build Signed-off-by: Tobias Guggenmos --- promql/engine.go | 13 -- promql/value.go | 321 ----------------------------------------------- 2 files changed, 334 deletions(-) delete mode 100644 promql/value.go diff --git a/promql/engine.go b/promql/engine.go index 723819006..c55742a40 100644 --- a/promql/engine.go +++ b/promql/engine.go @@ -2111,19 +2111,6 @@ func NewOriginContext(ctx context.Context, data map[string]interface{}) context. return context.WithValue(ctx, queryOrigin, data) } -// documentedType returns the internal type to the equivalent -// user facing terminology as defined in the documentation. -func documentedType(t ValueType) string { - switch t { - case "vector": - return "instant vector" - case "matrix": - return "range vector" - default: - return string(t) - } -} - func formatDate(t time.Time) string { return t.UTC().Format("2006-01-02T15:04:05.000Z07:00") } diff --git a/promql/value.go b/promql/value.go deleted file mode 100644 index 6993ee8b0..000000000 --- a/promql/value.go +++ /dev/null @@ -1,321 +0,0 @@ -// Copyright 2017 The Prometheus Authors -// Licensed under the Apache License, Version 2.0 (the "License"); -// you may not use this file except in compliance with the License. -// You may obtain a copy of the License at -// -// http://www.apache.org/licenses/LICENSE-2.0 -// -// Unless required by applicable law or agreed to in writing, software -// distributed under the License is distributed on an "AS IS" BASIS, -// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -// See the License for the specific language governing permissions and -// limitations under the License. - -package promql - -import ( - "encoding/json" - "fmt" - "strconv" - "strings" - - "github.com/pkg/errors" - - "github.com/prometheus/prometheus/pkg/labels" - "github.com/prometheus/prometheus/storage" -) - -// Value is a generic interface for values resulting from a query evaluation. -type Value interface { - Type() ValueType - String() string -} - -func (Matrix) Type() ValueType { return ValueTypeMatrix } -func (Vector) Type() ValueType { return ValueTypeVector } -func (Scalar) Type() ValueType { return ValueTypeScalar } -func (String) Type() ValueType { return ValueTypeString } - -// ValueType describes a type of a value. -type ValueType string - -// The valid value types. -const ( - ValueTypeNone = "none" - ValueTypeVector = "vector" - ValueTypeScalar = "scalar" - ValueTypeMatrix = "matrix" - ValueTypeString = "string" -) - -// String represents a string value. -type String struct { - T int64 - V string -} - -func (s String) String() string { - return s.V -} - -func (s String) MarshalJSON() ([]byte, error) { - return json.Marshal([...]interface{}{float64(s.T) / 1000, s.V}) -} - -// Scalar is a data point that's explicitly not associated with a metric. -type Scalar struct { - T int64 - V float64 -} - -func (s Scalar) String() string { - v := strconv.FormatFloat(s.V, 'f', -1, 64) - return fmt.Sprintf("scalar: %v @[%v]", v, s.T) -} - -func (s Scalar) MarshalJSON() ([]byte, error) { - v := strconv.FormatFloat(s.V, 'f', -1, 64) - return json.Marshal([...]interface{}{float64(s.T) / 1000, v}) -} - -// Series is a stream of data points belonging to a metric. -type Series struct { - Metric labels.Labels `json:"metric"` - Points []Point `json:"values"` -} - -func (s Series) String() string { - vals := make([]string, len(s.Points)) - for i, v := range s.Points { - vals[i] = v.String() - } - return fmt.Sprintf("%s =>\n%s", s.Metric, strings.Join(vals, "\n")) -} - -// Point represents a single data point for a given timestamp. -type Point struct { - T int64 - V float64 -} - -func (p Point) String() string { - v := strconv.FormatFloat(p.V, 'f', -1, 64) - return fmt.Sprintf("%v @[%v]", v, p.T) -} - -// MarshalJSON implements json.Marshaler. -func (p Point) MarshalJSON() ([]byte, error) { - v := strconv.FormatFloat(p.V, 'f', -1, 64) - return json.Marshal([...]interface{}{float64(p.T) / 1000, v}) -} - -// Sample is a single sample belonging to a metric. -type Sample struct { - Point - - Metric labels.Labels -} - -func (s Sample) String() string { - return fmt.Sprintf("%s => %s", s.Metric, s.Point) -} - -func (s Sample) MarshalJSON() ([]byte, error) { - v := struct { - M labels.Labels `json:"metric"` - V Point `json:"value"` - }{ - M: s.Metric, - V: s.Point, - } - return json.Marshal(v) -} - -// Vector is basically only an alias for model.Samples, but the -// contract is that in a Vector, all Samples have the same timestamp. -type Vector []Sample - -func (vec Vector) String() string { - entries := make([]string, len(vec)) - for i, s := range vec { - entries[i] = s.String() - } - return strings.Join(entries, "\n") -} - -// ContainsSameLabelset checks if a vector has samples with the same labelset -// Such a behavior is semantically undefined -// https://github.com/prometheus/prometheus/issues/4562 -func (vec Vector) ContainsSameLabelset() bool { - l := make(map[uint64]struct{}, len(vec)) - for _, s := range vec { - hash := s.Metric.Hash() - if _, ok := l[hash]; ok { - return true - } - l[hash] = struct{}{} - } - return false -} - -// Matrix is a slice of Series that implements sort.Interface and -// has a String method. -type Matrix []Series - -func (m Matrix) String() string { - // TODO(fabxc): sort, or can we rely on order from the querier? - strs := make([]string, len(m)) - - for i, ss := range m { - strs[i] = ss.String() - } - - return strings.Join(strs, "\n") -} - -// TotalSamples returns the total number of samples in the series within a matrix. -func (m Matrix) TotalSamples() int { - numSamples := 0 - for _, series := range m { - numSamples += len(series.Points) - } - return numSamples -} - -func (m Matrix) Len() int { return len(m) } -func (m Matrix) Less(i, j int) bool { return labels.Compare(m[i].Metric, m[j].Metric) < 0 } -func (m Matrix) Swap(i, j int) { m[i], m[j] = m[j], m[i] } - -// ContainsSameLabelset checks if a matrix has samples with the same labelset -// Such a behavior is semantically undefined -// https://github.com/prometheus/prometheus/issues/4562 -func (m Matrix) ContainsSameLabelset() bool { - l := make(map[uint64]struct{}, len(m)) - for _, ss := range m { - hash := ss.Metric.Hash() - if _, ok := l[hash]; ok { - return true - } - l[hash] = struct{}{} - } - return false -} - -// Result holds the resulting value of an execution or an error -// if any occurred. -type Result struct { - Err error - Value Value - Warnings storage.Warnings -} - -// Vector returns a Vector if the result value is one. An error is returned if -// the result was an error or the result value is not a Vector. -func (r *Result) Vector() (Vector, error) { - if r.Err != nil { - return nil, r.Err - } - v, ok := r.Value.(Vector) - if !ok { - return nil, errors.New("query result is not a Vector") - } - return v, nil -} - -// Matrix returns a Matrix. An error is returned if -// the result was an error or the result value is not a Matrix. -func (r *Result) Matrix() (Matrix, error) { - if r.Err != nil { - return nil, r.Err - } - v, ok := r.Value.(Matrix) - if !ok { - return nil, errors.New("query result is not a range Vector") - } - return v, nil -} - -// Scalar returns a Scalar value. An error is returned if -// the result was an error or the result value is not a Scalar. -func (r *Result) Scalar() (Scalar, error) { - if r.Err != nil { - return Scalar{}, r.Err - } - v, ok := r.Value.(Scalar) - if !ok { - return Scalar{}, errors.New("query result is not a Scalar") - } - return v, nil -} - -func (r *Result) String() string { - if r.Err != nil { - return r.Err.Error() - } - if r.Value == nil { - return "" - } - return r.Value.String() -} - -// StorageSeries simulates promql.Series as storage.Series. -type StorageSeries struct { - series Series -} - -// NewStorageSeries returns a StorageSeries from a Series. -func NewStorageSeries(series Series) *StorageSeries { - return &StorageSeries{ - series: series, - } -} - -func (ss *StorageSeries) Labels() labels.Labels { - return ss.series.Metric -} - -// Iterator returns a new iterator of the data of the series. -func (ss *StorageSeries) Iterator() storage.SeriesIterator { - return newStorageSeriesIterator(ss.series) -} - -type storageSeriesIterator struct { - points []Point - curr int -} - -func newStorageSeriesIterator(series Series) *storageSeriesIterator { - return &storageSeriesIterator{ - points: series.Points, - curr: -1, - } -} - -func (ssi *storageSeriesIterator) Seek(t int64) bool { - i := ssi.curr - if i < 0 { - i = 0 - } - for ; i < len(ssi.points); i++ { - if ssi.points[i].T >= t { - ssi.curr = i - return true - } - } - ssi.curr = len(ssi.points) - 1 - return false -} - -func (ssi *storageSeriesIterator) At() (t int64, v float64) { - p := ssi.points[ssi.curr] - return p.T, p.V -} - -func (ssi *storageSeriesIterator) Next() bool { - ssi.curr++ - return ssi.curr < len(ssi.points) -} - -func (ssi *storageSeriesIterator) Err() error { - return nil -}