Module: PrimaAwsClient
- Defined in:
- lib/prima_aws_client.rb
Instance Method Summary collapse
- #artifact_exists?(bucket, path) ⇒ Boolean
- #asg_client ⇒ Object
- #cf_client ⇒ Object
- #cluster_list ⇒ Object
- #create_stack(stack_name, stack_body, parameters = [], tags = [], role = nil) ⇒ Object
- #delete_stack(stack_name) ⇒ Object
- #ec2_client ⇒ Object
- #get_autoscaling_capacity(asg_name) ⇒ Object
- #get_spotfleet_capacity(fleet_arn) ⇒ Object
- #get_stack_outputs(name) ⇒ Object
- #get_stack_parameters(name) ⇒ Object
- #get_stack_tags(name) ⇒ Object
- #get_stack_template(name) ⇒ Object
- #hashes_to_tags(hashes) ⇒ Object
- #list_exports ⇒ Object
- #list_import_stacks(export_name) ⇒ Object
- #s3_client ⇒ Object
- #stack_exists?(stack_name) ⇒ Boolean
- #stack_list ⇒ Object
- #stack_ready?(stack_name, failed_statuses = ['CREATE_FAILED', 'ROLLBACK_IN_PROGRESS', 'ROLLBACK_FAILED', 'DELETE_IN_PROGRESS', 'DELETE_FAILED', 'DELETE_COMPLETE', 'UPDATE_ROLLBACK_FAILED', 'UPDATE_ROLLBACK_COMPLETE_CLEANUP_IN_PROGRESS']) ⇒ Boolean
- #tags_to_hashes(tags) ⇒ Object
- #update_stack(stack_name, template_body, parameters = [], tags = [], role = nil) ⇒ Object
- #update_stack_url(stack_name, template_url, parameters = [], tags = [], role = nil) ⇒ Object
- #upload_artifact(source_path, destination_path, bucket_name_override = nil) ⇒ Object
- #wait_for_stack_ready(stack_name, failed_statuses = ['CREATE_FAILED', 'ROLLBACK_IN_PROGRESS', 'ROLLBACK_FAILED', 'DELETE_IN_PROGRESS', 'DELETE_FAILED', 'DELETE_COMPLETE', 'UPDATE_ROLLBACK_FAILED', 'UPDATE_ROLLBACK_COMPLETE_CLEANUP_IN_PROGRESS']) ⇒ Object
Instance Method Details
#artifact_exists?(bucket, path) ⇒ Boolean
270 271 272 273 274 275 276 277 |
# File 'lib/prima_aws_client.rb', line 270 def artifact_exists?(bucket, path) resp = s3_client.list_objects( bucket: bucket, max_keys: 1, prefix: path ) !resp.contents.empty? end |
#asg_client ⇒ Object
13 14 15 |
# File 'lib/prima_aws_client.rb', line 13 def asg_client @asg ||= Aws::AutoScaling::Client.new end |
#cf_client ⇒ Object
9 10 11 |
# File 'lib/prima_aws_client.rb', line 9 def cf_client @cf ||= Aws::CloudFormation::Client.new end |
#cluster_list ⇒ Object
43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
# File 'lib/prima_aws_client.rb', line 43 def cluster_list stacks = [] next_token = '' loop do print '.'.yellow; STDOUT.flush = next_token != '' ? { next_token: next_token } : {} begin resp = cf_client.describe_stacks() rescue Aws::CloudFormation::Errors::Throttling => e output 'Throttling, retrying in 15 seconds'.red sleep 15 resp = cf_client.describe_stacks() end stacks += resp.stacks break unless resp.next_token next_token = resp.next_token end puts '.'.yellow; STDOUT.flush stacks.keep_if { |stack| stack.stack_name.include? 'ecs-cluster-qa-' } stacks end |
#create_stack(stack_name, stack_body, parameters = [], tags = [], role = nil) ⇒ Object
86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 |
# File 'lib/prima_aws_client.rb', line 86 def create_stack(stack_name, stack_body, parameters = [], = [], role = nil) cf_args = { stack_name: stack_name, template_body: stack_body, parameters: parameters, tags: , capabilities: ['CAPABILITY_IAM'], on_failure: 'ROLLBACK' } if role != nil then cf_args.merge!(role_arn: role) end begin cf_client.create_stack(cf_args) rescue Aws::CloudFormation::Errors::Throttling, Aws::CloudFormation::Errors::LimitExcedeedException => e output 'Throttling, retrying in 15 seconds'.red sleep 15 create_stack(stack_name, stack_body, parameters = [], = []) else output "La creazione dello stack #{stack_name} è stata avviata".green end end |
#delete_stack(stack_name) ⇒ Object
178 179 180 181 182 183 184 185 186 187 188 |
# File 'lib/prima_aws_client.rb', line 178 def delete_stack(stack_name) begin cf_client.delete_stack(stack_name: stack_name) rescue Aws::CloudFormation::Errors::Throttling => e output 'Throttling, retrying in 15 seconds'.red sleep 15 delete_stack(stack_name) else output "Stack #{stack_name} spenta con successo\n".green end end |
#ec2_client ⇒ Object
17 18 19 |
# File 'lib/prima_aws_client.rb', line 17 def ec2_client @ec2 ||= Aws::EC2::Client.new end |
#get_autoscaling_capacity(asg_name) ⇒ Object
310 311 312 313 |
# File 'lib/prima_aws_client.rb', line 310 def get_autoscaling_capacity(asg_name) resp = asg_client.describe_auto_scaling_groups(auto_scaling_group_names: [asg_name]) resp.auto_scaling_groups[0].desired_capacity end |
#get_spotfleet_capacity(fleet_arn) ⇒ Object
315 316 317 318 |
# File 'lib/prima_aws_client.rb', line 315 def get_spotfleet_capacity(fleet_arn) resp = ec2_client.describe_spot_fleet_requests(spot_fleet_request_ids: [fleet_arn]) resp.spot_fleet_request_configs[0].spot_fleet_request_config.target_capacity end |
#get_stack_outputs(name) ⇒ Object
231 232 233 234 235 236 237 238 239 240 241 |
# File 'lib/prima_aws_client.rb', line 231 def get_stack_outputs(name) begin resp = cf_client.describe_stacks(stack_name: name) rescue Aws::CloudFormation::Errors::Throttling => e output 'Throttling, retrying in 15 seconds'.red sleep 15 get_stack_outputs(name) else resp.stacks[0].outputs end end |
#get_stack_parameters(name) ⇒ Object
219 220 221 222 223 224 225 226 227 228 229 |
# File 'lib/prima_aws_client.rb', line 219 def get_stack_parameters(name) begin resp = cf_client.describe_stacks(stack_name: name) rescue Aws::CloudFormation::Errors::Throttling => e output 'Throttling, retrying in 15 seconds'.red sleep 15 get_stack_parameters(name) else resp.stacks[0].parameters end end |
#get_stack_tags(name) ⇒ Object
207 208 209 210 211 212 213 214 215 216 217 |
# File 'lib/prima_aws_client.rb', line 207 def (name) begin resp = cf_client.describe_stacks(stack_name: name) rescue Aws::CloudFormation::Errors::Throttling => e output 'Throttling, retrying in 15 seconds'.red sleep 15 (name) else resp.stacks[0]. end end |
#get_stack_template(name) ⇒ Object
243 244 245 246 247 248 249 250 251 252 253 254 |
# File 'lib/prima_aws_client.rb', line 243 def get_stack_template(name) begin resp = cf_client.get_template(stack_name: name) rescue Aws::CloudFormation::Errors::Throttling => e output 'Throttling, retrying in 15 seconds'.red sleep 15 get_stack_template(name) else resp.template_body end end |
#hashes_to_tags(hashes) ⇒ Object
320 321 322 323 324 325 326 327 |
# File 'lib/prima_aws_client.rb', line 320 def (hashes) = [] hkeys = hashes.keys hkeys.each do |hkey| .insert(0, { key: hkey, value: hashes[hkey].to_s }) end end |
#list_exports ⇒ Object
65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 |
# File 'lib/prima_aws_client.rb', line 65 def list_exports exports = [] next_token = '' loop do print '.'.yellow; STDOUT.flush = next_token != '' ? { next_token: next_token } : {} begin resp = cf_client.list_exports() rescue Aws::CloudFormation::Errors::Throttling => e output 'Throttling, retrying in 15 seconds'.red sleep 15 resp = cf_client.list_exports() end exports += resp.exports break unless resp.next_token next_token = resp.next_token end puts '.'.yellow; STDOUT.flush exports end |
#list_import_stacks(export_name) ⇒ Object
290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 |
# File 'lib/prima_aws_client.rb', line 290 def list_import_stacks(export_name) stacks = [] next_token = '' loop do print '.'.yellow; STDOUT.flush = next_token != '' ? { export_name: export_name, next_token: next_token } : {export_name: export_name} begin resp = cf_client.list_imports() rescue Aws::CloudFormation::Errors::Throttling => e output 'Throttling, retrying in 15 seconds'.red sleep 15 resp = cf_client.list_imports() end stacks += resp.imports break unless resp.next_token next_token = resp.next_token end stacks end |
#s3_client ⇒ Object
5 6 7 |
# File 'lib/prima_aws_client.rb', line 5 def s3_client @s3 ||= Aws::S3::Client.new end |
#stack_exists?(stack_name) ⇒ Boolean
163 164 165 166 167 168 169 170 171 172 173 174 175 176 |
# File 'lib/prima_aws_client.rb', line 163 def stack_exists?(stack_name) begin cf_client.describe_stacks(stack_name: stack_name) rescue Aws::CloudFormation::Errors::Throttling => e output 'Throttling, retrying in 15 seconds'.red sleep 15 stack_exists?(stack_name) rescue Aws::CloudFormation::Errors::ValidationError => e return false if e..include? 'does not exist' raise e else true end end |
#stack_list ⇒ Object
21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
# File 'lib/prima_aws_client.rb', line 21 def stack_list stacks = [] next_token = '' loop do print '.'.yellow; STDOUT.flush = next_token != '' ? { next_token: next_token } : {} begin resp = cf_client.describe_stacks() rescue Aws::CloudFormation::Errors::Throttling => e output 'Throttling, retrying in 15 seconds'.red sleep 15 resp = cf_client.describe_stacks() end stacks += resp.stacks break unless resp.next_token next_token = resp.next_token end puts '.'.yellow; STDOUT.flush stacks.keep_if { |stack| stack.stack_name.include? '-qa-' } stacks end |
#stack_ready?(stack_name, failed_statuses = ['CREATE_FAILED', 'ROLLBACK_IN_PROGRESS', 'ROLLBACK_FAILED', 'DELETE_IN_PROGRESS', 'DELETE_FAILED', 'DELETE_COMPLETE', 'UPDATE_ROLLBACK_FAILED', 'UPDATE_ROLLBACK_COMPLETE_CLEANUP_IN_PROGRESS']) ⇒ Boolean
256 257 258 259 260 261 262 263 264 265 266 267 268 |
# File 'lib/prima_aws_client.rb', line 256 def stack_ready?(stack_name, failed_statuses = ['CREATE_FAILED', 'ROLLBACK_IN_PROGRESS', 'ROLLBACK_FAILED', 'DELETE_IN_PROGRESS', 'DELETE_FAILED', 'DELETE_COMPLETE', 'UPDATE_ROLLBACK_FAILED', 'UPDATE_ROLLBACK_COMPLETE_CLEANUP_IN_PROGRESS']) begin resp = cf_client.describe_stacks( stack_name: stack_name ) stack_status = resp.stacks[0].stack_status rescue Aws::CloudFormation::Errors::Throttling => e print 'Throttling'.red; STDOUT.flush return false end raise "The stack #{stack_name} errored out" if failed_statuses.include? stack_status ['CREATE_COMPLETE', 'UPDATE_COMPLETE', 'UPDATE_ROLLBACK_COMPLETE', 'ROLLBACK_COMPLETE'].include? stack_status end |
#tags_to_hashes(tags) ⇒ Object
329 330 331 332 333 334 335 |
# File 'lib/prima_aws_client.rb', line 329 def () hash = Hash.new .each do || hash[.key] = .value end hash end |
#update_stack(stack_name, template_body, parameters = [], tags = [], role = nil) ⇒ Object
111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 |
# File 'lib/prima_aws_client.rb', line 111 def update_stack(stack_name, template_body, parameters = [], = [], role = nil) cf_args = { stack_name: stack_name, template_body: template_body, parameters: parameters, tags: , capabilities: ['CAPABILITY_IAM'] } if role != nil then cf_args.merge!(role_arn: role) end begin cf_client.update_stack(cf_args) rescue Aws::CloudFormation::Errors::Throttling => e output 'Throttling, retrying in 15 seconds'.red sleep 15 update_stack(stack_name, template_body, parameters = [], = []) rescue Aws::CloudFormation::Errors::ValidationError => e raise e else output "L'update dello stack #{stack_name} è stato avviato".green end end |
#update_stack_url(stack_name, template_url, parameters = [], tags = [], role = nil) ⇒ Object
137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 |
# File 'lib/prima_aws_client.rb', line 137 def update_stack_url(stack_name, template_url, parameters = [], = [], role = nil) cf_args = { stack_name: stack_name, template_url: template_url, parameters: parameters, tags: , capabilities: ['CAPABILITY_IAM'] } if role != nil then cf_args.merge!(role_arn: role) end begin cf_client.update_stack(cf_args) rescue Aws::CloudFormation::Errors::Throttling => e output 'Throttling, retrying in 15 seconds'.red sleep 15 update_stack_url(stack_name, template_url, parameters = [], = []) rescue Aws::CloudFormation::Errors::ValidationError => e raise e else output "L'update dello stack #{stack_name} è stato avviato".green end end |
#upload_artifact(source_path, destination_path, bucket_name_override = nil) ⇒ Object
279 280 281 282 283 284 285 286 287 288 |
# File 'lib/prima_aws_client.rb', line 279 def upload_artifact(source_path, destination_path, bucket_name_override=nil) output "Upload dell'artifact in corso (#{(File.size(source_path).to_f / 2**20).round(2)} MiB)\n".yellow s3 = Aws::S3::Resource.new s3_bucket = if !bucket_name_override.nil? then bucket_name_override else @s3_bucket end puts s3_bucket obj = s3.bucket(s3_bucket).object(destination_path) obj.upload_file(source_path) output "#{s3_bucket}/#{destination_path} uploadato con successo!\n".green end |
#wait_for_stack_ready(stack_name, failed_statuses = ['CREATE_FAILED', 'ROLLBACK_IN_PROGRESS', 'ROLLBACK_FAILED', 'DELETE_IN_PROGRESS', 'DELETE_FAILED', 'DELETE_COMPLETE', 'UPDATE_ROLLBACK_FAILED', 'UPDATE_ROLLBACK_COMPLETE_CLEANUP_IN_PROGRESS']) ⇒ Object
190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 |
# File 'lib/prima_aws_client.rb', line 190 def wait_for_stack_ready(stack_name, failed_statuses = ['CREATE_FAILED', 'ROLLBACK_IN_PROGRESS', 'ROLLBACK_FAILED', 'DELETE_IN_PROGRESS', 'DELETE_FAILED', 'DELETE_COMPLETE', 'UPDATE_ROLLBACK_FAILED', 'UPDATE_ROLLBACK_COMPLETE_CLEANUP_IN_PROGRESS']) ready = false sleep_seconds = 13 output "Attendo che lo stack #{stack_name} finisca di essere inizializzato...\n".yellow while !ready ready = true if stack_ready?(stack_name, failed_statuses) seconds_elapsed = 0 loop do break if seconds_elapsed >= sleep_seconds print '.'.yellow; STDOUT.flush sleep 1 seconds_elapsed += 1 end end output "\nStack #{stack_name} pronto!\n".green end |