Module: MovableInk::AWS::Autoscaling

Included in:
MovableInk::AWS
Defined in:
lib/movable_ink/aws/autoscaling.rb

Instance Method Summary collapse

Instance Method Details

#autoscaling(region: my_region) ⇒ Object 



6
7
8
9
 
# File 'lib/movable_ink/aws/autoscaling.rb', line 6

def autoscaling(region: my_region)
  @autoscaling_client ||= {}
  @autoscaling_client[region] ||= Aws::AutoScaling::Client.new(region: region)
end

#autoscaling_with_retries(region: my_region) ⇒ Object 



11
12
13
14
15
 
# File 'lib/movable_ink/aws/autoscaling.rb', line 11

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

#complete_lifecycle_action(lifecycle_hook_name:, auto_scaling_group_name:, lifecycle_action_token: nil, instance_id: nil) ⇒ Object

Raises:

  • (ArgumentError) 


69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
 
# File 'lib/movable_ink/aws/autoscaling.rb', line 69

def complete_lifecycle_action(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.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.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

#complete_lifecycle_action_with_retries(lifecycle_hook_name:, auto_scaling_group_name:, lifecycle_action_token: nil, instance_id: nil) ⇒ Object

Raises:

  • (ArgumentError) 


93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
 
# File 'lib/movable_ink/aws/autoscaling.rb', line 93

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

#delete_role_tag(role:) ⇒ Object 



41
42
43
44
45
46
47
48
49
50
51
52
53
 
# File 'lib/movable_ink/aws/autoscaling.rb', line 41

def delete_role_tag(role:)
  run_with_backoff do
    ec2.delete_tags({
      resources: [instance_id],
      tags: [
        {
          key: "mi:roles",
          value: role
        }
      ]
    })
  end
end

#delete_role_tag_with_retries(role:) ⇒ Object 



55
56
57
58
59
60
61
62
63
64
65
66
67
 
# File 'lib/movable_ink/aws/autoscaling.rb', line 55

def delete_role_tag_with_retries(role:)
  run_with_backoff do
    ec2_with_retries.delete_tags({
      resources: [instance_id],
      tags: [
        {
          key: "mi:roles",
          value: role
        }
      ]
    })
  end
end

#keep_instance_alive(lifecycle_hook_name:, auto_scaling_group_name:, lifecycle_action_token: nil, instance_id: nil) ⇒ Object 



139
140
141
142
143
144
145
146
147
148
149
 
# File 'lib/movable_ink/aws/autoscaling.rb', line 139

def keep_instance_alive(lifecycle_hook_name:, auto_scaling_group_name:, lifecycle_action_token: nil, instance_id: nil)
  24.downto(1) do |hours|
    record_lifecycle_action_heartbeat(
      lifecycle_hook_name: lifecycle_hook_name,
      auto_scaling_group_name: auto_scaling_group_name,
      lifecycle_action_token: lifecycle_action_token,
      instance_id: instance_id
    )
    sleep 3600
  end
end

#mark_me_as_healthy(role:) ⇒ Object 



27
28
29
30
31
32
33
34
35
36
37
38
39
 
# File 'lib/movable_ink/aws/autoscaling.rb', line 27

def mark_me_as_healthy(role:)
  run_with_backoff do
    ec2.create_tags({
      resources: [instance_id],
      tags: [
        {
          key: "mi:roles",
          value: role
        }
      ]
    })
  end
end

#mark_me_as_unhealthy ⇒ Object 



17
18
19
20
21
22
23
24
25
 
# File 'lib/movable_ink/aws/autoscaling.rb', line 17

def mark_me_as_unhealthy
  run_with_backoff do
    autoscaling.set_instance_health({
      health_status: "Unhealthy",
      instance_id: instance_id,
      should_respect_grace_period: false
    })
  end
end

#record_lifecycle_action_heartbeat(lifecycle_hook_name:, auto_scaling_group_name:, lifecycle_action_token: nil, instance_id: nil) ⇒ Object

Raises:

  • (ArgumentError) 


117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
 
# File 'lib/movable_ink/aws/autoscaling.rb', line 117

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?

  if lifecycle_action_token
    run_with_backoff do
      autoscaling.record_lifecycle_action_heartbeat({
        lifecycle_hook_name:     lifecycle_hook_name,
        auto_scaling_group_name: auto_scaling_group_name,
        lifecycle_action_token:  lifecycle_action_token
      })
    end
  else
    run_with_backoff do
      autoscaling.record_lifecycle_action_heartbeat({
        instance_id:             instance_id,
        lifecycle_hook_name:     lifecycle_hook_name,
        auto_scaling_group_name: auto_scaling_group_name,
      })
    end
  end
end