Skip to content

Commit 4d28f31

Browse files
committed
Move IsOneOfTheExpectedErrors to bucket package
Signed-off-by: SungJin1212 <tjdwls1201@gmail.com>
1 parent 919e0d1 commit 4d28f31

File tree

6 files changed

+16
-20
lines changed

6 files changed

+16
-20
lines changed

pkg/alertmanager/alertstore/bucketclient/bucket_client.go

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@ import (
1818
"github.com/cortexproject/cortex/pkg/alertmanager/alertspb"
1919
"github.com/cortexproject/cortex/pkg/storage/bucket"
2020
"github.com/cortexproject/cortex/pkg/util/concurrency"
21-
cortex_errors "github.com/cortexproject/cortex/pkg/util/errors"
2221
"github.com/cortexproject/cortex/pkg/util/runutil"
2322
"github.com/cortexproject/cortex/pkg/util/users"
2423
)
@@ -229,5 +228,5 @@ func (s *BucketAlertStore) getUserBucket(userID string) objstore.Bucket {
229228

230229
func (s *BucketAlertStore) getAlertmanagerUserBucket(userID string) objstore.Bucket {
231230
uBucket := bucket.NewUserBucketClient(userID, s.amBucket, s.cfgProvider)
232-
return uBucket.WithExpectedErrs(cortex_errors.IsOneOfTheExpectedErrors(uBucket.IsAccessDeniedErr, uBucket.IsObjNotFoundErr))
231+
return uBucket.WithExpectedErrs(bucket.IsOneOfTheExpectedErrors(uBucket.IsAccessDeniedErr, uBucket.IsObjNotFoundErr))
233232
}

pkg/storage/bucket/bucket_util.go

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,3 +49,14 @@ func ListPrefixes(ctx context.Context, bkt objstore.Bucket, prefix string, logge
4949
})
5050
return keys, err
5151
}
52+
53+
func IsOneOfTheExpectedErrors(f ...objstore.IsOpFailureExpectedFunc) objstore.IsOpFailureExpectedFunc {
54+
return func(err error) bool {
55+
for _, f := range f {
56+
if f(err) {
57+
return true
58+
}
59+
}
60+
return false
61+
}
62+
}

pkg/storage/parquet/converter_marker.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ import (
1313
"github.com/thanos-io/objstore"
1414
"github.com/thanos-io/thanos/pkg/runutil"
1515

16-
cortex_errors "github.com/cortexproject/cortex/pkg/util/errors"
16+
"github.com/cortexproject/cortex/pkg/storage/bucket"
1717
)
1818

1919
const (
@@ -28,7 +28,7 @@ type ConverterMark struct {
2828

2929
func ReadConverterMark(ctx context.Context, id ulid.ULID, userBkt objstore.InstrumentedBucket, logger log.Logger) (*ConverterMark, error) {
3030
markerPath := path.Join(id.String(), ConverterMarkerFileName)
31-
reader, err := userBkt.WithExpectedErrs(cortex_errors.IsOneOfTheExpectedErrors(userBkt.IsAccessDeniedErr, userBkt.IsObjNotFoundErr)).Get(ctx, markerPath)
31+
reader, err := userBkt.WithExpectedErrs(bucket.IsOneOfTheExpectedErrors(userBkt.IsAccessDeniedErr, userBkt.IsObjNotFoundErr)).Get(ctx, markerPath)
3232
if err != nil {
3333
if userBkt.IsObjNotFoundErr(err) || userBkt.IsAccessDeniedErr(err) {
3434
return &ConverterMark{}, nil

pkg/storage/tsdb/bucketindex/storage.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ func ReadIndex(ctx context.Context, bkt objstore.Bucket, userID string, cfgProvi
6969
userBkt := bucket.NewUserBucketClient(userID, bkt, cfgProvider)
7070

7171
// Get the bucket index.
72-
reader, err := userBkt.WithExpectedErrs(cortex_errors.IsOneOfTheExpectedErrors(userBkt.IsAccessDeniedErr, userBkt.IsObjNotFoundErr)).Get(ctx, IndexCompressedFilename)
72+
reader, err := userBkt.WithExpectedErrs(bucket.IsOneOfTheExpectedErrors(userBkt.IsAccessDeniedErr, userBkt.IsObjNotFoundErr)).Get(ctx, IndexCompressedFilename)
7373
if err != nil {
7474
if userBkt.IsObjNotFoundErr(err) {
7575
return nil, ErrIndexNotFound

pkg/storage/tsdb/bucketindex/updater.go

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@ import (
1717

1818
"github.com/cortexproject/cortex/pkg/storage/bucket"
1919
"github.com/cortexproject/cortex/pkg/storage/parquet"
20-
cortex_errors "github.com/cortexproject/cortex/pkg/util/errors"
2120
util_log "github.com/cortexproject/cortex/pkg/util/log"
2221
"github.com/cortexproject/cortex/pkg/util/runutil"
2322
)
@@ -150,7 +149,7 @@ func (w *Updater) updateBlockIndexEntry(ctx context.Context, id ulid.ULID) (*Blo
150149
metaFile := path.Join(id.String(), block.MetaFilename)
151150

152151
// Get the block's meta.json file.
153-
r, err := w.bkt.ReaderWithExpectedErrs(cortex_errors.IsOneOfTheExpectedErrors(w.bkt.IsObjNotFoundErr, w.bkt.IsAccessDeniedErr)).Get(ctx, metaFile)
152+
r, err := w.bkt.ReaderWithExpectedErrs(bucket.IsOneOfTheExpectedErrors(w.bkt.IsObjNotFoundErr, w.bkt.IsAccessDeniedErr)).Get(ctx, metaFile)
154153
if w.bkt.IsObjNotFoundErr(err) {
155154
return nil, ErrBlockMetaNotFound
156155
}

pkg/util/errors/errors.go

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,6 @@ package errors
22

33
import (
44
"errors"
5-
6-
"github.com/thanos-io/objstore"
75
)
86

97
type errWithCause struct {
@@ -54,14 +52,3 @@ func ErrorIs(err error, f func(err error) bool) bool {
5452
}
5553
}
5654
}
57-
58-
func IsOneOfTheExpectedErrors(f ...objstore.IsOpFailureExpectedFunc) objstore.IsOpFailureExpectedFunc {
59-
return func(err error) bool {
60-
for _, f := range f {
61-
if f(err) {
62-
return true
63-
}
64-
}
65-
return false
66-
}
67-
}

0 commit comments

Comments
 (0)