Merge pull request #13133 from mmorel-35/patch-8

tsdb/record: use Go standard errors package
pull/13138/head
Julien Pivotto 1 year ago committed by GitHub
commit 17fe6bc296
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -16,10 +16,10 @@
package record package record
import ( import (
"errors"
"fmt"
"math" "math"
"github.com/pkg/errors"
"github.com/prometheus/prometheus/model/histogram" "github.com/prometheus/prometheus/model/histogram"
"github.com/prometheus/prometheus/model/labels" "github.com/prometheus/prometheus/model/labels"
"github.com/prometheus/prometheus/model/textparse" "github.com/prometheus/prometheus/model/textparse"
@ -229,7 +229,7 @@ func (d *Decoder) Series(rec []byte, series []RefSeries) ([]RefSeries, error) {
return nil, dec.Err() return nil, dec.Err()
} }
if len(dec.B) > 0 { if len(dec.B) > 0 {
return nil, errors.Errorf("unexpected %d bytes left in entry", len(dec.B)) return nil, fmt.Errorf("unexpected %d bytes left in entry", len(dec.B))
} }
return series, nil return series, nil
} }
@ -272,7 +272,7 @@ func (d *Decoder) Metadata(rec []byte, metadata []RefMetadata) ([]RefMetadata, e
return nil, dec.Err() return nil, dec.Err()
} }
if len(dec.B) > 0 { if len(dec.B) > 0 {
return nil, errors.Errorf("unexpected %d bytes left in entry", len(dec.B)) return nil, fmt.Errorf("unexpected %d bytes left in entry", len(dec.B))
} }
return metadata, nil return metadata, nil
} }
@ -321,10 +321,10 @@ func (d *Decoder) Samples(rec []byte, samples []RefSample) ([]RefSample, error)
} }
if dec.Err() != nil { if dec.Err() != nil {
return nil, errors.Wrapf(dec.Err(), "decode error after %d samples", len(samples)) return nil, fmt.Errorf("decode error after %d samples: %w", len(samples), dec.Err())
} }
if len(dec.B) > 0 { if len(dec.B) > 0 {
return nil, errors.Errorf("unexpected %d bytes left in entry", len(dec.B)) return nil, fmt.Errorf("unexpected %d bytes left in entry", len(dec.B))
} }
return samples, nil return samples, nil
} }
@ -348,7 +348,7 @@ func (d *Decoder) Tombstones(rec []byte, tstones []tombstones.Stone) ([]tombston
return nil, dec.Err() return nil, dec.Err()
} }
if len(dec.B) > 0 { if len(dec.B) > 0 {
return nil, errors.Errorf("unexpected %d bytes left in entry", len(dec.B)) return nil, fmt.Errorf("unexpected %d bytes left in entry", len(dec.B))
} }
return tstones, nil return tstones, nil
} }
@ -386,10 +386,10 @@ func (d *Decoder) ExemplarsFromBuffer(dec *encoding.Decbuf, exemplars []RefExemp
} }
if dec.Err() != nil { if dec.Err() != nil {
return nil, errors.Wrapf(dec.Err(), "decode error after %d exemplars", len(exemplars)) return nil, fmt.Errorf("decode error after %d exemplars: %w", len(exemplars), dec.Err())
} }
if len(dec.B) > 0 { if len(dec.B) > 0 {
return nil, errors.Errorf("unexpected %d bytes left in entry", len(dec.B)) return nil, fmt.Errorf("unexpected %d bytes left in entry", len(dec.B))
} }
return exemplars, nil return exemplars, nil
} }
@ -414,10 +414,10 @@ func (d *Decoder) MmapMarkers(rec []byte, markers []RefMmapMarker) ([]RefMmapMar
} }
if dec.Err() != nil { if dec.Err() != nil {
return nil, errors.Wrapf(dec.Err(), "decode error after %d mmap markers", len(markers)) return nil, fmt.Errorf("decode error after %d mmap markers: %w", len(markers), dec.Err())
} }
if len(dec.B) > 0 { if len(dec.B) > 0 {
return nil, errors.Errorf("unexpected %d bytes left in entry", len(dec.B)) return nil, fmt.Errorf("unexpected %d bytes left in entry", len(dec.B))
} }
return markers, nil return markers, nil
} }
@ -450,10 +450,10 @@ func (d *Decoder) HistogramSamples(rec []byte, histograms []RefHistogramSample)
} }
if dec.Err() != nil { if dec.Err() != nil {
return nil, errors.Wrapf(dec.Err(), "decode error after %d histograms", len(histograms)) return nil, fmt.Errorf("decode error after %d histograms: %w", len(histograms), dec.Err())
} }
if len(dec.B) > 0 { if len(dec.B) > 0 {
return nil, errors.Errorf("unexpected %d bytes left in entry", len(dec.B)) return nil, fmt.Errorf("unexpected %d bytes left in entry", len(dec.B))
} }
return histograms, nil return histograms, nil
} }
@ -532,10 +532,10 @@ func (d *Decoder) FloatHistogramSamples(rec []byte, histograms []RefFloatHistogr
} }
if dec.Err() != nil { if dec.Err() != nil {
return nil, errors.Wrapf(dec.Err(), "decode error after %d histograms", len(histograms)) return nil, fmt.Errorf("decode error after %d histograms: %w", len(histograms), dec.Err())
} }
if len(dec.B) > 0 { if len(dec.B) > 0 {
return nil, errors.Errorf("unexpected %d bytes left in entry", len(dec.B)) return nil, fmt.Errorf("unexpected %d bytes left in entry", len(dec.B))
} }
return histograms, nil return histograms, nil
} }

