25 lines
506 B
Go
25 lines
506 B
Go
|
package ddb
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/aws/aws-sdk-go/aws/awserr"
|
||
|
|
"github.com/aws/aws-sdk-go/service/dynamodb"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Retryer interface contains one method that decides whether to retry based on error
|
||
|
|
type Retryer interface {
|
||
|
|
ShouldRetry(error) bool
|
||
|
|
}
|
||
|
|
|
||
|
|
type DefaultRetryer struct {
|
||
|
|
Retryer
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *DefaultRetryer) ShouldRetry(err error) bool {
|
||
|
|
if awsErr, ok := err.(awserr.Error); ok {
|
||
|
|
if awsErr.Code() == dynamodb.ErrCodeProvisionedThroughputExceededException {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return false
|
||
|
|
}
|