2024-05-15 02:55:44 +00:00
|
|
|
package logstorage
|
|
|
|
|
|
|
|
import (
|
|
|
|
"strconv"
|
|
|
|
)
|
|
|
|
|
|
|
|
type statsSumLen struct {
|
2024-05-22 19:01:20 +00:00
|
|
|
fields []string
|
2024-05-15 02:55:44 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (ss *statsSumLen) String() string {
|
2024-05-22 19:01:20 +00:00
|
|
|
return "sum_len(" + statsFuncFieldsToString(ss.fields) + ")"
|
2024-05-15 02:55:44 +00:00
|
|
|
}
|
|
|
|
|
2024-05-20 02:08:30 +00:00
|
|
|
func (ss *statsSumLen) updateNeededFields(neededFields fieldsSet) {
|
2024-05-22 19:01:20 +00:00
|
|
|
updateNeededFieldsForStatsFunc(neededFields, ss.fields)
|
2024-05-15 02:55:44 +00:00
|
|
|
}
|
|
|
|
|
2024-12-22 01:09:36 +00:00
|
|
|
func (ss *statsSumLen) newStatsProcessor(a *chunkedAllocator) statsProcessor {
|
2025-01-13 03:48:19 +00:00
|
|
|
return a.newStatsSumLenProcessor()
|
2024-05-15 02:55:44 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
type statsSumLenProcessor struct {
|
|
|
|
sumLen uint64
|
|
|
|
}
|
|
|
|
|
2025-01-13 03:48:19 +00:00
|
|
|
func (ssp *statsSumLenProcessor) updateStatsForAllRows(sf statsFunc, br *blockResult) int {
|
|
|
|
ss := sf.(*statsSumLen)
|
|
|
|
fields := ss.fields
|
2024-05-22 19:01:20 +00:00
|
|
|
if len(fields) == 0 {
|
2024-05-15 02:55:44 +00:00
|
|
|
// Sum all the columns
|
|
|
|
for _, c := range br.getColumns() {
|
|
|
|
ssp.sumLen += c.sumLenValues(br)
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// Sum the requested columns
|
2024-05-22 19:01:20 +00:00
|
|
|
for _, field := range fields {
|
2024-05-15 02:55:44 +00:00
|
|
|
c := br.getColumnByName(field)
|
|
|
|
ssp.sumLen += c.sumLenValues(br)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
2025-01-13 03:48:19 +00:00
|
|
|
func (ssp *statsSumLenProcessor) updateStatsForRow(sf statsFunc, br *blockResult, rowIdx int) int {
|
|
|
|
ss := sf.(*statsSumLen)
|
|
|
|
fields := ss.fields
|
2024-05-22 19:01:20 +00:00
|
|
|
if len(fields) == 0 {
|
2024-05-15 02:55:44 +00:00
|
|
|
// Sum all the fields for the given row
|
|
|
|
for _, c := range br.getColumns() {
|
|
|
|
v := c.getValueAtRow(br, rowIdx)
|
|
|
|
ssp.sumLen += uint64(len(v))
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// Sum only the given fields for the given row
|
2024-05-22 19:01:20 +00:00
|
|
|
for _, field := range fields {
|
2024-05-15 02:55:44 +00:00
|
|
|
c := br.getColumnByName(field)
|
|
|
|
v := c.getValueAtRow(br, rowIdx)
|
|
|
|
ssp.sumLen += uint64(len(v))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
|
2025-02-14 14:30:03 +00:00
|
|
|
func (ssp *statsSumLenProcessor) mergeState(_ *chunkedAllocator, _ statsFunc, sfp statsProcessor) {
|
2024-05-15 02:55:44 +00:00
|
|
|
src := sfp.(*statsSumLenProcessor)
|
|
|
|
ssp.sumLen += src.sumLen
|
|
|
|
}
|
|
|
|
|
2025-01-13 03:48:19 +00:00
|
|
|
func (ssp *statsSumLenProcessor) finalizeStats(_ statsFunc, dst []byte, _ <-chan struct{}) []byte {
|
2024-12-17 14:16:03 +00:00
|
|
|
return strconv.AppendUint(dst, ssp.sumLen, 10)
|
2024-05-15 02:55:44 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func parseStatsSumLen(lex *lexer) (*statsSumLen, error) {
|
2024-05-22 19:01:20 +00:00
|
|
|
fields, err := parseStatsFuncFields(lex, "sum_len")
|
2024-05-15 02:55:44 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
ss := &statsSumLen{
|
2024-05-22 19:01:20 +00:00
|
|
|
fields: fields,
|
2024-05-15 02:55:44 +00:00
|
|
|
}
|
|
|
|
return ss, nil
|
|
|
|
}
|