dynamodb_retry.go

dynamodb_retry.go

var timeout time.Time = 30 * time.Millisecond
var maxRetry int = 3

func queryWithRetry(ctx context.Context, input *dynamodb.QueryInput) (*dynamodb.QueryOutput, error) {
	for i := 0; i < maxRetry; i++ {
		res, err := func() (*dynamodb.QueryOutput, error) {
			timeoutCtx, cancel := context.WithTimeout(ctx, timeout)
			defer cancel()
			return dynamoDB.Query(input)
		}()

		if isRetryable(err) {
			continue
		} else if err != nil {
			return nil, err
		}

		return res, nil
	}

	return nil, errors.New("max retry exceeded")
}