aws-ruby-samples/create_auto_scaling_group.rb at master · d-smith/aws-ruby-samples · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
require File.expand_path(File.dirname(__FILE__) + '/config')

(vpc_id, lb_name) = ARGV
unless vpc_id && lb_name
  puts "Usage: create_auto_scaling_group <VPC_ID> <load balancer name>"
  exit 1
end

ec2Client = Aws::EC2::Client::new


# Grab the private subnets. Note this assumes we know the subnets we
# are looking for via knowledge of how the VPC was created.
subnet_infos = ec2Client.describe_subnets({
  :filters => [
    {
      :name => "vpc-id",
      :values => [vpc_id]
    },
    {
      :name => "cidr",
      :values => ["10.0.1.0/24", "10.0.3.0/24"]
    }
  ]
})[:subnets]

subnet_ids = subnet_infos.map do |subnet|
  subnet[:subnet_id]
end


azs = subnet_infos.map do |subnet|
  subnet[:availability_zone]
end


subnet_ids_str = subnet_ids.inject{|l,r| l + "," + r}

puts "subnet ids: '#{subnet_ids_str}'"


cw = Aws::CloudWatch::Client::new


asgClient = Aws::AutoScaling::Client::new

group_base = "b2bnext"
group_name = group_base + "-auto-scaling-group"

asgClient.create_auto_scaling_group({
    :auto_scaling_group_name => group_name,
    :launch_configuration_name => "b2bnext-launch-config",
    :min_size => 1,
    :max_size => 4,
    :default_cooldown => 300,
    :availability_zones => azs,
    :load_balancer_names => [lb_name],
    :health_check_type => "EC2",
    :health_check_grace_period => 300,
    :vpc_zone_identifier => subnet_ids_str
})


scale_up_policy_arn = asgClient.put_scaling_policy({
    :auto_scaling_group_name => group_name,
    :policy_name => "scale up policy",
    :scaling_adjustment => 1,
    :adjustment_type => "ChangeInCapacity",
    :cooldown => 600
})[:policy_arn]


scale_down_policy_arn = asgClient.put_scaling_policy({
    :auto_scaling_group_name => group_name,
    :policy_name => "scale down policy",
    :scaling_adjustment => -1,
    :adjustment_type => "ChangeInCapacity",
    :cooldown => 600
})[:policy_arn]

cw.put_metric_alarm({
    :alarm_name => group_base + " high cpu demo alarm",
    :actions_enabled => true,
    :alarm_actions => [scale_up_policy_arn],
    :metric_name => "CPUUtilization",
    :namespace => "AWS/EC2",
    :statistic => "Average",
    :dimensions => [{:name => "AutoScalingGroupName",:value => group_name}],
    :period=>300,
    :evaluation_periods=>1,
    :threshold=>50.0,
    :comparison_operator=>"GreaterThanOrEqualToThreshold"
})

cw.put_metric_alarm({
    :alarm_name => group_base + " low cpu demo alarm",
    :actions_enabled => true,
    :alarm_actions => [scale_down_policy_arn],
    :metric_name => "CPUUtilization",
    :namespace => "AWS/EC2",
    :statistic => "Average",
    :dimensions => [{:name => "AutoScalingGroupName",:value => group_name}],
    :period=>300,
    :evaluation_periods=>1,
    :threshold=>30.0,
    :comparison_operator=>"LessThanOrEqualToThreshold"
})