forked from grafana-cold-storage/metrictank
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathfunc_divideseries.go
More file actions
73 lines (65 loc) · 1.67 KB
/
Copy pathfunc_divideseries.go
File metadata and controls
73 lines (65 loc) · 1.67 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package expr
import (
"errors"
"fmt"
"math"
"github.com/raintank/metrictank/api/models"
"gopkg.in/raintank/schema.v1"
)
type FuncDivideSeries struct {
dividend GraphiteFunc
divisor GraphiteFunc
}
func NewDivideSeries() GraphiteFunc {
return &FuncDivideSeries{}
}
func (s *FuncDivideSeries) Signature() ([]Arg, []Arg) {
return []Arg{
ArgSeriesList{val: &s.dividend},
ArgSeries{val: &s.divisor},
}, []Arg{ArgSeries{}}
}
func (s *FuncDivideSeries) Context(context Context) Context {
return context
}
func (s *FuncDivideSeries) Exec(cache map[Req][]models.Series) ([]models.Series, error) {
dividends, err := s.dividend.Exec(cache)
if err != nil {
return nil, err
}
divisors, err := s.divisor.Exec(cache)
if err != nil {
return nil, err
}
if len(divisors) != 1 {
return nil, errors.New(fmt.Sprintf("need 1 divisor series, not %d", len(divisors)))
}
divisor := divisors[0]
var series []models.Series
for _, dividend := range dividends {
out := pointSlicePool.Get().([]schema.Point)
for i := 0; i < len(dividend.Datapoints); i++ {
p := schema.Point{
Ts: dividend.Datapoints[i].Ts,
}
if divisor.Datapoints[i].Val == 0 {
p.Val = math.NaN()
} else {
p.Val = dividend.Datapoints[i].Val / divisor.Datapoints[i].Val
}
out = append(out, p)
}
name := fmt.Sprintf("divideSeries(%s,%s)", dividend.QueryPatt, divisor.QueryPatt)
output := models.Series{
Target: name,
QueryPatt: name,
Datapoints: out,
Interval: divisor.Interval,
Consolidator: dividend.Consolidator,
QueryCons: dividend.QueryCons,
}
cache[Req{}] = append(cache[Req{}], output)
series = append(series, output)
}
return series, nil
}