2017-04-10 18:59:45 +00:00
|
|
|
// 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.
|
|
|
|
|
2016-12-04 12:16:11 +00:00
|
|
|
package tsdb
|
|
|
|
|
|
|
|
import (
|
2017-01-16 13:18:32 +00:00
|
|
|
"fmt"
|
2017-01-04 13:06:40 +00:00
|
|
|
"math"
|
2017-01-13 15:14:40 +00:00
|
|
|
"math/rand"
|
2017-01-19 10:22:47 +00:00
|
|
|
"os"
|
2017-05-18 14:09:30 +00:00
|
|
|
"path/filepath"
|
2016-12-14 17:38:46 +00:00
|
|
|
"sort"
|
2016-12-04 12:16:11 +00:00
|
|
|
"sync"
|
2017-02-04 10:53:52 +00:00
|
|
|
"sync/atomic"
|
2017-01-06 14:18:06 +00:00
|
|
|
"time"
|
2016-12-04 12:16:11 +00:00
|
|
|
|
2017-05-17 14:43:01 +00:00
|
|
|
"encoding/binary"
|
|
|
|
|
2017-01-06 14:18:06 +00:00
|
|
|
"github.com/go-kit/kit/log"
|
2017-02-27 09:46:15 +00:00
|
|
|
"github.com/oklog/ulid"
|
2017-01-19 10:22:47 +00:00
|
|
|
"github.com/pkg/errors"
|
2017-04-04 09:27:26 +00:00
|
|
|
"github.com/prometheus/tsdb/chunks"
|
|
|
|
"github.com/prometheus/tsdb/labels"
|
2016-12-04 12:16:11 +00:00
|
|
|
)
|
|
|
|
|
2017-01-17 15:33:58 +00:00
|
|
|
var (
|
|
|
|
// ErrNotFound is returned if a looked up resource was not found.
|
2017-03-21 09:11:23 +00:00
|
|
|
ErrNotFound = errors.Errorf("not found")
|
2017-01-17 15:33:58 +00:00
|
|
|
|
|
|
|
// ErrOutOfOrderSample is returned if an appended sample has a
|
|
|
|
// timestamp larger than the most recent sample.
|
|
|
|
ErrOutOfOrderSample = errors.New("out of order sample")
|
|
|
|
|
|
|
|
// ErrAmendSample is returned if an appended sample has the same timestamp
|
|
|
|
// as the most recent sample but a different value.
|
|
|
|
ErrAmendSample = errors.New("amending sample")
|
|
|
|
|
|
|
|
// ErrOutOfBounds is returned if an appended sample is out of the
|
|
|
|
// writable time range.
|
|
|
|
ErrOutOfBounds = errors.New("out of bounds")
|
|
|
|
)
|
|
|
|
|
2017-05-12 14:34:41 +00:00
|
|
|
// HeadBlock handles reads and writes of time series data within a time window.
|
|
|
|
type HeadBlock struct {
|
2017-03-20 07:41:56 +00:00
|
|
|
mtx sync.RWMutex
|
|
|
|
dir string
|
2017-05-13 15:09:26 +00:00
|
|
|
wal WAL
|
2017-01-19 10:22:47 +00:00
|
|
|
|
2017-02-04 10:53:52 +00:00
|
|
|
activeWriters uint64
|
2017-06-25 17:02:02 +00:00
|
|
|
highTimestamp int64
|
2017-03-17 11:12:50 +00:00
|
|
|
closed bool
|
2017-02-04 10:53:52 +00:00
|
|
|
|
2016-12-22 00:12:28 +00:00
|
|
|
// descs holds all chunk descs for the head block. Each chunk implicitly
|
|
|
|
// is assigned the index as its ID.
|
2017-01-11 12:02:38 +00:00
|
|
|
series []*memSeries
|
2016-12-22 00:12:28 +00:00
|
|
|
// hashes contains a collision map of label set hashes of chunks
|
2016-12-31 09:19:02 +00:00
|
|
|
// to their chunk descs.
|
2017-01-11 12:02:38 +00:00
|
|
|
hashes map[uint64][]*memSeries
|
2016-12-22 00:12:28 +00:00
|
|
|
|
2017-08-05 11:31:48 +00:00
|
|
|
symbols map[string]struct{}
|
2016-12-22 00:12:28 +00:00
|
|
|
values map[string]stringset // label names to possible values
|
|
|
|
postings *memPostings // postings lists for terms
|
2016-12-07 16:10:49 +00:00
|
|
|
|
2017-05-24 05:54:24 +00:00
|
|
|
tombstones tombstoneReader
|
2017-05-15 17:58:14 +00:00
|
|
|
|
2017-03-23 17:27:20 +00:00
|
|
|
meta BlockMeta
|
2017-01-19 13:01:38 +00:00
|
|
|
}
|
|
|
|
|
2017-05-13 16:14:18 +00:00
|
|
|
// TouchHeadBlock atomically touches a new head block in dir for
|
|
|
|
// samples in the range [mint,maxt).
|
2017-05-18 14:09:30 +00:00
|
|
|
func TouchHeadBlock(dir string, mint, maxt int64) (string, error) {
|
2017-03-20 13:45:27 +00:00
|
|
|
entropy := rand.New(rand.NewSource(time.Now().UnixNano()))
|
|
|
|
|
2017-02-27 09:46:15 +00:00
|
|
|
ulid, err := ulid.New(ulid.Now(), entropy)
|
|
|
|
if err != nil {
|
2017-05-18 14:09:30 +00:00
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
|
|
|
|
// Make head block creation appear atomic.
|
|
|
|
dir = filepath.Join(dir, ulid.String())
|
|
|
|
tmp := dir + ".tmp"
|
|
|
|
|
|
|
|
if err := os.MkdirAll(tmp, 0777); err != nil {
|
|
|
|
return "", err
|
2017-01-19 13:01:38 +00:00
|
|
|
}
|
2016-12-22 11:05:24 +00:00
|
|
|
|
2017-03-02 08:13:29 +00:00
|
|
|
if err := writeMetaFile(tmp, &BlockMeta{
|
2017-05-18 14:09:30 +00:00
|
|
|
ULID: ulid,
|
|
|
|
MinTime: mint,
|
|
|
|
MaxTime: maxt,
|
2017-01-29 07:11:47 +00:00
|
|
|
}); err != nil {
|
2017-05-18 14:09:30 +00:00
|
|
|
return "", err
|
2017-03-02 08:13:29 +00:00
|
|
|
}
|
2017-05-15 17:58:14 +00:00
|
|
|
|
2017-05-18 14:09:30 +00:00
|
|
|
return dir, renameFile(tmp, dir)
|
2016-12-04 12:16:11 +00:00
|
|
|
}
|
|
|
|
|
2017-05-12 14:34:41 +00:00
|
|
|
// OpenHeadBlock opens the head block in dir.
|
2017-05-13 16:14:18 +00:00
|
|
|
func OpenHeadBlock(dir string, l log.Logger, wal WAL) (*HeadBlock, error) {
|
2017-01-19 13:01:38 +00:00
|
|
|
meta, err := readMetaFile(dir)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2016-12-22 11:05:24 +00:00
|
|
|
|
2017-05-12 14:34:41 +00:00
|
|
|
h := &HeadBlock{
|
2017-05-15 17:58:14 +00:00
|
|
|
dir: dir,
|
|
|
|
wal: wal,
|
|
|
|
series: []*memSeries{nil}, // 0 is not a valid posting, filled with nil.
|
|
|
|
hashes: map[uint64][]*memSeries{},
|
|
|
|
values: map[string]stringset{},
|
2017-08-05 11:31:48 +00:00
|
|
|
symbols: map[string]struct{}{},
|
2017-05-15 17:58:14 +00:00
|
|
|
postings: &memPostings{m: make(map[term][]uint32)},
|
|
|
|
meta: *meta,
|
2017-05-19 17:24:29 +00:00
|
|
|
tombstones: newEmptyTombstoneReader(),
|
2017-01-07 15:20:32 +00:00
|
|
|
}
|
2017-05-13 16:14:18 +00:00
|
|
|
return h, h.init()
|
|
|
|
}
|
2017-01-06 11:37:28 +00:00
|
|
|
|
2017-05-13 16:14:18 +00:00
|
|
|
func (h *HeadBlock) init() error {
|
|
|
|
r := h.wal.Reader()
|
2017-02-15 05:54:59 +00:00
|
|
|
|
2017-05-23 10:45:16 +00:00
|
|
|
seriesFunc := func(series []labels.Labels) error {
|
2017-02-15 05:54:59 +00:00
|
|
|
for _, lset := range series {
|
2017-01-19 10:22:47 +00:00
|
|
|
h.create(lset.Hash(), lset)
|
|
|
|
h.meta.Stats.NumSeries++
|
2017-02-15 05:54:59 +00:00
|
|
|
}
|
2017-05-23 10:45:16 +00:00
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
samplesFunc := func(samples []RefSample) error {
|
2017-02-15 05:54:59 +00:00
|
|
|
for _, s := range samples {
|
2017-05-12 15:06:26 +00:00
|
|
|
if int(s.Ref) >= len(h.series) {
|
2017-05-23 10:45:16 +00:00
|
|
|
return errors.Errorf("unknown series reference %d (max %d); abort WAL restore",
|
|
|
|
s.Ref, len(h.series))
|
2017-03-23 17:27:20 +00:00
|
|
|
}
|
2017-05-12 15:06:26 +00:00
|
|
|
h.series[s.Ref].append(s.T, s.V)
|
2017-01-06 08:26:39 +00:00
|
|
|
|
2017-05-12 15:06:26 +00:00
|
|
|
if !h.inBounds(s.T) {
|
2017-05-13 16:14:18 +00:00
|
|
|
return errors.Wrap(ErrOutOfBounds, "consume WAL")
|
2017-01-19 10:22:47 +00:00
|
|
|
}
|
|
|
|
h.meta.Stats.NumSamples++
|
2017-02-15 05:54:59 +00:00
|
|
|
}
|
2017-05-23 10:45:16 +00:00
|
|
|
|
|
|
|
return nil
|
2017-02-15 05:54:59 +00:00
|
|
|
}
|
2017-05-26 15:56:31 +00:00
|
|
|
deletesFunc := func(stones []Stone) error {
|
2017-05-23 10:45:16 +00:00
|
|
|
for _, s := range stones {
|
|
|
|
for _, itv := range s.intervals {
|
2017-05-26 15:56:31 +00:00
|
|
|
h.tombstones.add(s.ref, itv)
|
2017-05-23 10:45:16 +00:00
|
|
|
}
|
|
|
|
}
|
2017-05-15 17:58:14 +00:00
|
|
|
|
2017-05-23 10:45:16 +00:00
|
|
|
return nil
|
2017-05-15 17:58:14 +00:00
|
|
|
}
|
|
|
|
|
2017-05-23 10:45:16 +00:00
|
|
|
if err := r.Read(seriesFunc, samplesFunc, deletesFunc); err != nil {
|
|
|
|
return errors.Wrap(err, "consume WAL")
|
2017-05-15 17:58:14 +00:00
|
|
|
}
|
2017-05-23 10:45:16 +00:00
|
|
|
|
|
|
|
return nil
|
2016-12-22 11:05:24 +00:00
|
|
|
}
|
|
|
|
|
2017-01-19 13:01:38 +00:00
|
|
|
// inBounds returns true if the given timestamp is within the valid
|
|
|
|
// time bounds of the block.
|
2017-05-12 14:34:41 +00:00
|
|
|
func (h *HeadBlock) inBounds(t int64) bool {
|
2017-02-01 14:29:48 +00:00
|
|
|
return t >= h.meta.MinTime && t <= h.meta.MaxTime
|
2017-01-19 13:01:38 +00:00
|
|
|
}
|
|
|
|
|
2017-05-12 14:34:41 +00:00
|
|
|
func (h *HeadBlock) String() string {
|
2017-05-18 14:09:30 +00:00
|
|
|
return h.meta.ULID.String()
|
2017-03-20 09:41:43 +00:00
|
|
|
}
|
|
|
|
|
2016-12-22 11:05:24 +00:00
|
|
|
// Close syncs all data and closes underlying resources of the head block.
|
2017-05-12 14:34:41 +00:00
|
|
|
func (h *HeadBlock) Close() error {
|
2017-03-04 15:50:48 +00:00
|
|
|
h.mtx.Lock()
|
2017-03-17 11:12:50 +00:00
|
|
|
defer h.mtx.Unlock()
|
2017-03-04 15:50:48 +00:00
|
|
|
|
2017-02-27 09:46:15 +00:00
|
|
|
if err := h.wal.Close(); err != nil {
|
2017-03-08 15:53:07 +00:00
|
|
|
return errors.Wrapf(err, "close WAL for head %s", h.dir)
|
2017-02-02 06:58:54 +00:00
|
|
|
}
|
2017-02-27 09:46:15 +00:00
|
|
|
// Check whether the head block still exists in the underlying dir
|
2017-03-02 13:32:09 +00:00
|
|
|
// or has already been replaced with a compacted version or removed.
|
2017-02-27 09:46:15 +00:00
|
|
|
meta, err := readMetaFile(h.dir)
|
2017-03-02 13:32:09 +00:00
|
|
|
if os.IsNotExist(err) {
|
|
|
|
return nil
|
|
|
|
}
|
2017-02-27 09:46:15 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if meta.ULID == h.meta.ULID {
|
|
|
|
return writeMetaFile(h.dir, &h.meta)
|
|
|
|
}
|
2017-03-17 11:12:50 +00:00
|
|
|
|
|
|
|
h.closed = true
|
2017-02-27 09:46:15 +00:00
|
|
|
return nil
|
2016-12-09 09:41:51 +00:00
|
|
|
}
|
|
|
|
|
2017-05-17 14:43:01 +00:00
|
|
|
// Meta returns a BlockMeta for the head block.
|
2017-05-12 14:34:41 +00:00
|
|
|
func (h *HeadBlock) Meta() BlockMeta {
|
2017-03-23 17:27:20 +00:00
|
|
|
m := BlockMeta{
|
|
|
|
ULID: h.meta.ULID,
|
|
|
|
MinTime: h.meta.MinTime,
|
|
|
|
MaxTime: h.meta.MaxTime,
|
|
|
|
Compaction: h.meta.Compaction,
|
|
|
|
}
|
2017-01-19 10:22:47 +00:00
|
|
|
|
2017-03-23 17:27:20 +00:00
|
|
|
m.Stats.NumChunks = atomic.LoadUint64(&h.meta.Stats.NumChunks)
|
|
|
|
m.Stats.NumSeries = atomic.LoadUint64(&h.meta.Stats.NumSeries)
|
|
|
|
m.Stats.NumSamples = atomic.LoadUint64(&h.meta.Stats.NumSamples)
|
|
|
|
|
|
|
|
return m
|
2017-01-19 10:22:47 +00:00
|
|
|
}
|
|
|
|
|
2017-05-22 07:28:38 +00:00
|
|
|
// Tombstones returns the TombstoneReader against the block.
|
2017-05-16 14:18:28 +00:00
|
|
|
func (h *HeadBlock) Tombstones() TombstoneReader {
|
2017-05-24 05:54:24 +00:00
|
|
|
return h.tombstones
|
2017-05-16 14:18:28 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Delete implements headBlock.
|
2017-05-15 17:58:14 +00:00
|
|
|
func (h *HeadBlock) Delete(mint int64, maxt int64, ms ...labels.Matcher) error {
|
|
|
|
ir := h.Index()
|
|
|
|
|
|
|
|
pr := newPostingsReader(ir)
|
|
|
|
p, absent := pr.Select(ms...)
|
|
|
|
|
2017-05-26 15:56:31 +00:00
|
|
|
var stones []Stone
|
|
|
|
|
2017-05-15 17:58:14 +00:00
|
|
|
Outer:
|
|
|
|
for p.Next() {
|
|
|
|
ref := p.At()
|
|
|
|
lset := h.series[ref].lset
|
|
|
|
for _, abs := range absent {
|
|
|
|
if lset.Get(abs) != "" {
|
|
|
|
continue Outer
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-05-22 05:58:24 +00:00
|
|
|
// Delete only until the current values and not beyond.
|
2017-05-27 16:29:49 +00:00
|
|
|
tmin, tmax := clampInterval(mint, maxt, h.series[ref].chunks[0].minTime, h.series[ref].head().maxTime)
|
|
|
|
stones = append(stones, Stone{ref, intervals{{tmin, tmax}}})
|
2017-05-15 17:58:14 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if p.Err() != nil {
|
|
|
|
return p.Err()
|
|
|
|
}
|
2017-05-26 15:56:31 +00:00
|
|
|
if err := h.wal.LogDeletes(stones); err != nil {
|
2017-05-23 10:45:16 +00:00
|
|
|
return err
|
|
|
|
}
|
2017-05-15 17:58:14 +00:00
|
|
|
|
2017-05-26 15:56:31 +00:00
|
|
|
for _, s := range stones {
|
|
|
|
h.tombstones.add(s.ref, s.intervals[0])
|
2017-05-23 10:45:16 +00:00
|
|
|
}
|
|
|
|
|
2017-05-26 15:56:31 +00:00
|
|
|
h.meta.Stats.NumTombstones = uint64(len(h.tombstones))
|
2017-05-23 10:45:16 +00:00
|
|
|
return nil
|
2017-05-15 17:58:14 +00:00
|
|
|
}
|
2017-05-14 09:06:26 +00:00
|
|
|
|
2017-06-05 08:18:31 +00:00
|
|
|
// Snapshot persists the current state of the headblock to the given directory.
|
2017-06-06 12:45:54 +00:00
|
|
|
// TODO(gouthamve): Snapshot must be called when there are no active appenders.
|
|
|
|
// This has been ensured by acquiring a Lock on DB.mtx, but this limitation should
|
|
|
|
// be removed in the future.
|
2017-06-05 08:18:31 +00:00
|
|
|
func (h *HeadBlock) Snapshot(snapshotDir string) error {
|
|
|
|
if h.meta.Stats.NumSeries == 0 {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
entropy := rand.New(rand.NewSource(time.Now().UnixNano()))
|
|
|
|
uid := ulid.MustNew(ulid.Now(), entropy)
|
|
|
|
|
|
|
|
dir := filepath.Join(snapshotDir, uid.String())
|
|
|
|
tmp := dir + ".tmp"
|
|
|
|
|
|
|
|
if err := os.RemoveAll(tmp); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := os.MkdirAll(tmp, 0777); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
// Populate chunk and index files into temporary directory with
|
|
|
|
// data of all blocks.
|
|
|
|
chunkw, err := newChunkWriter(chunkDir(tmp))
|
|
|
|
if err != nil {
|
|
|
|
return errors.Wrap(err, "open chunk writer")
|
|
|
|
}
|
|
|
|
indexw, err := newIndexWriter(tmp)
|
|
|
|
if err != nil {
|
|
|
|
return errors.Wrap(err, "open index writer")
|
|
|
|
}
|
|
|
|
|
|
|
|
meta, err := populateBlock([]Block{h}, indexw, chunkw)
|
|
|
|
if err != nil {
|
|
|
|
return errors.Wrap(err, "write snapshot")
|
|
|
|
}
|
|
|
|
meta.ULID = uid
|
2017-07-12 11:48:13 +00:00
|
|
|
meta.MaxTime = h.highTimestamp
|
2017-06-05 08:18:31 +00:00
|
|
|
|
|
|
|
if err = writeMetaFile(tmp, meta); err != nil {
|
|
|
|
return errors.Wrap(err, "write merged meta")
|
|
|
|
}
|
|
|
|
|
|
|
|
if err = chunkw.Close(); err != nil {
|
|
|
|
return errors.Wrap(err, "close chunk writer")
|
|
|
|
}
|
|
|
|
if err = indexw.Close(); err != nil {
|
|
|
|
return errors.Wrap(err, "close index writer")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Create an empty tombstones file.
|
|
|
|
if err := writeTombstoneFile(tmp, newEmptyTombstoneReader()); err != nil {
|
|
|
|
return errors.Wrap(err, "write new tombstones file")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Block successfully written, make visible
|
|
|
|
if err := renameFile(tmp, dir); err != nil {
|
|
|
|
return errors.Wrap(err, "rename block dir")
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2017-05-17 14:43:01 +00:00
|
|
|
// Dir returns the directory of the block.
|
|
|
|
func (h *HeadBlock) Dir() string { return h.dir }
|
|
|
|
|
|
|
|
// Index returns an IndexReader against the block.
|
2017-08-05 11:31:48 +00:00
|
|
|
func (h *HeadBlock) Index() IndexReader {
|
|
|
|
h.mtx.RLock()
|
|
|
|
defer h.mtx.RUnlock()
|
|
|
|
|
|
|
|
return &headIndexReader{HeadBlock: h, maxSeries: uint32(len(h.series) - 1)}
|
|
|
|
}
|
2017-05-17 14:43:01 +00:00
|
|
|
|
|
|
|
// Chunks returns a ChunkReader against the block.
|
2017-05-12 14:34:41 +00:00
|
|
|
func (h *HeadBlock) Chunks() ChunkReader { return &headChunkReader{h} }
|
2017-01-10 14:28:22 +00:00
|
|
|
|
2017-05-17 14:43:01 +00:00
|
|
|
// Querier returns a new Querier against the block for the range [mint, maxt].
|
2017-05-12 14:34:41 +00:00
|
|
|
func (h *HeadBlock) Querier(mint, maxt int64) Querier {
|
2017-03-20 09:21:21 +00:00
|
|
|
h.mtx.RLock()
|
|
|
|
if h.closed {
|
|
|
|
panic(fmt.Sprintf("block %s already closed", h.dir))
|
|
|
|
}
|
2017-08-05 11:31:48 +00:00
|
|
|
h.mtx.RUnlock()
|
2017-03-21 11:12:33 +00:00
|
|
|
|
2017-03-20 09:21:21 +00:00
|
|
|
return &blockQuerier{
|
2017-05-17 09:19:42 +00:00
|
|
|
mint: mint,
|
|
|
|
maxt: maxt,
|
|
|
|
index: h.Index(),
|
|
|
|
chunks: h.Chunks(),
|
2017-05-24 05:54:24 +00:00
|
|
|
tombstones: h.Tombstones(),
|
2017-03-20 09:21:21 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-05-17 14:43:01 +00:00
|
|
|
// Appender returns a new Appender against the head block.
|
2017-05-12 14:34:41 +00:00
|
|
|
func (h *HeadBlock) Appender() Appender {
|
2017-02-04 10:53:52 +00:00
|
|
|
atomic.AddUint64(&h.activeWriters, 1)
|
|
|
|
|
2017-01-12 18:18:51 +00:00
|
|
|
h.mtx.RLock()
|
2017-03-17 11:12:50 +00:00
|
|
|
|
|
|
|
if h.closed {
|
|
|
|
panic(fmt.Sprintf("block %s already closed", h.dir))
|
|
|
|
}
|
2017-05-12 14:34:41 +00:00
|
|
|
return &headAppender{HeadBlock: h, samples: getHeadAppendBuffer()}
|
2017-01-12 19:00:36 +00:00
|
|
|
}
|
|
|
|
|
2017-06-25 17:02:02 +00:00
|
|
|
// ActiveWriters returns true if the block has open write transactions.
|
|
|
|
func (h *HeadBlock) ActiveWriters() int {
|
|
|
|
return int(atomic.LoadUint64(&h.activeWriters))
|
|
|
|
}
|
|
|
|
|
|
|
|
// HighTimestamp returns the highest inserted sample timestamp.
|
|
|
|
func (h *HeadBlock) HighTimestamp() int64 {
|
|
|
|
return atomic.LoadInt64(&h.highTimestamp)
|
2017-03-20 07:41:56 +00:00
|
|
|
}
|
|
|
|
|
2017-01-12 19:00:36 +00:00
|
|
|
var headPool = sync.Pool{}
|
|
|
|
|
2017-05-12 15:06:26 +00:00
|
|
|
func getHeadAppendBuffer() []RefSample {
|
2017-01-12 19:00:36 +00:00
|
|
|
b := headPool.Get()
|
|
|
|
if b == nil {
|
2017-05-12 15:06:26 +00:00
|
|
|
return make([]RefSample, 0, 512)
|
2017-01-12 19:00:36 +00:00
|
|
|
}
|
2017-05-12 15:06:26 +00:00
|
|
|
return b.([]RefSample)
|
2017-01-12 19:00:36 +00:00
|
|
|
}
|
|
|
|
|
2017-05-12 15:06:26 +00:00
|
|
|
func putHeadAppendBuffer(b []RefSample) {
|
2017-01-12 19:00:36 +00:00
|
|
|
headPool.Put(b[:0])
|
2017-01-12 18:18:51 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
type headAppender struct {
|
2017-05-12 14:34:41 +00:00
|
|
|
*HeadBlock
|
2017-01-12 18:18:51 +00:00
|
|
|
|
2017-05-17 14:43:01 +00:00
|
|
|
newSeries []*hashedLabels
|
2017-01-12 18:18:51 +00:00
|
|
|
newLabels []labels.Labels
|
2017-05-17 14:43:01 +00:00
|
|
|
newHashes map[uint64]uint64
|
2017-01-13 14:25:11 +00:00
|
|
|
|
2017-06-25 17:02:02 +00:00
|
|
|
samples []RefSample
|
|
|
|
highTimestamp int64
|
2017-01-12 18:18:51 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
type hashedLabels struct {
|
2017-05-17 14:43:01 +00:00
|
|
|
ref uint64
|
2017-01-12 18:18:51 +00:00
|
|
|
hash uint64
|
|
|
|
labels labels.Labels
|
|
|
|
}
|
|
|
|
|
2017-05-17 14:43:01 +00:00
|
|
|
func (a *headAppender) Add(lset labels.Labels, t int64, v float64) (string, error) {
|
2017-05-05 14:22:11 +00:00
|
|
|
if !a.inBounds(t) {
|
2017-05-17 14:43:01 +00:00
|
|
|
return "", ErrOutOfBounds
|
2017-05-05 14:22:11 +00:00
|
|
|
}
|
|
|
|
|
2017-03-06 13:27:33 +00:00
|
|
|
hash := lset.Hash()
|
2017-05-17 14:43:01 +00:00
|
|
|
refb := make([]byte, 8)
|
2017-01-12 18:18:51 +00:00
|
|
|
|
2017-05-17 14:43:01 +00:00
|
|
|
// Series exists already in the block.
|
2017-01-12 18:18:51 +00:00
|
|
|
if ms := a.get(hash, lset); ms != nil {
|
2017-05-17 14:43:01 +00:00
|
|
|
binary.BigEndian.PutUint64(refb, uint64(ms.ref))
|
|
|
|
return string(refb), a.AddFast(string(refb), t, v)
|
2017-01-12 18:18:51 +00:00
|
|
|
}
|
2017-05-17 14:43:01 +00:00
|
|
|
// Series was added in this transaction previously.
|
2017-01-13 14:25:11 +00:00
|
|
|
if ref, ok := a.newHashes[hash]; ok {
|
2017-05-17 14:43:01 +00:00
|
|
|
binary.BigEndian.PutUint64(refb, ref)
|
|
|
|
// XXX(fabxc): there's no fast path for multiple samples for the same new series
|
|
|
|
// in the same transaction. We always return the invalid empty ref. It's has not
|
|
|
|
// been a relevant use case so far and is not worth the trouble.
|
2017-06-26 12:58:00 +00:00
|
|
|
return "", a.AddFast(string(refb), t, v)
|
2017-01-13 14:25:11 +00:00
|
|
|
}
|
2017-01-12 18:18:51 +00:00
|
|
|
|
2017-05-17 14:43:01 +00:00
|
|
|
// The series is completely new.
|
2017-01-12 18:18:51 +00:00
|
|
|
if a.newSeries == nil {
|
2017-01-13 15:14:40 +00:00
|
|
|
a.newHashes = map[uint64]uint64{}
|
2017-01-12 18:18:51 +00:00
|
|
|
}
|
2017-05-17 14:43:01 +00:00
|
|
|
// First sample for new series.
|
|
|
|
ref := uint64(len(a.newSeries))
|
|
|
|
|
|
|
|
a.newSeries = append(a.newSeries, &hashedLabels{
|
|
|
|
ref: ref,
|
|
|
|
hash: hash,
|
|
|
|
labels: lset,
|
|
|
|
})
|
|
|
|
// First bit indicates its a series created in this transaction.
|
|
|
|
ref |= (1 << 63)
|
|
|
|
|
2017-01-13 15:14:40 +00:00
|
|
|
a.newHashes[hash] = ref
|
2017-05-17 14:43:01 +00:00
|
|
|
binary.BigEndian.PutUint64(refb, ref)
|
2017-01-12 18:18:51 +00:00
|
|
|
|
2017-06-26 12:58:00 +00:00
|
|
|
return "", a.AddFast(string(refb), t, v)
|
2017-01-12 18:18:51 +00:00
|
|
|
}
|
|
|
|
|
2017-05-17 14:43:01 +00:00
|
|
|
func (a *headAppender) AddFast(ref string, t int64, v float64) error {
|
2017-05-18 14:09:30 +00:00
|
|
|
if len(ref) != 8 {
|
|
|
|
return errors.Wrap(ErrNotFound, "invalid ref length")
|
|
|
|
}
|
2017-05-17 14:43:01 +00:00
|
|
|
var (
|
|
|
|
refn = binary.BigEndian.Uint64(yoloBytes(ref))
|
|
|
|
id = (refn << 1) >> 1
|
|
|
|
inTx = refn&(1<<63) != 0
|
|
|
|
)
|
2017-01-12 18:18:51 +00:00
|
|
|
// Distinguish between existing series and series created in
|
|
|
|
// this transaction.
|
2017-05-17 14:43:01 +00:00
|
|
|
if inTx {
|
|
|
|
if id > uint64(len(a.newSeries)-1) {
|
|
|
|
return errors.Wrap(ErrNotFound, "transaction series ID too high")
|
2017-01-12 18:18:51 +00:00
|
|
|
}
|
|
|
|
// TODO(fabxc): we also have to validate here that the
|
|
|
|
// sample sequence is valid.
|
2017-05-17 14:43:01 +00:00
|
|
|
// We also have to revalidate it as we switch locks and create
|
2017-01-12 18:18:51 +00:00
|
|
|
// the new series.
|
2017-05-17 14:43:01 +00:00
|
|
|
} else if id > uint64(len(a.series)) {
|
|
|
|
return errors.Wrap(ErrNotFound, "transaction series ID too high")
|
2017-01-17 07:40:31 +00:00
|
|
|
} else {
|
2017-05-17 14:43:01 +00:00
|
|
|
ms := a.series[id]
|
2017-01-17 07:40:31 +00:00
|
|
|
if ms == nil {
|
2017-05-17 14:43:01 +00:00
|
|
|
return errors.Wrap(ErrNotFound, "nil series")
|
2017-01-17 07:40:31 +00:00
|
|
|
}
|
|
|
|
// TODO(fabxc): memory series should be locked here already.
|
|
|
|
// Only problem is release of locks in case of a rollback.
|
2017-01-19 14:03:57 +00:00
|
|
|
c := ms.head()
|
|
|
|
|
|
|
|
if !a.inBounds(t) {
|
2017-02-01 14:29:48 +00:00
|
|
|
return ErrOutOfBounds
|
2017-01-19 14:03:57 +00:00
|
|
|
}
|
2017-01-17 07:40:31 +00:00
|
|
|
if t < c.maxTime {
|
|
|
|
return ErrOutOfOrderSample
|
|
|
|
}
|
2017-05-02 21:06:40 +00:00
|
|
|
|
|
|
|
// We are allowing exact duplicates as we can encounter them in valid cases
|
|
|
|
// like federation and erroring out at that time would be extremely noisy.
|
2017-04-12 15:25:32 +00:00
|
|
|
if c.maxTime == t && math.Float64bits(ms.lastValue) != math.Float64bits(v) {
|
2017-01-17 07:40:31 +00:00
|
|
|
return ErrAmendSample
|
|
|
|
}
|
2017-01-12 18:18:51 +00:00
|
|
|
}
|
|
|
|
|
2017-06-25 17:02:02 +00:00
|
|
|
if t > a.highTimestamp {
|
|
|
|
a.highTimestamp = t
|
|
|
|
}
|
|
|
|
|
2017-05-12 15:06:26 +00:00
|
|
|
a.samples = append(a.samples, RefSample{
|
2017-05-17 14:43:01 +00:00
|
|
|
Ref: refn,
|
2017-05-12 15:06:26 +00:00
|
|
|
T: t,
|
|
|
|
V: v,
|
2017-01-12 18:18:51 +00:00
|
|
|
})
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2017-06-08 10:39:39 +00:00
|
|
|
func (a *headAppender) createSeries() error {
|
2017-01-12 18:18:51 +00:00
|
|
|
if len(a.newSeries) == 0 {
|
2017-06-08 10:39:39 +00:00
|
|
|
return nil
|
2017-01-12 18:18:51 +00:00
|
|
|
}
|
|
|
|
a.newLabels = make([]labels.Labels, 0, len(a.newSeries))
|
2017-01-13 15:14:40 +00:00
|
|
|
base0 := len(a.series)
|
2017-01-12 18:18:51 +00:00
|
|
|
|
|
|
|
a.mtx.RUnlock()
|
2017-06-08 10:39:39 +00:00
|
|
|
defer a.mtx.RLock()
|
2017-01-12 18:18:51 +00:00
|
|
|
a.mtx.Lock()
|
2017-06-08 10:39:39 +00:00
|
|
|
defer a.mtx.Unlock()
|
2017-01-12 18:18:51 +00:00
|
|
|
|
2017-01-13 15:14:40 +00:00
|
|
|
base1 := len(a.series)
|
|
|
|
|
2017-05-17 14:43:01 +00:00
|
|
|
for _, l := range a.newSeries {
|
2017-01-12 18:18:51 +00:00
|
|
|
// We switched locks and have to re-validate that the series were not
|
|
|
|
// created by another goroutine in the meantime.
|
2017-01-13 15:14:40 +00:00
|
|
|
if base1 > base0 {
|
|
|
|
if ms := a.get(l.hash, l.labels); ms != nil {
|
2017-05-17 14:43:01 +00:00
|
|
|
l.ref = uint64(ms.ref)
|
2017-01-13 15:14:40 +00:00
|
|
|
continue
|
|
|
|
}
|
2017-01-12 18:18:51 +00:00
|
|
|
}
|
|
|
|
// Series is still new.
|
|
|
|
a.newLabels = append(a.newLabels, l.labels)
|
2017-05-17 14:43:01 +00:00
|
|
|
l.ref = uint64(len(a.series))
|
2017-01-12 18:18:51 +00:00
|
|
|
|
2017-01-13 15:14:40 +00:00
|
|
|
a.create(l.hash, l.labels)
|
2017-01-12 18:18:51 +00:00
|
|
|
}
|
|
|
|
|
2017-06-08 10:39:39 +00:00
|
|
|
// Write all new series to the WAL.
|
|
|
|
if err := a.wal.LogSeries(a.newLabels); err != nil {
|
|
|
|
return errors.Wrap(err, "WAL log series")
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
2017-01-12 18:18:51 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (a *headAppender) Commit() error {
|
2017-02-04 10:53:52 +00:00
|
|
|
defer atomic.AddUint64(&a.activeWriters, ^uint64(0))
|
2017-01-12 19:00:36 +00:00
|
|
|
defer putHeadAppendBuffer(a.samples)
|
2017-05-23 10:45:16 +00:00
|
|
|
defer a.mtx.RUnlock()
|
2017-01-12 18:18:51 +00:00
|
|
|
|
2017-06-08 10:39:39 +00:00
|
|
|
if err := a.createSeries(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2017-01-13 14:25:11 +00:00
|
|
|
|
2017-05-17 14:43:01 +00:00
|
|
|
// We have to update the refs of samples for series we just created.
|
2017-01-13 15:14:40 +00:00
|
|
|
for i := range a.samples {
|
|
|
|
s := &a.samples[i]
|
2017-05-17 14:43:01 +00:00
|
|
|
if s.Ref&(1<<63) != 0 {
|
|
|
|
s.Ref = a.newSeries[(s.Ref<<1)>>1].ref
|
2017-01-13 15:14:40 +00:00
|
|
|
}
|
2017-01-12 18:18:51 +00:00
|
|
|
}
|
2017-01-17 07:40:31 +00:00
|
|
|
|
2017-06-08 10:39:39 +00:00
|
|
|
// Write all new samples to the WAL and add them to the
|
2017-01-17 07:40:31 +00:00
|
|
|
// in-mem database on success.
|
2017-05-26 15:56:31 +00:00
|
|
|
if err := a.wal.LogSamples(a.samples); err != nil {
|
|
|
|
return errors.Wrap(err, "WAL log samples")
|
2017-01-17 07:40:31 +00:00
|
|
|
}
|
|
|
|
|
2017-04-20 12:45:07 +00:00
|
|
|
total := uint64(len(a.samples))
|
2017-01-19 13:01:38 +00:00
|
|
|
|
2017-01-17 15:33:58 +00:00
|
|
|
for _, s := range a.samples {
|
2017-05-12 15:06:26 +00:00
|
|
|
if !a.series[s.Ref].append(s.T, s.V) {
|
2017-01-17 15:33:58 +00:00
|
|
|
total--
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-03-23 17:27:20 +00:00
|
|
|
atomic.AddUint64(&a.meta.Stats.NumSamples, total)
|
|
|
|
atomic.AddUint64(&a.meta.Stats.NumSeries, uint64(len(a.newSeries)))
|
2017-01-12 18:18:51 +00:00
|
|
|
|
2017-06-25 17:02:02 +00:00
|
|
|
for {
|
|
|
|
ht := a.HeadBlock.HighTimestamp()
|
|
|
|
if a.highTimestamp <= ht {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
if atomic.CompareAndSwapInt64(&a.HeadBlock.highTimestamp, ht, a.highTimestamp) {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-01-12 18:18:51 +00:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (a *headAppender) Rollback() error {
|
|
|
|
a.mtx.RUnlock()
|
2017-02-04 10:53:52 +00:00
|
|
|
atomic.AddUint64(&a.activeWriters, ^uint64(0))
|
|
|
|
putHeadAppendBuffer(a.samples)
|
2017-01-12 18:18:51 +00:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2017-02-23 09:50:22 +00:00
|
|
|
type headChunkReader struct {
|
2017-05-12 14:34:41 +00:00
|
|
|
*HeadBlock
|
2017-01-07 17:02:17 +00:00
|
|
|
}
|
2016-12-14 17:38:46 +00:00
|
|
|
|
2016-12-15 15:14:33 +00:00
|
|
|
// Chunk returns the chunk for the reference number.
|
2017-02-23 09:50:22 +00:00
|
|
|
func (h *headChunkReader) Chunk(ref uint64) (chunks.Chunk, error) {
|
2017-01-06 16:23:12 +00:00
|
|
|
h.mtx.RLock()
|
|
|
|
defer h.mtx.RUnlock()
|
|
|
|
|
2017-02-18 16:33:20 +00:00
|
|
|
si := ref >> 32
|
|
|
|
ci := (ref << 32) >> 32
|
|
|
|
|
2017-01-11 12:02:38 +00:00
|
|
|
c := &safeChunk{
|
2017-02-18 16:33:20 +00:00
|
|
|
Chunk: h.series[si].chunks[ci].chunk,
|
|
|
|
s: h.series[si],
|
|
|
|
i: int(ci),
|
2016-12-14 17:38:46 +00:00
|
|
|
}
|
2017-01-11 12:02:38 +00:00
|
|
|
return c, nil
|
2016-12-14 17:38:46 +00:00
|
|
|
}
|
|
|
|
|
2017-01-09 15:51:39 +00:00
|
|
|
type safeChunk struct {
|
2017-01-11 12:02:38 +00:00
|
|
|
chunks.Chunk
|
|
|
|
s *memSeries
|
|
|
|
i int
|
2017-01-09 15:51:39 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (c *safeChunk) Iterator() chunks.Iterator {
|
2017-01-11 12:02:38 +00:00
|
|
|
c.s.mtx.RLock()
|
|
|
|
defer c.s.mtx.RUnlock()
|
|
|
|
return c.s.iterator(c.i)
|
2017-01-09 15:51:39 +00:00
|
|
|
}
|
|
|
|
|
2017-01-11 12:02:38 +00:00
|
|
|
// func (c *safeChunk) Appender() (chunks.Appender, error) { panic("illegal") }
|
|
|
|
// func (c *safeChunk) Bytes() []byte { panic("illegal") }
|
|
|
|
// func (c *safeChunk) Encoding() chunks.Encoding { panic("illegal") }
|
2017-01-09 15:51:39 +00:00
|
|
|
|
2017-01-10 14:28:22 +00:00
|
|
|
type headIndexReader struct {
|
2017-05-12 14:34:41 +00:00
|
|
|
*HeadBlock
|
2017-08-05 11:31:48 +00:00
|
|
|
// Highest series that existed when the index reader was instantiated.
|
|
|
|
maxSeries uint32
|
|
|
|
}
|
|
|
|
|
|
|
|
func (h *headIndexReader) Symbols() (map[string]struct{}, error) {
|
|
|
|
return h.symbols, nil
|
2016-12-14 17:38:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// LabelValues returns the possible label values
|
2017-01-10 14:28:22 +00:00
|
|
|
func (h *headIndexReader) LabelValues(names ...string) (StringTuples, error) {
|
2017-01-06 16:23:12 +00:00
|
|
|
h.mtx.RLock()
|
|
|
|
defer h.mtx.RUnlock()
|
|
|
|
|
2016-12-14 17:38:46 +00:00
|
|
|
if len(names) != 1 {
|
|
|
|
return nil, errInvalidSize
|
|
|
|
}
|
|
|
|
var sl []string
|
|
|
|
|
2016-12-22 00:12:28 +00:00
|
|
|
for s := range h.values[names[0]] {
|
2016-12-14 17:38:46 +00:00
|
|
|
sl = append(sl, s)
|
|
|
|
}
|
|
|
|
sort.Strings(sl)
|
|
|
|
|
2017-01-05 14:13:01 +00:00
|
|
|
return &stringTuples{l: len(names), s: sl}, nil
|
2016-12-14 17:38:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Postings returns the postings list iterator for the label pair.
|
2017-01-10 14:28:22 +00:00
|
|
|
func (h *headIndexReader) Postings(name, value string) (Postings, error) {
|
2017-01-06 16:23:12 +00:00
|
|
|
h.mtx.RLock()
|
|
|
|
defer h.mtx.RUnlock()
|
|
|
|
|
2016-12-22 00:12:28 +00:00
|
|
|
return h.postings.get(term{name: name, value: value}), nil
|
2016-12-14 17:38:46 +00:00
|
|
|
}
|
|
|
|
|
2017-08-05 11:31:48 +00:00
|
|
|
func (h *headIndexReader) SortedPostings(p Postings) Postings {
|
|
|
|
h.mtx.RLock()
|
|
|
|
defer h.mtx.RUnlock()
|
|
|
|
|
|
|
|
ep := make([]uint32, 0, 1024)
|
|
|
|
|
|
|
|
for p.Next() {
|
|
|
|
// Skip posting entries that include series added after we
|
|
|
|
// instantiated the index reader.
|
|
|
|
if p.At() > h.maxSeries {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
ep = append(ep, p.At())
|
|
|
|
}
|
|
|
|
if err := p.Err(); err != nil {
|
|
|
|
return errPostings{err: errors.Wrap(err, "expand postings")}
|
|
|
|
}
|
|
|
|
|
|
|
|
sort.Slice(ep, func(i, j int) bool {
|
|
|
|
return labels.Compare(h.series[ep[i]].lset, h.series[ep[j]].lset) < 0
|
|
|
|
})
|
|
|
|
return newListPostings(ep)
|
|
|
|
}
|
|
|
|
|
2016-12-14 17:38:46 +00:00
|
|
|
// Series returns the series for the given reference.
|
2017-08-06 18:41:24 +00:00
|
|
|
func (h *headIndexReader) Series(ref uint32, lbls *labels.Labels, chks *[]ChunkMeta) error {
|
2017-01-06 16:23:12 +00:00
|
|
|
h.mtx.RLock()
|
|
|
|
defer h.mtx.RUnlock()
|
|
|
|
|
2017-08-05 11:31:48 +00:00
|
|
|
if ref > h.maxSeries {
|
|
|
|
return ErrNotFound
|
2016-12-14 17:38:46 +00:00
|
|
|
}
|
2017-05-15 17:58:14 +00:00
|
|
|
|
2017-01-11 12:02:38 +00:00
|
|
|
s := h.series[ref]
|
2017-05-17 14:43:01 +00:00
|
|
|
if s == nil {
|
2017-08-05 11:31:48 +00:00
|
|
|
return ErrNotFound
|
2017-05-17 14:43:01 +00:00
|
|
|
}
|
2017-08-05 11:31:48 +00:00
|
|
|
*lbls = append((*lbls)[:0], s.lset...)
|
2017-01-11 12:02:38 +00:00
|
|
|
|
|
|
|
s.mtx.RLock()
|
|
|
|
defer s.mtx.RUnlock()
|
|
|
|
|
2017-08-05 11:31:48 +00:00
|
|
|
*chks = (*chks)[:0]
|
|
|
|
|
2017-01-11 12:02:38 +00:00
|
|
|
for i, c := range s.chunks {
|
2017-08-06 18:41:24 +00:00
|
|
|
*chks = append(*chks, ChunkMeta{
|
2017-01-11 12:02:38 +00:00
|
|
|
MinTime: c.minTime,
|
|
|
|
MaxTime: c.maxTime,
|
2017-02-18 16:33:20 +00:00
|
|
|
Ref: (uint64(ref) << 32) | uint64(i),
|
2017-01-11 12:02:38 +00:00
|
|
|
})
|
2017-01-03 14:43:26 +00:00
|
|
|
}
|
2017-01-11 12:02:38 +00:00
|
|
|
|
2017-08-05 11:31:48 +00:00
|
|
|
return nil
|
2016-12-31 14:35:08 +00:00
|
|
|
}
|
|
|
|
|
2017-01-10 14:28:22 +00:00
|
|
|
func (h *headIndexReader) LabelIndices() ([][]string, error) {
|
2017-01-06 16:23:12 +00:00
|
|
|
h.mtx.RLock()
|
|
|
|
defer h.mtx.RUnlock()
|
|
|
|
|
2016-12-31 14:35:08 +00:00
|
|
|
res := [][]string{}
|
|
|
|
|
|
|
|
for s := range h.values {
|
|
|
|
res = append(res, []string{s})
|
2016-12-14 17:38:46 +00:00
|
|
|
}
|
2016-12-31 14:35:08 +00:00
|
|
|
return res, nil
|
2016-12-14 17:38:46 +00:00
|
|
|
}
|
|
|
|
|
2016-12-04 12:16:11 +00:00
|
|
|
// get retrieves the chunk with the hash and label set and creates
|
|
|
|
// a new one if it doesn't exist yet.
|
2017-05-12 14:34:41 +00:00
|
|
|
func (h *HeadBlock) get(hash uint64, lset labels.Labels) *memSeries {
|
2017-01-11 12:02:38 +00:00
|
|
|
series := h.hashes[hash]
|
2016-12-22 00:12:28 +00:00
|
|
|
|
2017-01-11 12:02:38 +00:00
|
|
|
for _, s := range series {
|
|
|
|
if s.lset.Equals(lset) {
|
|
|
|
return s
|
2016-12-04 12:16:11 +00:00
|
|
|
}
|
|
|
|
}
|
2016-12-31 09:19:02 +00:00
|
|
|
return nil
|
2016-12-22 11:05:24 +00:00
|
|
|
}
|
|
|
|
|
2017-05-12 14:34:41 +00:00
|
|
|
func (h *HeadBlock) create(hash uint64, lset labels.Labels) *memSeries {
|
2017-06-07 11:42:53 +00:00
|
|
|
s := newMemSeries(lset, uint32(len(h.series)), h.meta.MaxTime)
|
2017-01-04 13:06:40 +00:00
|
|
|
|
2017-01-12 18:18:51 +00:00
|
|
|
// Allocate empty space until we can insert at the given index.
|
2017-01-13 15:14:40 +00:00
|
|
|
h.series = append(h.series, s)
|
2016-12-22 00:12:28 +00:00
|
|
|
|
2017-01-11 12:02:38 +00:00
|
|
|
h.hashes[hash] = append(h.hashes[hash], s)
|
2016-12-22 00:12:28 +00:00
|
|
|
|
|
|
|
for _, l := range lset {
|
|
|
|
valset, ok := h.values[l.Name]
|
|
|
|
if !ok {
|
|
|
|
valset = stringset{}
|
|
|
|
h.values[l.Name] = valset
|
|
|
|
}
|
|
|
|
valset.set(l.Value)
|
2017-01-03 14:43:26 +00:00
|
|
|
|
2017-01-11 12:02:38 +00:00
|
|
|
h.postings.add(s.ref, term{name: l.Name, value: l.Value})
|
2017-08-05 11:31:48 +00:00
|
|
|
|
|
|
|
h.symbols[l.Name] = struct{}{}
|
|
|
|
h.symbols[l.Value] = struct{}{}
|
2016-12-22 00:12:28 +00:00
|
|
|
}
|
2017-01-03 14:43:26 +00:00
|
|
|
|
2017-01-11 12:02:38 +00:00
|
|
|
h.postings.add(s.ref, term{})
|
2016-12-04 12:16:11 +00:00
|
|
|
|
2017-01-11 12:02:38 +00:00
|
|
|
return s
|
2016-12-04 12:16:11 +00:00
|
|
|
}
|
|
|
|
|
2017-03-24 09:20:39 +00:00
|
|
|
type sample struct {
|
|
|
|
t int64
|
|
|
|
v float64
|
|
|
|
}
|
|
|
|
|
2017-01-11 12:02:38 +00:00
|
|
|
type memSeries struct {
|
2017-01-09 15:51:39 +00:00
|
|
|
mtx sync.RWMutex
|
|
|
|
|
2017-01-11 12:02:38 +00:00
|
|
|
ref uint32
|
|
|
|
lset labels.Labels
|
|
|
|
chunks []*memChunk
|
2017-01-09 15:51:39 +00:00
|
|
|
|
2017-06-07 11:42:53 +00:00
|
|
|
nextAt int64 // timestamp at which to cut the next chunk.
|
|
|
|
maxt int64 // maximum timestamp for the series.
|
2017-01-11 12:02:38 +00:00
|
|
|
lastValue float64
|
2017-01-09 15:51:39 +00:00
|
|
|
sampleBuf [4]sample
|
|
|
|
|
2017-03-06 16:34:49 +00:00
|
|
|
app chunks.Appender // Current appender for the chunk.
|
2017-01-09 15:51:39 +00:00
|
|
|
}
|
|
|
|
|
2017-06-07 11:42:53 +00:00
|
|
|
func (s *memSeries) cut(mint int64) *memChunk {
|
2017-01-11 12:02:38 +00:00
|
|
|
c := &memChunk{
|
|
|
|
chunk: chunks.NewXORChunk(),
|
2017-06-07 11:42:53 +00:00
|
|
|
minTime: mint,
|
2017-01-11 12:02:38 +00:00
|
|
|
maxTime: math.MinInt64,
|
|
|
|
}
|
|
|
|
s.chunks = append(s.chunks, c)
|
|
|
|
|
|
|
|
app, err := c.chunk.Appender()
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
s.app = app
|
|
|
|
return c
|
|
|
|
}
|
|
|
|
|
2017-06-07 11:42:53 +00:00
|
|
|
func newMemSeries(lset labels.Labels, id uint32, maxt int64) *memSeries {
|
|
|
|
s := &memSeries{
|
|
|
|
lset: lset,
|
|
|
|
ref: id,
|
|
|
|
maxt: maxt,
|
|
|
|
nextAt: math.MinInt64,
|
|
|
|
}
|
|
|
|
return s
|
|
|
|
}
|
|
|
|
|
2017-01-11 12:02:38 +00:00
|
|
|
func (s *memSeries) append(t int64, v float64) bool {
|
2017-06-07 11:42:53 +00:00
|
|
|
const samplesPerChunk = 120
|
|
|
|
|
2017-03-20 13:45:27 +00:00
|
|
|
s.mtx.Lock()
|
|
|
|
defer s.mtx.Unlock()
|
|
|
|
|
2017-01-11 12:02:38 +00:00
|
|
|
var c *memChunk
|
|
|
|
|
2017-06-07 11:42:53 +00:00
|
|
|
if len(s.chunks) == 0 {
|
|
|
|
c = s.cut(t)
|
|
|
|
}
|
|
|
|
c = s.head()
|
|
|
|
if c.maxTime >= t {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
if c.samples > samplesPerChunk/4 && t >= s.nextAt {
|
|
|
|
c = s.cut(t)
|
2017-01-09 15:51:39 +00:00
|
|
|
}
|
2017-01-11 12:02:38 +00:00
|
|
|
s.app.Append(t, v)
|
|
|
|
|
|
|
|
c.maxTime = t
|
|
|
|
c.samples++
|
|
|
|
|
2017-06-07 11:42:53 +00:00
|
|
|
if c.samples == samplesPerChunk/4 {
|
|
|
|
s.nextAt = computeChunkEndTime(c.minTime, c.maxTime, s.maxt)
|
|
|
|
}
|
|
|
|
|
2017-01-11 12:02:38 +00:00
|
|
|
s.lastValue = v
|
2017-01-09 15:51:39 +00:00
|
|
|
|
2017-01-11 12:02:38 +00:00
|
|
|
s.sampleBuf[0] = s.sampleBuf[1]
|
|
|
|
s.sampleBuf[1] = s.sampleBuf[2]
|
|
|
|
s.sampleBuf[2] = s.sampleBuf[3]
|
|
|
|
s.sampleBuf[3] = sample{t: t, v: v}
|
2017-01-09 15:51:39 +00:00
|
|
|
|
2017-01-11 12:02:38 +00:00
|
|
|
return true
|
2017-01-09 15:51:39 +00:00
|
|
|
}
|
|
|
|
|
2017-06-07 11:42:53 +00:00
|
|
|
// computeChunkEndTime estimates the end timestamp based the beginning of a chunk,
|
|
|
|
// its current timestamp and the upper bound up to which we insert data.
|
|
|
|
// It assumes that the time range is 1/4 full.
|
|
|
|
func computeChunkEndTime(start, cur, max int64) int64 {
|
|
|
|
a := (max - start) / ((cur - start + 1) * 4)
|
|
|
|
if a == 0 {
|
|
|
|
return max
|
|
|
|
}
|
|
|
|
return start + (max-start)/a
|
|
|
|
}
|
|
|
|
|
2017-01-11 12:02:38 +00:00
|
|
|
func (s *memSeries) iterator(i int) chunks.Iterator {
|
|
|
|
c := s.chunks[i]
|
|
|
|
|
|
|
|
if i < len(s.chunks)-1 {
|
|
|
|
return c.chunk.Iterator()
|
|
|
|
}
|
|
|
|
|
2017-01-09 15:51:39 +00:00
|
|
|
it := &memSafeIterator{
|
2017-01-11 12:02:38 +00:00
|
|
|
Iterator: c.chunk.Iterator(),
|
2017-01-09 15:51:39 +00:00
|
|
|
i: -1,
|
2017-01-11 12:02:38 +00:00
|
|
|
total: c.samples,
|
|
|
|
buf: s.sampleBuf,
|
2017-01-09 15:51:39 +00:00
|
|
|
}
|
|
|
|
return it
|
|
|
|
}
|
|
|
|
|
2017-01-11 12:02:38 +00:00
|
|
|
func (s *memSeries) head() *memChunk {
|
|
|
|
return s.chunks[len(s.chunks)-1]
|
|
|
|
}
|
|
|
|
|
|
|
|
type memChunk struct {
|
|
|
|
chunk chunks.Chunk
|
|
|
|
minTime, maxTime int64
|
|
|
|
samples int
|
|
|
|
}
|
|
|
|
|
2017-01-09 15:51:39 +00:00
|
|
|
type memSafeIterator struct {
|
|
|
|
chunks.Iterator
|
|
|
|
|
|
|
|
i int
|
|
|
|
total int
|
|
|
|
buf [4]sample
|
|
|
|
}
|
|
|
|
|
|
|
|
func (it *memSafeIterator) Next() bool {
|
|
|
|
if it.i+1 >= it.total {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
it.i++
|
|
|
|
if it.total-it.i > 4 {
|
|
|
|
return it.Iterator.Next()
|
|
|
|
}
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
|
|
|
|
func (it *memSafeIterator) At() (int64, float64) {
|
|
|
|
if it.total-it.i > 4 {
|
|
|
|
return it.Iterator.At()
|
|
|
|
}
|
|
|
|
s := it.buf[4-(it.total-it.i)]
|
|
|
|
return s.t, s.v
|
|
|
|
}
|