aws_service.rb in cfn-status-0.4.5

- old
+ new

@@ -1,11 +1,11 @@
 require "aws-sdk-cloudformation"
 
 class CfnStatus
   module AwsService
     def cfn
-      @cfn ||= Aws::CloudFormation::Client.new
+      @cfn ||= Aws::CloudFormation::Client.new(aws_options)
     end
 
     def stack_exists?(stack_name)
       return true if ENV['TEST']
       return false if @options[:noop]
@@ -44,8 +44,39 @@
       end
     end
 
     def rollback_complete?(stack)
       stack&.stack_status == 'ROLLBACK_COMPLETE'
+    end
+
+    # Override the AWS retry settings.
+    #
+    # The aws-sdk-core has exponential backup with this formula:
+    #
+    #   2 ** c.retries * c.config.retry_base_delay
+    #
+    # Source:
+    #   https://github.com/aws/aws-sdk-ruby/blob/version-3/gems/aws-sdk-core/lib/aws-sdk-core/plugins/retry_errors.rb
+    #
+    # So the max delay will be 2 ** 7 * 0.6 = 76.8s
+    #
+    # Only scoping this to deploy because dont want to affect people's application that use the aws sdk.
+    #
+    # There is also additional rate backoff logic elsewhere, since this is only scoped to deploys.
+    #
+    # Useful links:
+    #   https://github.com/aws/aws-sdk-ruby/blob/master/gems/aws-sdk-core/lib/aws-sdk-core/plugins/retry_errors.rb
+    #   https://docs.aws.amazon.com/apigateway/latest/developerguide/limits.html
+    #
+    def aws_options
+      options = {
+        retry_limit: 7, # default: 3
+        retry_base_delay: 0.6, # default: 0.3
+      }
+      options.merge!(
+        log_level: :debug,
+        logger: Logger.new($stdout),
+      ) if ENV['CFN_STATUS_DEBUG_AWS_SDK']
+      options
     end
   end
 end