@ -15,10 +15,10 @@
package record package record
import ( import (
"errors"
"math/rand" "math/rand"
"testing" "testing"
"github.com/pkg/errors"
"github.com/stretchr/testify/require" "github.com/stretchr/testify/require"
"github.com/prometheus/prometheus/model/histogram" "github.com/prometheus/prometheus/model/histogram"
@ -209,7 +209,7 @@ func TestRecord_Corrupted(t *testing.T) {
corrupted := enc.Samples(samples, nil)[:8] corrupted := enc.Samples(samples, nil)[:8]
_, err := dec.Samples(corrupted, nil) _, err := dec.Samples(corrupted, nil)
require.Equal(t, errors.Cause(err), encoding.ErrInvalidSize) require.True(t, errors.Is(err, encoding.ErrInvalidSize))
}) })
t.Run("Test corrupted tombstone record", func(t *testing.T) { t.Run("Test corrupted tombstone record", func(t *testing.T) {
@ -232,7 +232,7 @@ func TestRecord_Corrupted(t *testing.T) {
corrupted := enc.Exemplars(exemplars, nil)[:8] corrupted := enc.Exemplars(exemplars, nil)[:8]
_, err := dec.Exemplars(corrupted, nil) _, err := dec.Exemplars(corrupted, nil)
require.Equal(t, errors.Cause(err), encoding.ErrInvalidSize) require.True(t, errors.Is(err, encoding.ErrInvalidSize))
}) })
t.Run("Test corrupted metadata record", func(t *testing.T) { t.Run("Test corrupted metadata record", func(t *testing.T) {
@ -242,7 +242,7 @@ func TestRecord_Corrupted(t *testing.T) {
corrupted := enc.Metadata(meta, nil)[:8] corrupted := enc.Metadata(meta, nil)[:8]
_, err := dec.Metadata(corrupted, nil) _, err := dec.Metadata(corrupted, nil)
require.Equal(t, errors.Cause(err), encoding.ErrInvalidSize) require.True(t, errors.Is(err, encoding.ErrInvalidSize))
}) })
t.Run("Test corrupted histogram record", func(t *testing.T) { t.Run("Test corrupted histogram record", func(t *testing.T) {
@ -267,7 +267,7 @@ func TestRecord_Corrupted(t *testing.T) {
corrupted := enc.HistogramSamples(histograms, nil)[:8] corrupted := enc.HistogramSamples(histograms, nil)[:8]
_, err := dec.HistogramSamples(corrupted, nil) _, err := dec.HistogramSamples(corrupted, nil)
require.Equal(t, errors.Cause(err), encoding.ErrInvalidSize) require.True(t, errors.Is(err, encoding.ErrInvalidSize))
}) })
} }

Loading…
Cancel
Save