Skip to content

Commit

Permalink
fix review
Browse files Browse the repository at this point in the history
  • Loading branch information
veceraj committed Feb 18, 2025
1 parent 92256b8 commit bb79546
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions pbm/storage/s3/retryer.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ package s3
import (
"context"
"time"

"github.com/aws/aws-sdk-go-v2/aws"
"github.com/aws/aws-sdk-go-v2/aws/retry"
)
Expand Down Expand Up @@ -37,15 +37,15 @@ func (r CustomRetryer) RetryDelay(attempt int, opErr error) (time.Duration, erro
return delay, nil
}

func (r CustomRetryer) GetRetryToken(ctx context.Context, opErr error) (releaseToken func(error) error, err error) {
func (r CustomRetryer) GetRetryToken(ctx context.Context, opErr error) (func(error) error, error) {
return r.base.GetRetryToken(ctx, opErr)
}

func (r CustomRetryer) GetInitialToken() (releaseToken func(error) error) {
func (r CustomRetryer) GetInitialToken() func(error) error {
return r.base.GetInitialToken()
}

func NewCustomRetryer(numMaxRetries int, minBackoff time.Duration, maxBackoff time.Duration) aws.Retryer {
func NewCustomRetryer(numMaxRetries int, minBackoff, maxBackoff time.Duration) aws.Retryer {
baseRetryer := retry.NewStandard(func(o *retry.StandardOptions) {
o.MaxAttempts = numMaxRetries
o.MaxBackoff = maxBackoff
Expand Down

0 comments on commit bb79546

Please sign in to comment.