mudhorn/prometheus
1
0
Fork 0
mirror of https://github.com/prometheus/prometheus.git synced 2025-03-05 20:59:13 -08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix TestOOOHeadChunkReader_Chunk on 32-bit

Browse source 
Signed-off-by: Joel Beckmeyer <joel@beckmeyer.us>
This commit is contained in:
Joel Beckmeyer 2024-12-15 17:53:36 -05:00
parent bdace97744
commit 39f5a07236
8 changed files with 57 additions and 57 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
scrape/manager_test.go
View file

@ -894,7 +894,7 @@ func findSamplesForMetric(floats []floatSample, metricName string) (ret []floatS
// generateTestHistogram generates the same thing as tsdbutil.GenerateTestHistogram, // generateTestHistogram generates the same thing as tsdbutil.GenerateTestHistogram,
// but in the form of dto.Histogram. // but in the form of dto.Histogram.
func generateTestHistogram(i int) *dto.Histogram { func generateTestHistogram(i int) *dto.Histogram {
helper := tsdbutil.GenerateTestHistogram(i) helper := tsdbutil.GenerateTestHistogram(int64(i))
h := &dto.Histogram{} h := &dto.Histogram{}
h.SampleCount = proto.Uint64(helper.Count) h.SampleCount = proto.Uint64(helper.Count)
h.SampleSum = proto.Float64(helper.Sum) h.SampleSum = proto.Float64(helper.Sum)

4
storage/merge_test.go
View file

@ -385,13 +385,13 @@ func TestMergeChunkQuerierWithNoVerticalChunkSeriesMerger(t *testing.T) {
} }
func histogramSample(ts int64, hint histogram.CounterResetHint) hSample { func histogramSample(ts int64, hint histogram.CounterResetHint) hSample {
h := tsdbutil.GenerateTestHistogram(int(ts + 1)) h := tsdbutil.GenerateTestHistogram(ts + 1)
h.CounterResetHint = hint h.CounterResetHint = hint
return hSample{t: ts, h: h} return hSample{t: ts, h: h}
} }
func floatHistogramSample(ts int64, hint histogram.CounterResetHint) fhSample { func floatHistogramSample(ts int64, hint histogram.CounterResetHint) fhSample {
fh := tsdbutil.GenerateTestFloatHistogram(int(ts + 1)) fh := tsdbutil.GenerateTestFloatHistogram(ts + 1)
fh.CounterResetHint = hint fh.CounterResetHint = hint
return fhSample{t: ts, fh: fh} return fhSample{t: ts, fh: fh}
} }

40
tsdb/db_test.go
View file

@ -4101,7 +4101,7 @@ func TestOOOWALWrite(t *testing.T) {
}, },
"integer histogram": { "integer histogram": {
appendSample: func(app storage.Appender, l labels.Labels, mins int64) (storage.SeriesRef, error) { appendSample: func(app storage.Appender, l labels.Labels, mins int64) (storage.SeriesRef, error) {
seriesRef, err := app.AppendHistogram(0, l, minutes(mins), tsdbutil.GenerateTestHistogram(int(mins)), nil) seriesRef, err := app.AppendHistogram(0, l, minutes(mins), tsdbutil.GenerateTestHistogram(mins), nil)
require.NoError(t, err) require.NoError(t, err)
return seriesRef, nil return seriesRef, nil
}, },
@ -4192,7 +4192,7 @@ func TestOOOWALWrite(t *testing.T) {
}, },
"float histogram": { "float histogram": {
appendSample: func(app storage.Appender, l labels.Labels, mins int64) (storage.SeriesRef, error) { appendSample: func(app storage.Appender, l labels.Labels, mins int64) (storage.SeriesRef, error) {
seriesRef, err := app.AppendHistogram(0, l, minutes(mins), nil, tsdbutil.GenerateTestFloatHistogram(int(mins))) seriesRef, err := app.AppendHistogram(0, l, minutes(mins), nil, tsdbutil.GenerateTestFloatHistogram(mins))
require.NoError(t, err) require.NoError(t, err)
return seriesRef, nil return seriesRef, nil
}, },
@ -4736,12 +4736,12 @@ func TestMultipleEncodingsCommitOrder(t *testing.T) {
return sample{t: ts, f: float64(ts)} return sample{t: ts, f: float64(ts)}
} }
if valType == chunkenc.ValHistogram { if valType == chunkenc.ValHistogram {
h := tsdbutil.GenerateTestHistogram(int(ts)) h := tsdbutil.GenerateTestHistogram(ts)
_, err := app.AppendHistogram(0, labels.FromStrings("foo", "bar1"), ts, h, nil) _, err := app.AppendHistogram(0, labels.FromStrings("foo", "bar1"), ts, h, nil)
require.NoError(t, err) require.NoError(t, err)
return sample{t: ts, h: h} return sample{t: ts, h: h}
} }
fh := tsdbutil.GenerateTestFloatHistogram(int(ts)) fh := tsdbutil.GenerateTestFloatHistogram(ts)
_, err := app.AppendHistogram(0, labels.FromStrings("foo", "bar1"), ts, nil, fh) _, err := app.AppendHistogram(0, labels.FromStrings("foo", "bar1"), ts, nil, fh)
require.NoError(t, err) require.NoError(t, err)
return sample{t: ts, fh: fh} return sample{t: ts, fh: fh}
@ -5427,37 +5427,37 @@ func TestQuerierOOOQuery(t *testing.T) {
}, },
"integer histogram": { "integer histogram": {
appendFunc: func(app storage.Appender, ts int64, counterReset bool) (storage.SeriesRef, error) { appendFunc: func(app storage.Appender, ts int64, counterReset bool) (storage.SeriesRef, error) {
h := tsdbutil.GenerateTestHistogram(int(ts)) h := tsdbutil.GenerateTestHistogram(ts)
if counterReset { if counterReset {
h.CounterResetHint = histogram.CounterReset h.CounterResetHint = histogram.CounterReset
} }
return app.AppendHistogram(0, labels.FromStrings("foo", "bar1"), ts, h, nil) return app.AppendHistogram(0, labels.FromStrings("foo", "bar1"), ts, h, nil)
}, },
sampleFunc: func(ts int64) chunks.Sample { sampleFunc: func(ts int64) chunks.Sample {
return sample{t: ts, h: tsdbutil.GenerateTestHistogram(int(ts))} return sample{t: ts, h: tsdbutil.GenerateTestHistogram(ts)}
}, },
}, },
"float histogram": { "float histogram": {
appendFunc: func(app storage.Appender, ts int64, counterReset bool) (storage.SeriesRef, error) { appendFunc: func(app storage.Appender, ts int64, counterReset bool) (storage.SeriesRef, error) {
fh := tsdbutil.GenerateTestFloatHistogram(int(ts)) fh := tsdbutil.GenerateTestFloatHistogram(ts)
if counterReset { if counterReset {
fh.CounterResetHint = histogram.CounterReset fh.CounterResetHint = histogram.CounterReset
} }
return app.AppendHistogram(0, labels.FromStrings("foo", "bar1"), ts, nil, fh) return app.AppendHistogram(0, labels.FromStrings("foo", "bar1"), ts, nil, fh)
}, },
sampleFunc: func(ts int64) chunks.Sample { sampleFunc: func(ts int64) chunks.Sample {
return sample{t: ts, fh: tsdbutil.GenerateTestFloatHistogram(int(ts))} return sample{t: ts, fh: tsdbutil.GenerateTestFloatHistogram(ts)}
}, },
}, },
"integer histogram counter resets": { "integer histogram counter resets": {
// Adding counter reset to all histograms means each histogram will have its own chunk. // Adding counter reset to all histograms means each histogram will have its own chunk.
appendFunc: func(app storage.Appender, ts int64, counterReset bool) (storage.SeriesRef, error) { appendFunc: func(app storage.Appender, ts int64, counterReset bool) (storage.SeriesRef, error) {
h := tsdbutil.GenerateTestHistogram(int(ts)) h := tsdbutil.GenerateTestHistogram(ts)
h.CounterResetHint = histogram.CounterReset // For this scenario, ignore the counterReset argument. h.CounterResetHint = histogram.CounterReset // For this scenario, ignore the counterReset argument.
return app.AppendHistogram(0, labels.FromStrings("foo", "bar1"), ts, h, nil) return app.AppendHistogram(0, labels.FromStrings("foo", "bar1"), ts, h, nil)
}, },
sampleFunc: func(ts int64) chunks.Sample { sampleFunc: func(ts int64) chunks.Sample {
return sample{t: ts, h: tsdbutil.GenerateTestHistogram(int(ts))} return sample{t: ts, h: tsdbutil.GenerateTestHistogram(ts)}
}, },
}, },
} }
@ -5743,37 +5743,37 @@ func TestChunkQuerierOOOQuery(t *testing.T) {
}, },
"integer histogram": { "integer histogram": {
appendFunc: func(app storage.Appender, ts int64, counterReset bool) (storage.SeriesRef, error) { appendFunc: func(app storage.Appender, ts int64, counterReset bool) (storage.SeriesRef, error) {
h := tsdbutil.GenerateTestHistogram(int(ts)) h := tsdbutil.GenerateTestHistogram(ts)
if counterReset { if counterReset {
h.CounterResetHint = histogram.CounterReset h.CounterResetHint = histogram.CounterReset
} }
return app.AppendHistogram(0, labels.FromStrings("foo", "bar1"), ts, h, nil) return app.AppendHistogram(0, labels.FromStrings("foo", "bar1"), ts, h, nil)
}, },
sampleFunc: func(ts int64) chunks.Sample { sampleFunc: func(ts int64) chunks.Sample {
return sample{t: ts, h: tsdbutil.GenerateTestHistogram(int(ts))} return sample{t: ts, h: tsdbutil.GenerateTestHistogram(ts)}
}, },
}, },
"float histogram": { "float histogram": {
appendFunc: func(app storage.Appender, ts int64, counterReset bool) (storage.SeriesRef, error) { appendFunc: func(app storage.Appender, ts int64, counterReset bool) (storage.SeriesRef, error) {
fh := tsdbutil.GenerateTestFloatHistogram(int(ts)) fh := tsdbutil.GenerateTestFloatHistogram(ts)
if counterReset { if counterReset {
fh.CounterResetHint = histogram.CounterReset fh.CounterResetHint = histogram.CounterReset
} }
return app.AppendHistogram(0, labels.FromStrings("foo", "bar1"), ts, nil, fh) return app.AppendHistogram(0, labels.FromStrings("foo", "bar1"), ts, nil, fh)
}, },
sampleFunc: func(ts int64) chunks.Sample { sampleFunc: func(ts int64) chunks.Sample {
return sample{t: ts, fh: tsdbutil.GenerateTestFloatHistogram(int(ts))} return sample{t: ts, fh: tsdbutil.GenerateTestFloatHistogram(ts)}
}, },
}, },
"integer histogram counter resets": { "integer histogram counter resets": {
// Adding counter reset to all histograms means each histogram will have its own chunk. // Adding counter reset to all histograms means each histogram will have its own chunk.
appendFunc: func(app storage.Appender, ts int64, counterReset bool) (storage.SeriesRef, error) { appendFunc: func(app storage.Appender, ts int64, counterReset bool) (storage.SeriesRef, error) {
h := tsdbutil.GenerateTestHistogram(int(ts)) h := tsdbutil.GenerateTestHistogram(ts)
h.CounterResetHint = histogram.CounterReset // For this scenario, ignore the counterReset argument. h.CounterResetHint = histogram.CounterReset // For this scenario, ignore the counterReset argument.
return app.AppendHistogram(0, labels.FromStrings("foo", "bar1"), ts, h, nil) return app.AppendHistogram(0, labels.FromStrings("foo", "bar1"), ts, h, nil)
}, },
sampleFunc: func(ts int64) chunks.Sample { sampleFunc: func(ts int64) chunks.Sample {
return sample{t: ts, h: tsdbutil.GenerateTestHistogram(int(ts))} return sample{t: ts, h: tsdbutil.GenerateTestHistogram(ts)}
}, },
}, },
"integer histogram with recode": { "integer histogram with recode": {
@ -6908,7 +6908,7 @@ func TestOOOHistogramCompactionWithCounterResets(t *testing.T) {
app := db.Appender(context.Background()) app := db.Appender(context.Background())
tsMs := ts * time.Minute.Milliseconds() tsMs := ts * time.Minute.Milliseconds()
if floatHistogram { if floatHistogram {
h := tsdbutil.GenerateTestFloatHistogram(val) h := tsdbutil.GenerateTestFloatHistogram(int64(val))
h.CounterResetHint = hint h.CounterResetHint = hint
_, err = app.AppendHistogram(0, l, tsMs, nil, h) _, err = app.AppendHistogram(0, l, tsMs, nil, h)
require.NoError(t, err) require.NoError(t, err)
@ -6916,7 +6916,7 @@ func TestOOOHistogramCompactionWithCounterResets(t *testing.T) {
return sample{t: tsMs, fh: h.Copy()} return sample{t: tsMs, fh: h.Copy()}
} }
h := tsdbutil.GenerateTestHistogram(val) h := tsdbutil.GenerateTestHistogram(int64(val))
h.CounterResetHint = hint h.CounterResetHint = hint
_, err = app.AppendHistogram(0, l, tsMs, h, nil) _, err = app.AppendHistogram(0, l, tsMs, h, nil)
require.NoError(t, err) require.NoError(t, err)
@ -7267,14 +7267,14 @@ func TestInterleavedInOrderAndOOOHistogramCompactionWithCounterResets(t *testing
app := db.Appender(context.Background()) app := db.Appender(context.Background())
tsMs := ts tsMs := ts
if floatHistogram { if floatHistogram {
h := tsdbutil.GenerateTestFloatHistogram(val) h := tsdbutil.GenerateTestFloatHistogram(int64(val))
_, err = app.AppendHistogram(0, l, tsMs, nil, h) _, err = app.AppendHistogram(0, l, tsMs, nil, h)
require.NoError(t, err) require.NoError(t, err)
require.NoError(t, app.Commit()) require.NoError(t, app.Commit())
return sample{t: tsMs, fh: h.Copy()} return sample{t: tsMs, fh: h.Copy()}
} }
h := tsdbutil.GenerateTestHistogram(val) h := tsdbutil.GenerateTestHistogram(int64(val))
_, err = app.AppendHistogram(0, l, tsMs, h, nil) _, err = app.AppendHistogram(0, l, tsMs, h, nil)
require.NoError(t, err) require.NoError(t, err)
require.NoError(t, app.Commit()) require.NoError(t, app.Commit())

4
tsdb/head_test.go
View file

@ -4732,7 +4732,7 @@ func TestOOOHistogramCounterResetHeaders(t *testing.T) {
// OOO histogram // OOO histogram
for i := 1; i <= 5; i++ { for i := 1; i <= 5; i++ {
appendHistogram(100+int64(i), tsdbutil.GenerateTestHistogram(1000+i)) appendHistogram(100+int64(i), tsdbutil.GenerateTestHistogram(1000+int64(i)))
} }
// Nothing mmapped yet. // Nothing mmapped yet.
checkOOOExpCounterResetHeader() checkOOOExpCounterResetHeader()
@ -4820,7 +4820,7 @@ func TestOOOHistogramCounterResetHeaders(t *testing.T) {
appendHistogram(300, tsdbutil.SetHistogramCounterReset(tsdbutil.GenerateTestHistogram(3000))) appendHistogram(300, tsdbutil.SetHistogramCounterReset(tsdbutil.GenerateTestHistogram(3000)))
for i := 1; i <= 4; i++ { for i := 1; i <= 4; i++ {
appendHistogram(300+int64(i), tsdbutil.GenerateTestHistogram(3000+i)) appendHistogram(300+int64(i), tsdbutil.GenerateTestHistogram(3000+int64(i)))
} }
// One mmapped chunk with (ts, val) [(300, 3000), (301, 3001), (302, 3002), (303, 3003), (350, 4000)]. // One mmapped chunk with (ts, val) [(300, 3000), (301, 3001), (302, 3002), (303, 3003), (350, 4000)].

8
tsdb/ooo_head_test.go
View file

@ -54,12 +54,12 @@ func TestOOOInsert(t *testing.T) {
}, },
"integer histogram": { "integer histogram": {
sampleFunc: func(ts int64) sample { sampleFunc: func(ts int64) sample {
return sample{t: ts, h: tsdbutil.GenerateTestHistogram(int(ts))} return sample{t: ts, h: tsdbutil.GenerateTestHistogram(ts)}
}, },
}, },
"float histogram": { "float histogram": {
sampleFunc: func(ts int64) sample { sampleFunc: func(ts int64) sample {
return sample{t: ts, fh: tsdbutil.GenerateTestFloatHistogram(int(ts))} return sample{t: ts, fh: tsdbutil.GenerateTestFloatHistogram(ts)}
}, },
}, },
} }
@ -118,12 +118,12 @@ func TestOOOInsertDuplicate(t *testing.T) {
}, },
"integer histogram": { "integer histogram": {
sampleFunc: func(ts int64) sample { sampleFunc: func(ts int64) sample {
return sample{t: ts, h: tsdbutil.GenerateTestHistogram(int(ts))} return sample{t: ts, h: tsdbutil.GenerateTestHistogram(ts)}
}, },
}, },
"float histogram": { "float histogram": {
sampleFunc: func(ts int64) sample { sampleFunc: func(ts int64) sample {
return sample{t: ts, fh: tsdbutil.GenerateTestFloatHistogram(int(ts))} return sample{t: ts, fh: tsdbutil.GenerateTestFloatHistogram(ts)}
}, },
}, },
} }

12
tsdb/querier_test.go
View file

@ -1826,12 +1826,12 @@ func checkCurrVal(t *testing.T, valType chunkenc.ValueType, it *populateWithDelS
ts, h := it.AtHistogram(nil) ts, h := it.AtHistogram(nil)
require.Equal(t, int64(expectedTs), ts) require.Equal(t, int64(expectedTs), ts)
h.CounterResetHint = histogram.UnknownCounterReset h.CounterResetHint = histogram.UnknownCounterReset
require.Equal(t, tsdbutil.GenerateTestHistogram(expectedValue), h) require.Equal(t, tsdbutil.GenerateTestHistogram(int64(expectedValue)), h)
case chunkenc.ValFloatHistogram: case chunkenc.ValFloatHistogram:
ts, h := it.AtFloatHistogram(nil) ts, h := it.AtFloatHistogram(nil)
require.Equal(t, int64(expectedTs), ts) require.Equal(t, int64(expectedTs), ts)
h.CounterResetHint = histogram.UnknownCounterReset h.CounterResetHint = histogram.UnknownCounterReset
require.Equal(t, tsdbutil.GenerateTestFloatHistogram(expectedValue), h) require.Equal(t, tsdbutil.GenerateTestFloatHistogram(int64(expectedValue)), h)
default: default:
panic("unexpected value type") panic("unexpected value type")
} }
@ -3579,16 +3579,16 @@ func TestQueryWithDeletedHistograms(t *testing.T) {
ctx := context.Background() ctx := context.Background()
testcases := map[string]func(int) (*histogram.Histogram, *histogram.FloatHistogram){ testcases := map[string]func(int) (*histogram.Histogram, *histogram.FloatHistogram){
"intCounter": func(i int) (*histogram.Histogram, *histogram.FloatHistogram) { "intCounter": func(i int) (*histogram.Histogram, *histogram.FloatHistogram) {
return tsdbutil.GenerateTestHistogram(i), nil return tsdbutil.GenerateTestHistogram(int64(i)), nil
}, },
"intgauge": func(i int) (*histogram.Histogram, *histogram.FloatHistogram) { "intgauge": func(i int) (*histogram.Histogram, *histogram.FloatHistogram) {
return tsdbutil.GenerateTestGaugeHistogram(rand.Int() % 1000), nil return tsdbutil.GenerateTestGaugeHistogram(rand.Int63() % 1000), nil
}, },
"floatCounter": func(i int) (*histogram.Histogram, *histogram.FloatHistogram) { "floatCounter": func(i int) (*histogram.Histogram, *histogram.FloatHistogram) {
return nil, tsdbutil.GenerateTestFloatHistogram(i) return nil, tsdbutil.GenerateTestFloatHistogram(int64(i))
}, },
"floatGauge": func(i int) (*histogram.Histogram, *histogram.FloatHistogram) { "floatGauge": func(i int) (*histogram.Histogram, *histogram.FloatHistogram) {
return nil, tsdbutil.GenerateTestGaugeFloatHistogram(rand.Int() % 1000) return nil, tsdbutil.GenerateTestGaugeFloatHistogram(rand.Int63() % 1000)
}, },
} }

16
tsdb/testutil.go
View file

@ -63,45 +63,45 @@ var sampleTypeScenarios = map[string]sampleTypeScenario{
intHistogram: { intHistogram: {
sampleType: sampleMetricTypeHistogram, sampleType: sampleMetricTypeHistogram,
appendFunc: func(appender storage.Appender, lbls labels.Labels, ts, value int64) (storage.SeriesRef, sample, error) { appendFunc: func(appender storage.Appender, lbls labels.Labels, ts, value int64) (storage.SeriesRef, sample, error) {
s := sample{t: ts, h: tsdbutil.GenerateTestHistogram(int(value))} s := sample{t: ts, h: tsdbutil.GenerateTestHistogram(value)}
ref, err := appender.AppendHistogram(0, lbls, ts, s.h, nil) ref, err := appender.AppendHistogram(0, lbls, ts, s.h, nil)
return ref, s, err return ref, s, err
}, },
sampleFunc: func(ts, value int64) sample { sampleFunc: func(ts, value int64) sample {
return sample{t: ts, h: tsdbutil.GenerateTestHistogram(int(value))} return sample{t: ts, h: tsdbutil.GenerateTestHistogram(value)}
}, },
}, },
floatHistogram: { floatHistogram: {
sampleType: sampleMetricTypeHistogram, sampleType: sampleMetricTypeHistogram,
appendFunc: func(appender storage.Appender, lbls labels.Labels, ts, value int64) (storage.SeriesRef, sample, error) { appendFunc: func(appender storage.Appender, lbls labels.Labels, ts, value int64) (storage.SeriesRef, sample, error) {
s := sample{t: ts, fh: tsdbutil.GenerateTestFloatHistogram(int(value))} s := sample{t: ts, fh: tsdbutil.GenerateTestFloatHistogram(value)}
ref, err := appender.AppendHistogram(0, lbls, ts, nil, s.fh) ref, err := appender.AppendHistogram(0, lbls, ts, nil, s.fh)
return ref, s, err return ref, s, err
}, },
sampleFunc: func(ts, value int64) sample { sampleFunc: func(ts, value int64) sample {
return sample{t: ts, fh: tsdbutil.GenerateTestFloatHistogram(int(value))} return sample{t: ts, fh: tsdbutil.GenerateTestFloatHistogram(value)}
}, },
}, },
gaugeIntHistogram: { gaugeIntHistogram: {
sampleType: sampleMetricTypeHistogram, sampleType: sampleMetricTypeHistogram,
appendFunc: func(appender storage.Appender, lbls labels.Labels, ts, value int64) (storage.SeriesRef, sample, error) { appendFunc: func(appender storage.Appender, lbls labels.Labels, ts, value int64) (storage.SeriesRef, sample, error) {
s := sample{t: ts, h: tsdbutil.GenerateTestGaugeHistogram(int(value))} s := sample{t: ts, h: tsdbutil.GenerateTestGaugeHistogram(value)}
ref, err := appender.AppendHistogram(0, lbls, ts, s.h, nil) ref, err := appender.AppendHistogram(0, lbls, ts, s.h, nil)
return ref, s, err return ref, s, err
}, },
sampleFunc: func(ts, value int64) sample { sampleFunc: func(ts, value int64) sample {
return sample{t: ts, h: tsdbutil.GenerateTestGaugeHistogram(int(value))} return sample{t: ts, h: tsdbutil.GenerateTestGaugeHistogram(value)}
}, },
}, },
gaugeFloatHistogram: { gaugeFloatHistogram: {
sampleType: sampleMetricTypeHistogram, sampleType: sampleMetricTypeHistogram,
appendFunc: func(appender storage.Appender, lbls labels.Labels, ts, value int64) (storage.SeriesRef, sample, error) { appendFunc: func(appender storage.Appender, lbls labels.Labels, ts, value int64) (storage.SeriesRef, sample, error) {
s := sample{t: ts, fh: tsdbutil.GenerateTestGaugeFloatHistogram(int(value))} s := sample{t: ts, fh: tsdbutil.GenerateTestGaugeFloatHistogram(value)}
ref, err := appender.AppendHistogram(0, lbls, ts, nil, s.fh) ref, err := appender.AppendHistogram(0, lbls, ts, nil, s.fh)
return ref, s, err return ref, s, err
}, },
sampleFunc: func(ts, value int64) sample { sampleFunc: func(ts, value int64) sample {
return sample{t: ts, fh: tsdbutil.GenerateTestGaugeFloatHistogram(int(value))} return sample{t: ts, fh: tsdbutil.GenerateTestGaugeFloatHistogram(value)}
}, },
}, },
} }

