mudhorn/prometheus
1
0
Fork 0
mirror of https://github.com/prometheus/prometheus.git synced 2024-11-12 16:44:05 -08:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

storage: add AtT method to MemoizedSeriesIterator

Browse source 
Signed-off-by: Charles Korn <charles.korn@grafana.com>
This commit is contained in:
Charles Korn 2024-07-15 14:09:02 +10:00
parent 52a73ec871
commit 2ca4d16236
No known key found for this signature in database
2 changed files with 11 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
storage/memoized_iterator.go
View file

@ -136,6 +136,11 @@ func (b *MemoizedSeriesIterator) AtFloatHistogram() (int64, *histogram.FloatHist
return b.it.AtFloatHistogram(nil) return b.it.AtFloatHistogram(nil)
} }
// AtT returns the timestamp of the current element of the iterator.
func (b *MemoizedSeriesIterator) AtT() int64 {
return b.it.AtT()
}
// Err returns the last encountered error. // Err returns the last encountered error.
func (b *MemoizedSeriesIterator) Err() error { func (b *MemoizedSeriesIterator) Err() error {
return b.it.Err() return b.it.Err()

10
storage/memoized_iterator_test.go
View file

@ -29,13 +29,15 @@ func TestMemoizedSeriesIterator(t *testing.T) {
sampleEq := func(ets int64, ev float64, efh *histogram.FloatHistogram) { sampleEq := func(ets int64, ev float64, efh *histogram.FloatHistogram) {
if efh == nil { if efh == nil {
ts, v := it.At() ts, v := it.At()
require.Equal(t, ets, ts, "timestamp mismatch") require.Equal(t, ets, ts, "At() timestamp mismatch")
require.Equal(t, ev, v, "value mismatch") require.Equal(t, ev, v, "At() value mismatch")
} else { } else {
ts, fh := it.AtFloatHistogram() ts, fh := it.AtFloatHistogram()
require.Equal(t, ets, ts, "timestamp mismatch") require.Equal(t, ets, ts, "AtFloatHistogram() timestamp mismatch")
require.Equal(t, efh, fh, "histogram 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) { prevSampleEq := func(ets int64, ev float64, efh *histogram.FloatHistogram, eok bool) {
ts, v, fh, ok := it.PeekPrev() ts, v, fh, ok := it.PeekPrev()