Module: PrimaAwsClient
- Defined in:
- lib/prima_aws_client.rb
Instance Method Summary collapse
- #artifact_exists?(bucket, path) ⇒ Boolean
- #asg_client ⇒ Object
- #cf_client ⇒ 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
248 249 250 251 252 253 254 255 |
# File 'lib/prima_aws_client.rb', line 248 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 |
#create_stack(stack_name, stack_body, parameters = [], tags = [], role = nil) ⇒ Object
64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 |
# File 'lib/prima_aws_client.rb', line 64 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
156 157 158 159 160 161 162 163 164 165 166 |
# File 'lib/prima_aws_client.rb', line 156 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
288 289 290 291 |
# File 'lib/prima_aws_client.rb', line 288 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
293 294 295 296 |
# File 'lib/prima_aws_client.rb', line 293 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
209 210 211 212 213 214 215 216 217 218 219 |
# File 'lib/prima_aws_client.rb', line 209 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
197 198 199 200 201 202 203 204 205 206 207 |
# File 'lib/prima_aws_client.rb', line 197 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
185 186 187 188 189 190 191 192 193 194 195 |
# File 'lib/prima_aws_client.rb', line 185 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
221 222 223 224 225 226 227 228 229 230 231 232 |
# File 'lib/prima_aws_client.rb', line 221 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
298 299 300 301 302 303 304 305 |
# File 'lib/prima_aws_client.rb', line 298 def (hashes) = [] hkeys = hashes.keys hkeys.each do |hkey| .insert(0, { key: hkey, value: hashes[hkey].to_s }) end end |
#list_exports ⇒ Object
43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
# File 'lib/prima_aws_client.rb', line 43 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
268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 |
# File 'lib/prima_aws_client.rb', line 268 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
141 142 143 144 145 146 147 148 149 150 151 152 153 154 |
# File 'lib/prima_aws_client.rb', line 141 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
234 235 236 237 238 239 240 241 242 243 244 245 246 |
# File 'lib/prima_aws_client.rb', line 234 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
307 308 309 310 311 312 313 |
# File 'lib/prima_aws_client.rb', line 307 def () hash = Hash.new .each do || hash[.key] = .value end hash end |
#update_stack(stack_name, template_body, parameters = [], tags = [], role = nil) ⇒ Object
89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 |
# File 'lib/prima_aws_client.rb', line 89 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
115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 |
# File 'lib/prima_aws_client.rb', line 115 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
257 258 259 260 261 262 263 264 265 266 |
# File 'lib/prima_aws_client.rb', line 257 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
168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 |
# File 'lib/prima_aws_client.rb', line 168 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 |