28
tsdb/tsdbutil/histogram.go
View file

@ -21,7 +21,7 @@ import (
func GenerateTestHistograms(n int) (r []*histogram.Histogram) { func GenerateTestHistograms(n int) (r []*histogram.Histogram) {
for i := 0; i < n; i++ { for i := 0; i < n; i++ {
h := GenerateTestHistogram(i) h := GenerateTestHistogram(int64(i))
if i > 0 { if i > 0 {
h.CounterResetHint = histogram.NotCounterReset h.CounterResetHint = histogram.NotCounterReset
} }
@ -31,13 +31,13 @@ func GenerateTestHistograms(n int) (r []*histogram.Histogram) {
} }
func GenerateTestHistogramWithHint(n int, hint histogram.CounterResetHint) *histogram.Histogram { func GenerateTestHistogramWithHint(n int, hint histogram.CounterResetHint) *histogram.Histogram {
h := GenerateTestHistogram(n) h := GenerateTestHistogram(int64(n))
h.CounterResetHint = hint h.CounterResetHint = hint
return h return h
} }
// GenerateTestHistogram but it is up to the user to set any known counter reset hint. // GenerateTestHistogram but it is up to the user to set any known counter reset hint.
func GenerateTestHistogram(i int) *histogram.Histogram { func GenerateTestHistogram(i int64) *histogram.Histogram {
return &histogram.Histogram{ return &histogram.Histogram{
Count: 12 + uint64(i*9), Count: 12 + uint64(i*9),
ZeroCount: 2 + uint64(i), ZeroCount: 2 + uint64(i),
@ -48,16 +48,16 @@ func GenerateTestHistogram(i int) *histogram.Histogram {
{Offset: 0, Length: 2}, {Offset: 0, Length: 2},
{Offset: 1, Length: 2}, {Offset: 1, Length: 2},
}, },
PositiveBuckets: []int64{int64(i + 1), 1, -1, 0}, PositiveBuckets: []int64{i + 1, 1, -1, 0},
NegativeSpans: []histogram.Span{ NegativeSpans: []histogram.Span{
{Offset: 0, Length: 2}, {Offset: 0, Length: 2},
{Offset: 1, Length: 2}, {Offset: 1, Length: 2},
}, },
NegativeBuckets: []int64{int64(i + 1), 1, -1, 0}, NegativeBuckets: []int64{i + 1, 1, -1, 0},
} }
} }
func GenerateTestCustomBucketsHistogram(i int) *histogram.Histogram { func GenerateTestCustomBucketsHistogram(i int64) *histogram.Histogram {
return &histogram.Histogram{ return &histogram.Histogram{
Count: 5 + uint64(i*4), Count: 5 + uint64(i*4),
Sum: 18.4 * float64(i+1), Sum: 18.4 * float64(i+1),
@ -66,20 +66,20 @@ func GenerateTestCustomBucketsHistogram(i int) *histogram.Histogram {
{Offset: 0, Length: 2}, {Offset: 0, Length: 2},
{Offset: 1, Length: 2}, {Offset: 1, Length: 2},
}, },
PositiveBuckets: []int64{int64(i + 1), 1, -1, 0}, PositiveBuckets: []int64{i + 1, 1, -1, 0},
CustomValues: []float64{0, 1, 2, 3, 4}, CustomValues: []float64{0, 1, 2, 3, 4},
} }
} }
func GenerateTestGaugeHistograms(n int) (r []*histogram.Histogram) { func GenerateTestGaugeHistograms(n int) (r []*histogram.Histogram) {
for x := 0; x < n; x++ { for x := 0; x < n; x++ {
i := int(math.Sin(float64(x))*100) + 100 i := int64(math.Sin(float64(x))*100) + 100
r = append(r, GenerateTestGaugeHistogram(i)) r = append(r, GenerateTestGaugeHistogram(i))
} }
return r return r
} }
func GenerateTestGaugeHistogram(i int) *histogram.Histogram { func GenerateTestGaugeHistogram(i int64) *histogram.Histogram {
h := GenerateTestHistogram(i) h := GenerateTestHistogram(i)
h.CounterResetHint = histogram.GaugeType h.CounterResetHint = histogram.GaugeType
return h return h
@ -87,7 +87,7 @@ func GenerateTestGaugeHistogram(i int) *histogram.Histogram {
func GenerateTestFloatHistograms(n int) (r []*histogram.FloatHistogram) { func GenerateTestFloatHistograms(n int) (r []*histogram.FloatHistogram) {
for i := 0; i < n; i++ { for i := 0; i < n; i++ {
h := GenerateTestFloatHistogram(i) h := GenerateTestFloatHistogram(int64(i))
if i > 0 { if i > 0 {
h.CounterResetHint = histogram.NotCounterReset h.CounterResetHint = histogram.NotCounterReset
} }
@ -97,7 +97,7 @@ func GenerateTestFloatHistograms(n int) (r []*histogram.FloatHistogram) {
} }
// GenerateTestFloatHistogram but it is up to the user to set any known counter reset hint. // GenerateTestFloatHistogram but it is up to the user to set any known counter reset hint.
func GenerateTestFloatHistogram(i int) *histogram.FloatHistogram { func GenerateTestFloatHistogram(i int64) *histogram.FloatHistogram {
return &histogram.FloatHistogram{ return &histogram.FloatHistogram{
Count: 12 + float64(i*9), Count: 12 + float64(i*9),
ZeroCount: 2 + float64(i), ZeroCount: 2 + float64(i),
@ -117,7 +117,7 @@ func GenerateTestFloatHistogram(i int) *histogram.FloatHistogram {
} }
} }
func GenerateTestCustomBucketsFloatHistogram(i int) *histogram.FloatHistogram { func GenerateTestCustomBucketsFloatHistogram(i int64) *histogram.FloatHistogram {
return &histogram.FloatHistogram{ return &histogram.FloatHistogram{
Count: 5 + float64(i*4), Count: 5 + float64(i*4),
Sum: 18.4 * float64(i+1), Sum: 18.4 * float64(i+1),
@ -133,13 +133,13 @@ func GenerateTestCustomBucketsFloatHistogram(i int) *histogram.FloatHistogram {
func GenerateTestGaugeFloatHistograms(n int) (r []*histogram.FloatHistogram) { func GenerateTestGaugeFloatHistograms(n int) (r []*histogram.FloatHistogram) {
for x := 0; x < n; x++ { for x := 0; x < n; x++ {
i := int(math.Sin(float64(x))*100) + 100 i := int64(math.Sin(float64(x))*100) + 100
r = append(r, GenerateTestGaugeFloatHistogram(i)) r = append(r, GenerateTestGaugeFloatHistogram(i))
} }
return r return r
} }
func GenerateTestGaugeFloatHistogram(i int) *histogram.FloatHistogram { func GenerateTestGaugeFloatHistogram(i int64) *histogram.FloatHistogram {
h := GenerateTestFloatHistogram(i) h := GenerateTestFloatHistogram(i)
h.CounterResetHint = histogram.GaugeType h.CounterResetHint = histogram.GaugeType
return h return h