mirror of https://github.com/prometheus/prometheus
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
121 lines
4.1 KiB
121 lines
4.1 KiB
// Copyright 2021 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. |
|
|
|
package storage |
|
|
|
import ( |
|
"testing" |
|
|
|
"github.com/stretchr/testify/require" |
|
|
|
"github.com/prometheus/prometheus/model/histogram" |
|
"github.com/prometheus/prometheus/tsdb/chunkenc" |
|
"github.com/prometheus/prometheus/tsdb/tsdbutil" |
|
) |
|
|
|
func TestMemoizedSeriesIterator(t *testing.T) { |
|
var it *MemoizedSeriesIterator |
|
|
|
sampleEq := func(ets int64, ev float64, efh *histogram.FloatHistogram) { |
|
if efh == nil { |
|
ts, v := it.At() |
|
require.Equal(t, ets, ts, "At() timestamp mismatch") |
|
require.Equal(t, ev, v, "At() value mismatch") |
|
} else { |
|
ts, fh := it.AtFloatHistogram() |
|
require.Equal(t, ets, ts, "AtFloatHistogram() timestamp mismatch") |
|
require.Equal(t, efh, fh, "AtFloatHistogram() histogram mismatch") |
|
} |
|
|
|
require.Equal(t, ets, it.AtT(), "AtT() timestamp mismatch") |
|
} |
|
prevSampleEq := func(ets int64, ev float64, efh *histogram.FloatHistogram, eok bool) { |
|
ts, v, fh, ok := it.PeekPrev() |
|
require.Equal(t, eok, ok, "exist mismatch") |
|
require.Equal(t, ets, ts, "timestamp mismatch") |
|
if efh == nil { |
|
require.Equal(t, ev, v, "value mismatch") |
|
} else { |
|
require.Equal(t, efh, fh, "histogram mismatch") |
|
} |
|
} |
|
|
|
it = NewMemoizedIterator(NewListSeriesIterator(samples{ |
|
fSample{t: 1, f: 2}, |
|
fSample{t: 2, f: 3}, |
|
fSample{t: 3, f: 4}, |
|
fSample{t: 4, f: 5}, |
|
fSample{t: 5, f: 6}, |
|
fSample{t: 99, f: 8}, |
|
fSample{t: 100, f: 9}, |
|
fSample{t: 101, f: 10}, |
|
hSample{t: 102, h: tsdbutil.GenerateTestHistogram(0)}, |
|
hSample{t: 103, h: tsdbutil.GenerateTestHistogram(1)}, |
|
fhSample{t: 104, fh: tsdbutil.GenerateTestFloatHistogram(2)}, |
|
fhSample{t: 199, fh: tsdbutil.GenerateTestFloatHistogram(3)}, |
|
hSample{t: 200, h: tsdbutil.GenerateTestHistogram(4)}, |
|
fhSample{t: 299, fh: tsdbutil.GenerateTestFloatHistogram(5)}, |
|
fSample{t: 300, f: 11}, |
|
hSample{t: 399, h: tsdbutil.GenerateTestHistogram(6)}, |
|
fSample{t: 400, f: 12}, |
|
}), 2) |
|
|
|
require.Equal(t, chunkenc.ValFloat, it.Seek(-123), "seek failed") |
|
sampleEq(1, 2, nil) |
|
prevSampleEq(0, 0, nil, false) |
|
|
|
require.Equal(t, chunkenc.ValFloat, it.Seek(5), "seek failed") |
|
sampleEq(5, 6, nil) |
|
prevSampleEq(4, 5, nil, true) |
|
|
|
// Seek to a histogram sample with a previous float sample. |
|
require.Equal(t, chunkenc.ValFloatHistogram, it.Seek(102), "seek failed") |
|
sampleEq(102, 10, tsdbutil.GenerateTestFloatHistogram(0)) |
|
prevSampleEq(101, 10, nil, true) |
|
|
|
// Attempt to seek backwards (no-op). |
|
require.Equal(t, chunkenc.ValFloatHistogram, it.Seek(50), "seek failed") |
|
sampleEq(102, 10, tsdbutil.GenerateTestFloatHistogram(0)) |
|
prevSampleEq(101, 10, nil, true) |
|
|
|
// Seek to a float histogram sample with a previous histogram sample. |
|
require.Equal(t, chunkenc.ValFloatHistogram, it.Seek(104), "seek failed") |
|
sampleEq(104, 0, tsdbutil.GenerateTestFloatHistogram(2)) |
|
prevSampleEq(103, 0, tsdbutil.GenerateTestFloatHistogram(1), true) |
|
|
|
// Seek to a float sample with a previous float histogram sample. |
|
require.Equal(t, chunkenc.ValFloat, it.Seek(300), "seek failed") |
|
sampleEq(300, 11, nil) |
|
prevSampleEq(299, 0, tsdbutil.GenerateTestFloatHistogram(5), true) |
|
|
|
// Seek to a float sample with a previous histogram sample. |
|
require.Equal(t, chunkenc.ValFloat, it.Seek(400), "seek failed") |
|
sampleEq(400, 12, nil) |
|
prevSampleEq(399, 0, tsdbutil.GenerateTestFloatHistogram(6), true) |
|
|
|
require.Equal(t, chunkenc.ValNone, it.Seek(1024), "seek succeeded unexpectedly") |
|
} |
|
|
|
func BenchmarkMemoizedSeriesIterator(b *testing.B) { |
|
// Simulate a 5 minute rate. |
|
it := NewMemoizedIterator(newFakeSeriesIterator(int64(b.N), 30), 5*60) |
|
|
|
b.SetBytes(16) |
|
b.ReportAllocs() |
|
b.ResetTimer() |
|
|
|
for it.Next() != chunkenc.ValNone { |
|
// Scan everything. |
|
} |
|
require.NoError(b, it.Err()) |
|
}
|
|
|