update of ec2-cost-calculate.rb to include ability to summarize by ASG

This commit is contained in:
Colin Johnson 2012-07-11 20:39:05 +00:00
parent 4f06b0af68
commit 0c22a72f56

View File

@ -2,7 +2,7 @@
access_key = ""
secret_key = ""
#default options are given below - user input can override any one of these
options = {:fileoutputlocation => "~/ec-cost-calculate-result.txt", :output => "screen",:seperator => ",",:region => "all",:period => "hour",:multiplier => 1,:status => :running, :awscredentialsfile => "", :awscredentialssource => "env", :user_selected_region => "all" }
options = {:fileoutputlocation => "~/ec-cost-calculate-result.txt", :output => "screen",:seperator => ",",:region => "all",:period => "hour",:multiplier => 1,:status => :running, :awscredentialsfile => "", :awscredentialssource => "env", :user_selected_region => "all", :mode => "byinstance" }
#ec2cc_resources holds resources needed by ec2_cost_calculate, such as the output file handle
ec2cc_resources = {}
#list of valid statuses for an instance - will be used to validate user input
@ -58,6 +58,30 @@ class Region_Resource
end
end
#CostASG - will contain one cost object per Auto Scaling Group
class CostASG
attr_accessor :name, :region, :instance_type,:total_instance_count, :total_cost
def initialize(name,region,instance_type,total_instance_count,total_cost)
@name = name
@region = region
@instance_type = instance_type
@total_instance_count = total_instance_count
@total_cost = total_cost
end
def output (options,asg_cost_object,ec2cc_resources)
outputstring = "#{asg_cost_object.name}","#{options[:seperator]}","#{asg_cost_object.region}","#{options[:seperator]}","#{asg_cost_object.instance_type}","#{options[:seperator]}","#{asg_cost_object.total_instance_count}","#{options[:seperator]}","#{asg_cost_object.total_cost}","\n"
case options[:output]
when "file"
ec2cc_resources[:ec2_output_file_handle].print outputstring
when "screen"
print outputstring
else
$stderr.print "error with output.\n"
exit 1
end
end
end
## ec2-cost-calculate ##
## Initialization of ec2-cost-calculate
#sets program name
@ -140,6 +164,17 @@ optparse = OptionParser.new do |opts|
exit 64
end
end
#mode - allowing either "byinstance" - in which the cost of each instance is listed or "byASG" in which the cost of each ASG is listed
opts.on("-m","--mode MODE","mode in which ", program_name, " runs. Accepts values \"byinstance\" or \"byASG.\" Default value is \"byinstance\".") do |mode|
#forces option to lowercase - easier to evaluate variables when always lowercase
mode.downcase!
if (mode == "byinstance" || mode == "byasg")
options[:mode] = mode
else
$stderr.print "You must specifiy a mode such as \"byinstance\" or \"byasg\". You specified \"", mode, ".\"\n"
exit 64
end
end
#options processing for aws credential file input
opts.on("--awscredentialfile CREDENTIAILFILE","path to AWS credential file. This is required if the path to an AWS credential file is not provided by an environment variable.") do |awscredentialfile|
options[:awscredentialfile] = awscredentialfile
@ -186,6 +221,8 @@ aws_interface = AWS::EC2.new( :access_key_id => access_key, :secret_access_key =
instance_collection = {};
#creates a collection (currently, an array) of all regions resources
region_collection = {};
#creates a collection (currently, an array) of all asg costs objects
asg_cost_collection = {};
#regions_aws is a list of all current Amazon regions
regions_array = aws_interface.regions.map
#file expansion and validation done outside of optparse
@ -253,6 +290,16 @@ AWS.memoize do
ec2_object.asg = instance.tags["aws:autoscaling:groupName"]
#gets price using Instance.price method
ec2_object.price = ec2_object.get_price(ec2_object.instance_type,ec2_object.region,ec2_object.platform,price_table,options[:multiplier])
#if the currently evaluated instance has an asg that is already in the asg_cost_collection, add the price of the current object to the asg_cost_collection.total_cost attribute, else create a new asg_cost_object and add to the asg_cost_collection
if asg_cost_collection.include?(ec2_object.asg)
asg_cost_collection[ec2_object.asg].total_instance_count += 1
asg_cost_collection[ec2_object.asg].total_cost += ec2_object.price
else
asg_cost_object = CostASG.new(ec2_object.asg,ec2_object.region,ec2_object.instance_type,1,ec2_object.price)
asg_cost_collection[ec2_object.asg] = asg_cost_object
end
#places each ec2_object into the instance_collection if the status of instance matches user requested status
if options[:status] == instance.status
instance_collection[instance.id] = ec2_object
@ -261,7 +308,13 @@ AWS.memoize do
end
end
#Prints Header
headerstring = "instanceid","#{options[:seperator]}","region","#{options[:seperator]}","platform","#{options[:seperator]}","status","#{options[:seperator]}","cost","#{options[:seperator]}","name","#{options[:seperator]}","autoscalinggroup","\n"
case options[:mode]
when "byinstance"
headerstring = "instanceid","#{options[:seperator]}","region","#{options[:seperator]}","platform","#{options[:seperator]}","status","#{options[:seperator]}","cost","#{options[:seperator]}","name","#{options[:seperator]}","autoscalinggroup","\n"
when "byasg"
headerstring = "asgname","#{options[:seperator]}","region","#{options[:seperator]}","instanceplatform","#{options[:seperator]}","instancecount","#{options[:seperator]}","asgcost","\n"
end
case options[:output]
when "screen"
@ -270,7 +323,13 @@ case options[:output]
ec2cc_resources[:ec2_output_file_handle].print headerstring
end
#Prints Output for each EC2 object
instance_collection.each do |ec2_instance_id,ec2_instance_object|
ec2_instance_object.output(options,ec2_instance_object,ec2cc_resources)
end
case options[:mode]
when "byinstance"
instance_collection.each do |ec2_instance_id,ec2_instance_object|
ec2_instance_object.output(options,ec2_instance_object,ec2cc_resources)
end
when "byasg"
asg_cost_collection.each do |asg_cost_object_asg,asg_cost_object|
asg_cost_object.output(options,asg_cost_object,ec2cc_resources)
end
end