diff --git a/wal/wal.go b/wal/wal.go index 0e95ba2a7..1aea24d86 100644 --- a/wal/wal.go +++ b/wal/wal.go @@ -255,15 +255,17 @@ Loop: // Repair attempts to repair the WAL based on the error. // It discards all data after the corruption. -func (w *WAL) Repair(err error) error { +func (w *WAL) Repair(origErr error) error { // We could probably have a mode that only discards torn records right around // the corruption to preserve as data much as possible. // But that's not generally applicable if the records have any kind of causality. // Maybe as an extra mode in the future if mid-WAL corruptions become // a frequent concern. + err := errors.Cause(origErr) // So that we can pick up errors even if wrapped. + cerr, ok := err.(*CorruptionErr) if !ok { - return errors.New("cannot handle error") + return errors.Wrap(origErr, "cannot handle error") } if cerr.Segment < 0 { return errors.New("corruption error does not specify position") diff --git a/wal/wal_test.go b/wal/wal_test.go index d1b724c7e..26ee8663d 100644 --- a/wal/wal_test.go +++ b/wal/wal_test.go @@ -23,6 +23,7 @@ import ( "os" "testing" + "github.com/pkg/errors" "github.com/prometheus/tsdb/testutil" ) @@ -286,8 +287,8 @@ func TestWAL_Repair(t *testing.T) { for r.Next() { } testutil.NotOk(t, r.Err()) - testutil.Ok(t, w.Repair(r.Err())) + testutil.Ok(t, w.Repair(errors.Wrap(r.Err(), "err"))) // See https://github.com/prometheus/prometheus/issues/4603 sr, err = NewSegmentsReader(dir) testutil.Ok(t, err)