Merge pull request #131 from movableink/arajeev/sc-162448/add-autoscaling-client-retries

add autoscaling client with retries [sc-162448]

Author: alecrajeev

Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    MovableInkAWS (2.11.0)
+    MovableInkAWS (2.11.2)
       aws-sdk-athena (~> 1)
       aws-sdk-autoscaling (~> 1)
       aws-sdk-cloudwatch (~> 1)

lib/movable_ink/aws/autoscaling.rb

@@ -8,6 +8,12 @@ def autoscaling(region: my_region)
         @autoscaling_client[region] ||= Aws::AutoScaling::Client.new(region: region)
       end
 
+      def autoscaling_with_retries(region: my_region)
+        @autoscaling_client_with_retries ||= {}
+        instance_credentials = Aws::InstanceProfileCredentials.new(retries: 5, disable_imds_v1: true)
+        @autoscaling_client_with_retries[region] ||= Aws::AutoScaling::Client.new(region: region, credentials: instance_credentials)
+      end
+
       def mark_me_as_unhealthy
         run_with_backoff do
           autoscaling.set_instance_health({
@@ -84,6 +90,30 @@ def complete_lifecycle_action(lifecycle_hook_name:, auto_scaling_group_name:, li
         end
       end
 
+      def complete_lifecycle_action_with_retries(lifecycle_hook_name:, auto_scaling_group_name:, lifecycle_action_token: nil, instance_id: nil)
+        raise ArgumentError.new('lifecycle_action_token or instance_id required') if lifecycle_action_token.nil? && instance_id.nil?
+
+        if lifecycle_action_token
+          run_with_backoff do
+            autoscaling_with_retries.complete_lifecycle_action({
+              lifecycle_hook_name:     lifecycle_hook_name,
+              auto_scaling_group_name: auto_scaling_group_name,
+              lifecycle_action_token:  lifecycle_action_token,
+              lifecycle_action_result: 'CONTINUE'
+            })
+          end
+        else
+          run_with_backoff do
+            autoscaling_with_retries.complete_lifecycle_action({
+              instance_id:             instance_id,
+              lifecycle_hook_name:     lifecycle_hook_name,
+              auto_scaling_group_name: auto_scaling_group_name,
+              lifecycle_action_result: 'CONTINUE'
+            })
+          end
+        end
+      end
+
       def record_lifecycle_action_heartbeat(lifecycle_hook_name:, auto_scaling_group_name:, lifecycle_action_token: nil, instance_id: nil)
         raise ArgumentError.new('lifecycle_action_token or instance_id required') if lifecycle_action_token.nil? && instance_id.nil?
 

lib/movable_ink/version.rb

@@ -1,5 +1,5 @@
 module MovableInk
   class AWS
-    VERSION = '2.11.0'
+    VERSION = '2.11.2'
   end
 end