You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
290 lines
12 KiB
290 lines
12 KiB
#!/home/alpcentaur/ProjektA/PrototypeWebApp/venv/bin/python3.5
|
|
# Copyright (c) 2011 Joel Barciauskas http://joel.barciausk.as/
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining a
|
|
# copy of this software and associated documentation files (the
|
|
# "Software"), to deal in the Software without restriction, including
|
|
# without limitation the rights to use, copy, modify, merge, publish, dis-
|
|
# tribute, sublicense, and/or sell copies of the Software, and to permit
|
|
# persons to whom the Software is furnished to do so, subject to the fol-
|
|
# lowing conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be included
|
|
# in all copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL-
|
|
# ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
|
|
# SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
|
# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
|
|
|
|
#
|
|
# Auto Scaling Groups Tool
|
|
#
|
|
VERSION="0.1"
|
|
usage = """%prog [options] [command]
|
|
Commands:
|
|
list|ls List all Auto Scaling Groups
|
|
list-lc|ls-lc List all Launch Configurations
|
|
delete <name> Delete ASG <name>
|
|
delete-lc <name> Delete Launch Configuration <name>
|
|
get <name> Get details of ASG <name>
|
|
create <name> Create an ASG
|
|
create-lc <name> Create a Launch Configuration
|
|
update <name> <prop> <value> Update a property of an ASG
|
|
update-image <asg-name> <lc-name> Update image ID for ASG by creating a new LC
|
|
migrate-instances <name> Shut down current instances one by one and wait for ASG to start up a new instance with the current AMI (useful in conjunction with update-image)
|
|
|
|
Examples:
|
|
|
|
1) Create launch configuration
|
|
bin/asadmin create-lc my-lc-1 -i ami-1234abcd -t c1.xlarge -k my-key -s web-group -m
|
|
|
|
2) Create auto scaling group in us-east-1a and us-east-1c with a load balancer and min size of 2 and max size of 6
|
|
bin/asadmin create my-asg -z us-east-1a -z us-east-1c -l my-lc-1 -b my-lb -H ELB -p 180 -x 2 -X 6
|
|
"""
|
|
|
|
def get_group(autoscale, name):
|
|
g = autoscale.get_all_groups(names=[name])
|
|
if len(g) < 1:
|
|
print "No auto scaling groups by the name of %s found" % name
|
|
return sys.exit(1)
|
|
return g[0]
|
|
|
|
def get_lc(autoscale, name):
|
|
l = autoscale.get_all_launch_configurations(names=[name])
|
|
if len(l) < 1:
|
|
print "No launch configurations by the name of %s found" % name
|
|
sys.exit(1)
|
|
return l[0]
|
|
|
|
def list(autoscale):
|
|
"""List all ASGs"""
|
|
print "%-20s %s" % ("Name", "LC Name")
|
|
print "-"*80
|
|
groups = autoscale.get_all_groups()
|
|
for g in groups:
|
|
print "%-20s %s" % (g.name, g.launch_config_name)
|
|
|
|
def list_lc(autoscale):
|
|
"""List all LCs"""
|
|
print "%-30s %-20s %s" % ("Name", "Image ID", "Instance Type")
|
|
print "-"*80
|
|
for l in autoscale.get_all_launch_configurations():
|
|
print "%-30s %-20s %s" % (l.name, l.image_id, l.instance_type)
|
|
|
|
def get(autoscale, name):
|
|
"""Get details about ASG <name>"""
|
|
g = get_group(autoscale, name)
|
|
print "="*80
|
|
print "%-30s %s" % ('Name:', g.name)
|
|
print "%-30s %s" % ('Launch configuration:', g.launch_config_name)
|
|
print "%-30s %s" % ('Minimum size:', g.min_size)
|
|
print "%-30s %s" % ('Maximum size:', g.max_size)
|
|
print "%-30s %s" % ('Desired capacity:', g.desired_capacity)
|
|
print "%-30s %s" % ('Load balancers:', ','.join(g.load_balancers))
|
|
|
|
print
|
|
|
|
print "Instances"
|
|
print "---------"
|
|
print "%-20s %-20s %-20s %s" % ("ID", "Status", "Health", "AZ")
|
|
for i in g.instances:
|
|
print "%-20s %-20s %-20s %s" % \
|
|
(i.instance_id, i.lifecycle_state, i.health_status, i.availability_zone)
|
|
|
|
print
|
|
|
|
def create(autoscale, name, zones, lc_name, load_balancers, hc_type, hc_period,
|
|
min_size, max_size, cooldown, capacity):
|
|
"""Create an ASG named <name>"""
|
|
g = AutoScalingGroup(name=name, launch_config=lc_name,
|
|
availability_zones=zones, load_balancers=load_balancers,
|
|
default_cooldown=cooldown, health_check_type=hc_type,
|
|
health_check_period=hc_period, desired_capacity=capacity,
|
|
min_size=min_size, max_size=max_size)
|
|
g = autoscale.create_auto_scaling_group(g)
|
|
return list(autoscale)
|
|
|
|
def create_lc(autoscale, name, image_id, instance_type, key_name,
|
|
security_groups, instance_monitoring):
|
|
l = LaunchConfiguration(name=name, image_id=image_id,
|
|
instance_type=instance_type,key_name=key_name,
|
|
security_groups=security_groups,
|
|
instance_monitoring=instance_monitoring)
|
|
l = autoscale.create_launch_configuration(l)
|
|
return list_lc(autoscale)
|
|
|
|
def update(autoscale, name, prop, value):
|
|
g = get_group(autoscale, name)
|
|
setattr(g, prop, value)
|
|
g.update()
|
|
return get(autoscale, name)
|
|
|
|
def delete(autoscale, name, force_delete=False):
|
|
"""Delete this ASG"""
|
|
g = get_group(autoscale, name)
|
|
autoscale.delete_auto_scaling_group(g.name, force_delete)
|
|
print "Auto scaling group %s deleted" % name
|
|
return list(autoscale)
|
|
|
|
def delete_lc(autoscale, name):
|
|
"""Delete this LC"""
|
|
l = get_lc(autoscale, name)
|
|
autoscale.delete_launch_configuration(name)
|
|
print "Launch configuration %s deleted" % name
|
|
return list_lc(autoscale)
|
|
|
|
def update_image(autoscale, name, lc_name, image_id, is_migrate_instances=False):
|
|
""" Get the current launch config,
|
|
Update its name and image id
|
|
Re-create it as a new launch config
|
|
Update the ASG with the new LC
|
|
Delete the old LC """
|
|
|
|
g = get_group(autoscale, name)
|
|
l = get_lc(autoscale, g.launch_config_name)
|
|
|
|
old_lc_name = l.name
|
|
l.name = lc_name
|
|
l.image_id = image_id
|
|
autoscale.create_launch_configuration(l)
|
|
g.launch_config_name = l.name
|
|
g.update()
|
|
|
|
if(is_migrate_instances):
|
|
migrate_instances(autoscale, name)
|
|
else:
|
|
return get(autoscale, name)
|
|
|
|
def migrate_instances(autoscale, name):
|
|
""" Shut down instances of the old image type one by one
|
|
and let the ASG start up instances with the new image """
|
|
g = get_group(autoscale, name)
|
|
|
|
old_instances = g.instances
|
|
ec2 = boto.connect_ec2()
|
|
for old_instance in old_instances:
|
|
print "Terminating instance " + old_instance.instance_id
|
|
ec2.terminate_instances([old_instance.instance_id])
|
|
while True:
|
|
g = get_group(autoscale, name)
|
|
new_instances = g.instances
|
|
for new_instance in new_instances:
|
|
hasOldInstance = False
|
|
instancesReady = True
|
|
if(old_instance.instance_id == new_instance.instance_id):
|
|
hasOldInstance = True
|
|
print "Waiting for old instance to shut down..."
|
|
break
|
|
elif(new_instance.lifecycle_state != 'InService'):
|
|
instancesReady = False
|
|
print "Waiting for instances to be ready...."
|
|
break
|
|
if(not hasOldInstance and instancesReady):
|
|
break
|
|
else:
|
|
time.sleep(20)
|
|
return get(autoscale, name)
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
import readline
|
|
except ImportError:
|
|
pass
|
|
import boto
|
|
import sys
|
|
import time
|
|
from optparse import OptionParser
|
|
from boto.mashups.iobject import IObject
|
|
from boto.ec2.autoscale import AutoScalingGroup
|
|
from boto.ec2.autoscale import LaunchConfiguration
|
|
parser = OptionParser(version=VERSION, usage=usage)
|
|
""" Create launch config options """
|
|
parser.add_option("-i", "--image-id",
|
|
help="Image (AMI) ID", action="store",
|
|
type="string", default=None, dest="image_id")
|
|
parser.add_option("-t", "--instance-type",
|
|
help="EC2 Instance Type (e.g., m1.large, c1.xlarge), default is m1.large",
|
|
action="store", type="string", default="m1.large", dest="instance_type")
|
|
parser.add_option("-k", "--key-name",
|
|
help="EC2 Key Name",
|
|
action="store", type="string", dest="key_name")
|
|
parser.add_option("-s", "--security-group",
|
|
help="EC2 Security Group",
|
|
action="append", default=[], dest="security_groups")
|
|
parser.add_option("-m", "--monitoring",
|
|
help="Enable instance monitoring",
|
|
action="store_true", default=False, dest="instance_monitoring")
|
|
|
|
""" Create auto scaling group options """
|
|
parser.add_option("-z", "--zone", help="Add availability zone", action="append", default=[], dest="zones")
|
|
parser.add_option("-l", "--lc-name",
|
|
help="Launch configuration name",
|
|
action="store", default=None, type="string", dest="lc_name")
|
|
parser.add_option("-b", "--load-balancer",
|
|
help="Load balancer name",
|
|
action="append", default=[], dest="load_balancers")
|
|
parser.add_option("-H", "--health-check-type",
|
|
help="Health check type (EC2 or ELB)",
|
|
action="store", default="EC2", type="string", dest="hc_type")
|
|
parser.add_option("-p", "--health-check-period",
|
|
help="Health check period in seconds (default 300s)",
|
|
action="store", default=300, type="int", dest="hc_period")
|
|
parser.add_option("-X", "--max-size",
|
|
help="Max size of ASG (default 10)",
|
|
action="store", default=10, type="int", dest="max_size")
|
|
parser.add_option("-x", "--min-size",
|
|
help="Min size of ASG (default 2)",
|
|
action="store", default=2, type="int", dest="min_size")
|
|
parser.add_option("-c", "--cooldown",
|
|
help="Cooldown time after a scaling activity in seconds (default 300s)",
|
|
action="store", default=300, type="int", dest="cooldown")
|
|
parser.add_option("-C", "--desired-capacity",
|
|
help="Desired capacity of the ASG",
|
|
action="store", default=None, type="int", dest="capacity")
|
|
parser.add_option("-f", "--force",
|
|
help="Force delete ASG",
|
|
action="store_true", default=False, dest="force")
|
|
parser.add_option("-y", "--migrate-instances",
|
|
help="Automatically migrate instances to new image when running update-image",
|
|
action="store_true", default=False, dest="migrate_instances")
|
|
|
|
(options, args) = parser.parse_args()
|
|
|
|
if len(args) < 1:
|
|
parser.print_help()
|
|
sys.exit(1)
|
|
|
|
autoscale = boto.connect_autoscale()
|
|
|
|
print "%s" % (autoscale.region.endpoint)
|
|
|
|
command = args[0].lower()
|
|
if command in ("ls", "list"):
|
|
list(autoscale)
|
|
elif command in ("ls-lc", "list-lc"):
|
|
list_lc(autoscale)
|
|
elif command == "get":
|
|
get(autoscale, args[1])
|
|
elif command == "create":
|
|
create(autoscale, args[1], options.zones, options.lc_name,
|
|
options.load_balancers, options.hc_type,
|
|
options.hc_period, options.min_size, options.max_size,
|
|
options.cooldown, options.capacity)
|
|
elif command == "create-lc":
|
|
create_lc(autoscale, args[1], options.image_id, options.instance_type,
|
|
options.key_name, options.security_groups,
|
|
options.instance_monitoring)
|
|
elif command == "update":
|
|
update(autoscale, args[1], args[2], args[3])
|
|
elif command == "delete":
|
|
delete(autoscale, args[1], options.force)
|
|
elif command == "delete-lc":
|
|
delete_lc(autoscale, args[1])
|
|
elif command == "update-image":
|
|
update_image(autoscale, args[1], args[2],
|
|
options.image_id, options.migrate_instances)
|
|
elif command == "migrate-instances":
|
|
migrate_instances(autoscale, args[1])
|