prometheus/head.go

70 lines
1.6 KiB
Go
Raw Normal View History

2016-12-04 04:16:11 -08:00
package tsdb
import (
"math"
"sync"
"github.com/fabxc/tsdb/chunks"
)
// HeadBlock handles reads and writes of time series data within a time window.
type HeadBlock struct {
mtx sync.RWMutex
descs map[uint64][]*chunkDesc // labels hash to possible chunks descs
index *memIndex
2016-12-09 04:41:38 -08:00
samples uint64 // total samples in the block
highTimestamp int64 // highest timestamp of any sample
baseTimestamp int64 // all samples are strictly later
2016-12-04 04:16:11 -08:00
}
2016-12-09 04:41:38 -08:00
// NewHeadBlock creates a new empty head block.
func NewHeadBlock(baseTime int64) *HeadBlock {
return &HeadBlock{
2016-12-09 04:41:38 -08:00
descs: make(map[uint64][]*chunkDesc, 2048),
index: newMemIndex(),
baseTimestamp: baseTime,
}
}
2016-12-04 04:16:11 -08:00
// get retrieves the chunk with the hash and label set and creates
// a new one if it doesn't exist yet.
2016-12-09 01:00:14 -08:00
func (h *HeadBlock) get(hash uint64, lset Labels) *chunkDesc {
2016-12-04 04:16:11 -08:00
cds := h.descs[hash]
for _, cd := range cds {
if cd.lset.Equals(lset) {
2016-12-09 01:00:14 -08:00
return cd
2016-12-04 04:16:11 -08:00
}
}
// None of the given chunks was for the series, create a new one.
cd := &chunkDesc{
lset: lset,
chunk: chunks.NewXORChunk(int(math.MaxInt64)),
}
h.index.add(cd)
2016-12-04 04:16:11 -08:00
h.descs[hash] = append(cds, cd)
2016-12-09 01:00:14 -08:00
return cd
2016-12-04 04:16:11 -08:00
}
2016-12-09 01:00:14 -08:00
// append adds the sample to the headblock.
func (h *HeadBlock) append(hash uint64, lset Labels, ts int64, v float64) error {
if err := h.get(hash, lset).append(ts, v); err != nil {
return err
}
h.samples++
return nil
2016-12-04 04:16:11 -08:00
}
type blockStats struct {
chunks uint32
samples uint64
}
func (h *HeadBlock) stats() *blockStats {
return &blockStats{
2016-12-09 04:41:38 -08:00
chunks: uint32(h.index.numSeries()),
2016-12-09 01:00:14 -08:00
samples: h.samples,
}
}