-
Notifications
You must be signed in to change notification settings - Fork 1.1k
/
stats_sum.go
110 lines (96 loc) · 2.17 KB
/
stats_sum.go
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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
package logstorage
import (
"math"
"strconv"
"unsafe"
)
type statsSum struct {
fields []string
}
func (ss *statsSum) String() string {
return "sum(" + statsFuncFieldsToString(ss.fields) + ")"
}
func (ss *statsSum) updateNeededFields(neededFields fieldsSet) {
updateNeededFieldsForStatsFunc(neededFields, ss.fields)
}
func (ss *statsSum) newStatsProcessor() (statsProcessor, int) {
ssp := &statsSumProcessor{
ss: ss,
sum: nan,
}
return ssp, int(unsafe.Sizeof(*ssp))
}
type statsSumProcessor struct {
ss *statsSum
sum float64
}
func (ssp *statsSumProcessor) updateStatsForAllRows(br *blockResult) int {
fields := ssp.ss.fields
if len(fields) == 0 {
// Sum all the columns
for _, c := range br.getColumns() {
ssp.updateStateForColumn(br, c)
}
} else {
// Sum the requested columns
for _, field := range fields {
c := br.getColumnByName(field)
ssp.updateStateForColumn(br, c)
}
}
return 0
}
func (ssp *statsSumProcessor) updateStatsForRow(br *blockResult, rowIdx int) int {
fields := ssp.ss.fields
if len(fields) == 0 {
// Sum all the fields for the given row
for _, c := range br.getColumns() {
f, ok := c.getFloatValueAtRow(br, rowIdx)
if ok {
ssp.updateState(f)
}
}
} else {
// Sum only the given fields for the given row
for _, field := range fields {
c := br.getColumnByName(field)
f, ok := c.getFloatValueAtRow(br, rowIdx)
if ok {
ssp.updateState(f)
}
}
}
return 0
}
func (ssp *statsSumProcessor) updateStateForColumn(br *blockResult, c *blockResultColumn) {
f, count := c.sumValues(br)
if count > 0 {
ssp.updateState(f)
}
}
func (ssp *statsSumProcessor) updateState(f float64) {
if math.IsNaN(ssp.sum) {
ssp.sum = f
} else {
ssp.sum += f
}
}
func (ssp *statsSumProcessor) mergeState(sfp statsProcessor) {
src := sfp.(*statsSumProcessor)
if !math.IsNaN(src.sum) {
ssp.updateState(src.sum)
}
}
func (ssp *statsSumProcessor) finalizeStats() string {
return strconv.FormatFloat(ssp.sum, 'f', -1, 64)
}
func parseStatsSum(lex *lexer) (*statsSum, error) {
fields, err := parseStatsFuncFields(lex, "sum")
if err != nil {
return nil, err
}
ss := &statsSum{
fields: fields,
}
return ss, nil
}