Adds support for VPC subnets to ELB

pull/7051/head
Mike Buzzetti 11 years ago
parent 09e0d1c51f
commit 5dada9403e

@ -66,6 +66,22 @@ options:
- The AWS region to use. If not specified then the value of the EC2_REGION environment variable, if any, is used. - The AWS region to use. If not specified then the value of the EC2_REGION environment variable, if any, is used.
required: false required: false
aliases: ['aws_region', 'ec2_region'] aliases: ['aws_region', 'ec2_region']
subnets:
description:
- A list of VPC subnets to use when creating ELB. Zones should be empty if using this.
required: false
default: None
aliases: []
version_added: "1.6"
validate_certs:
description:
- When set to "no", SSL certificates will not be validated for boto versions >= 2.6.0.
required: false
default: "yes"
choices: ["yes", "no"]
aliases: []
version_added: "1.5"
extends_documentation_fragment: aws extends_documentation_fragment: aws
""" """
@ -92,6 +108,21 @@ EXAMPLES = """
# ssl certificate required for https or ssl # ssl certificate required for https or ssl
ssl_certificate_id: "arn:aws:iam::123456789012:server-certificate/company/servercerts/ProdServerCert" ssl_certificate_id: "arn:aws:iam::123456789012:server-certificate/company/servercerts/ProdServerCert"
# Basic VPC provisioning example
- local_action:
module: ec2_elb_lb
name: "test-vpc"
state: present
subnets:
zones:
- subnet-abcd1234
- subnet-1a2b3c4d
listeners:
- protocol: http # options are http, https, ssl, tcp
load_balancer_port: 80
instance_port: 80
# Configure a health check # Configure a health check
- local_action: - local_action:
module: ec2_elb_lb module: ec2_elb_lb
@ -149,6 +180,19 @@ EXAMPLES = """
load_balancer_port: 80 load_balancer_port: 80
instance_port: 80 instance_port: 80
purge_zones: yes purge_zones: yes
# Creates a ELB and assigns a list of subnets to it.
- local_action:
module: ec2_elb_lb
state: present
name: 'New ELB'
security_group_ids: 'sg-123456, sg-67890'
region: us-west-2
subnets: 'subnet-123456, subnet-67890'
listeners:
- protocol: http
load_balancer_port: 80
instance_port: 80
""" """
import sys import sys
@ -168,8 +212,10 @@ class ElbManager(object):
"""Handles ELB creation and destruction""" """Handles ELB creation and destruction"""
def __init__(self, module, name, listeners=None, purge_listeners=None, def __init__(self, module, name, listeners=None, purge_listeners=None,
zones=None, purge_zones=None, security_group_ids=None, health_check=None, zones=None, purge_zones=None, security_group_ids=None,
health_check=None, subnets=None,
region=None, **aws_connect_params): region=None, **aws_connect_params):
self.module = module self.module = module
self.name = name self.name = name
self.listeners = listeners self.listeners = listeners
@ -178,6 +224,7 @@ class ElbManager(object):
self.purge_zones = purge_zones self.purge_zones = purge_zones
self.security_group_ids = security_group_ids self.security_group_ids = security_group_ids
self.health_check = health_check self.health_check = health_check
self.subnets = subnets
self.aws_connect_params = aws_connect_params self.aws_connect_params = aws_connect_params
self.region = region self.region = region
@ -196,6 +243,7 @@ class ElbManager(object):
self._set_zones() self._set_zones()
self._set_security_groups() self._set_security_groups()
self._set_elb_listeners() self._set_elb_listeners()
self._set_subnets()
self._set_health_check() self._set_health_check()
def ensure_gone(self): def ensure_gone(self):
@ -215,7 +263,8 @@ class ElbManager(object):
'dns_name': self.elb.dns_name, 'dns_name': self.elb.dns_name,
'zones': self.elb.availability_zones, 'zones': self.elb.availability_zones,
'security_group_ids': self.elb.security_groups, 'security_group_ids': self.elb.security_groups,
'status': self.status 'status': self.status,
'subnets': self.subnets
} }
if self.elb.health_check: if self.elb.health_check:
@ -266,7 +315,8 @@ class ElbManager(object):
self.elb = self.elb_conn.create_load_balancer(name=self.name, self.elb = self.elb_conn.create_load_balancer(name=self.name,
zones=self.zones, zones=self.zones,
security_groups=self.security_group_ids, security_groups=self.security_group_ids,
complex_listeners=listeners) complex_listeners=listeners,
subnets=self.subnets)
if self.elb: if self.elb:
self.changed = True self.changed = True
self.status = 'created' self.status = 'created'
@ -373,8 +423,28 @@ class ElbManager(object):
self.elb_conn.disable_availability_zones(self.name, zones) self.elb_conn.disable_availability_zones(self.name, zones)
self.changed = True self.changed = True
def _attach_subnets(self, subnets):
self.elb_conn.attach_lb_to_subnets(self.name, subnets)
self.changed = True
def _detach_subnets(self, subnets):
self.elb_conn.detach_lb_from_subnets(self.name, subnets)
self.changed = True
def _set_subnets(self):
"""Determine which subnets need to be attached or detached on the ELB"""
if self.subnets:
subnets_to_detach = list(set(self.elb.subnets) - set(self.subnets))
subnets_to_attach = list(set(self.subnets) - set(self.elb.subnets))
if subnets_to_attach:
self._attach_subnets(subnets_to_attach)
if subnets_to_detach:
self._detach_subnets(subnets_to_detach)
def _set_zones(self): def _set_zones(self):
"""Determine which zones need to be enabled or disabled on the ELB""" """Determine which zones need to be enabled or disabled on the ELB"""
if self.zones:
if self.purge_zones: if self.purge_zones:
zones_to_disable = list(set(self.elb.availability_zones) - zones_to_disable = list(set(self.elb.availability_zones) -
set(self.zones)) set(self.zones))
@ -447,6 +517,7 @@ def main():
purge_zones={'default': False, 'required': False, 'type': 'bool'}, purge_zones={'default': False, 'required': False, 'type': 'bool'},
security_group_ids={'default': None, 'required': False, 'type': 'list'}, security_group_ids={'default': None, 'required': False, 'type': 'list'},
health_check={'default': None, 'required': False, 'type': 'dict'}, health_check={'default': None, 'required': False, 'type': 'dict'},
subnets={'default': None, 'required': False, 'type': 'list'}
) )
) )
@ -466,15 +537,16 @@ def main():
purge_zones = module.params['purge_zones'] purge_zones = module.params['purge_zones']
security_group_ids = module.params['security_group_ids'] security_group_ids = module.params['security_group_ids']
health_check = module.params['health_check'] health_check = module.params['health_check']
subnets = module.params['subnets']
if state == 'present' and not listeners: if state == 'present' and not listeners:
module.fail_json(msg="At least one port is required for ELB creation") module.fail_json(msg="At least one port is required for ELB creation")
if state == 'present' and not zones: if state == 'present' and not (zones or subnets):
module.fail_json(msg="At least one availability zone is required for ELB creation") module.fail_json(msg="At least one availability zone or subnet is required for ELB creation")
elb_man = ElbManager(module, name, listeners, purge_listeners, zones, elb_man = ElbManager(module, name, listeners, purge_listeners, zones,
purge_zones, security_group_ids, health_check, purge_zones, security_group_ids, health_check, subnets,
region=region, **aws_connect_params) region=region, **aws_connect_params)
if state == 'present': if state == 'present':

Loading…
Cancel
Save