ansible/library/cloud/rax_clb

304 lines
8.9 KiB
Text
Raw Normal View History

#!/usr/bin/python
# This file is part of Ansible
#
# Ansible is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Ansible is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
# This is a DOCUMENTATION stub specific to this module, it extends
# a documentation fragment located in ansible.utils.module_docs_fragments
DOCUMENTATION = '''
---
module: rax_clb
short_description: create / delete a load balancer in Rackspace Public Cloud
description:
- creates / deletes a Rackspace Public Cloud load balancer.
version_added: "1.4"
options:
2013-11-06 18:39:47 +01:00
algorithm:
description:
2013-11-06 18:39:47 +01:00
- algorithm for the balancer being created
choices:
- RANDOM
- LEAST_CONNECTIONS
- ROUND_ROBIN
- WEIGHTED_LEAST_CONNECTIONS
- WEIGHTED_ROUND_ROBIN
2013-11-06 18:39:47 +01:00
default: LEAST_CONNECTIONS
meta:
description:
2013-11-06 18:39:47 +01:00
- A hash of metadata to associate with the instance
default: null
name:
description:
2013-11-06 18:39:47 +01:00
- Name to give the load balancer
default: null
port:
description:
2013-11-06 18:39:47 +01:00
- Port for the balancer being created
default: 80
protocol:
description:
2013-11-06 18:39:47 +01:00
- Protocol for the balancer being created
choices:
- DNS_TCP
- DNS_UDP
- FTP
- HTTP
- HTTPS
- IMAPS
- IMAPv4
- LDAP
- LDAPS
- MYSQL
- POP3
- POP3S
- SMTP
- TCP
- TCP_CLIENT_FIRST
- UDP
- UDP_STREAM
- SFTP
default: HTTP
2013-11-06 18:39:47 +01:00
state:
description:
- Indicate desired state of the resource
choices:
- present
- absent
2013-11-06 18:39:47 +01:00
default: present
timeout:
description:
2013-11-06 18:39:47 +01:00
- timeout for communication between the balancer and the node
default: 30
type:
description:
2013-11-06 18:39:47 +01:00
- type of interface for the balancer being created
choices:
- PUBLIC
- SERVICENET
default: PUBLIC
vip_id:
description:
- Virtual IP ID to use when creating the load balancer for purposes of
sharing an IP with another load balancer of another protocol
2013-12-12 16:29:20 +01:00
version_added: 1.5
wait:
description:
2013-11-06 18:39:47 +01:00
- wait for the balancer to be in state 'running' before returning
default: "no"
choices:
- "yes"
- "no"
wait_timeout:
description:
2013-11-06 18:39:47 +01:00
- how long before wait gives up, in seconds
default: 300
2013-11-06 18:39:47 +01:00
author: Christopher H. Laco, Matt Martz
extends_documentation_fragment: rackspace
'''
EXAMPLES = '''
- name: Build a Load Balancer
gather_facts: False
2013-11-06 18:39:47 +01:00
hosts: local
connection: local
tasks:
2013-11-06 18:39:47 +01:00
- name: Load Balancer create request
local_action:
module: rax_clb
credentials: ~/.raxpub
name: my-lb
2013-11-06 18:39:47 +01:00
port: 8080
protocol: HTTP
type: SERVICENET
timeout: 30
region: DFW
wait: yes
state: present
2013-11-06 18:39:47 +01:00
meta:
app: my-cool-app
register: my_lb
'''
try:
import pyrax
HAS_PYRAX = True
except ImportError:
HAS_PYRAX = False
2013-11-06 18:39:47 +01:00
def cloud_load_balancer(module, state, name, meta, algorithm, port, protocol,
vip_type, timeout, wait, wait_timeout, vip_id):
2013-11-06 18:39:47 +01:00
for arg in (state, name, port, protocol, vip_type):
if not arg:
2013-11-06 18:39:47 +01:00
module.fail_json(msg='%s is required for rax_clb' % arg)
if int(timeout) < 30:
module.fail_json(msg='"timeout" must be greater than or equal to 30')
changed = False
balancers = []
2013-11-06 18:39:47 +01:00
clb = pyrax.cloud_loadbalancers
if not clb:
module.fail_json(msg='Failed to instantiate client. This '
'typically indicates an invalid region or an '
'incorrectly capitalized region name.')
2013-11-06 18:39:47 +01:00
for balancer in clb.list():
if name != balancer.name and name != balancer.id:
continue
balancers.append(balancer)
2013-11-06 18:39:47 +01:00
if len(balancers) > 1:
module.fail_json(msg='Multiple Load Balancers were matched by name, '
'try using the Load Balancer ID instead')
if state == 'present':
2013-11-06 18:39:47 +01:00
if isinstance(meta, dict):
metadata = [dict(key=k, value=v) for k, v in meta.items()]
if not balancers:
try:
virtual_ips = [clb.VirtualIP(type=vip_type, id=vip_id)]
2013-11-06 18:39:47 +01:00
balancer = clb.create(name, metadata=metadata, port=port,
algorithm=algorithm, protocol=protocol,
timeout=timeout, virtual_ips=virtual_ips)
changed = True
except Exception, e:
module.fail_json(msg='%s' % e.message)
2013-11-06 18:39:47 +01:00
else:
balancer = balancers[0]
setattr(balancer, 'metadata',
[dict(key=k, value=v) for k, v in
balancer.get_metadata().items()])
atts = {
'name': name,
'algorithm': algorithm,
'port': port,
'protocol': protocol,
'timeout': timeout
}
for att, value in atts.iteritems():
current = getattr(balancer, att)
if current != value:
changed = True
2013-11-06 18:39:47 +01:00
if changed:
balancer.update(**atts)
if balancer.metadata != metadata:
balancer.set_metadata(meta)
changed = True
virtual_ips = [clb.VirtualIP(type=vip_type)]
current_vip_types = set([v.type for v in balancer.virtual_ips])
vip_types = set([v.type for v in virtual_ips])
if current_vip_types != vip_types:
module.fail_json(msg='Load balancer Virtual IP type cannot '
'be changed')
if wait:
attempts = wait_timeout / 5
pyrax.utils.wait_for_build(balancer, interval=5, attempts=attempts)
balancer.get()
instance = rax_to_dict(balancer, 'clb')
2013-11-06 18:39:47 +01:00
result = dict(changed=changed, balancer=instance)
if balancer.status == 'ERROR':
result['msg'] = '%s failed to build' % balancer.id
elif wait and balancer.status not in ('ACTIVE', 'ERROR'):
result['msg'] = 'Timeout waiting on %s' % balancer.id
if 'msg' in result:
module.fail_json(**result)
else:
module.exit_json(**result)
elif state == 'absent':
2013-11-06 18:39:47 +01:00
if balancers:
balancer = balancers[0]
try:
balancer.delete()
changed = True
except Exception, e:
module.fail_json(msg='%s' % e.message)
instance = rax_to_dict(balancer, 'clb')
2013-11-06 18:39:47 +01:00
if wait:
attempts = wait_timeout / 5
pyrax.utils.wait_until(balancer, 'status', ('DELETED'),
interval=5, attempts=attempts)
else:
instance = {}
2013-11-06 18:39:47 +01:00
module.exit_json(changed=changed, balancer=instance)
def main():
2013-11-19 19:51:04 +01:00
argument_spec = rax_argument_spec()
argument_spec.update(
dict(
algorithm=dict(choices=CLB_ALGORITHMS,
default='LEAST_CONNECTIONS'),
meta=dict(type='dict', default={}),
2013-11-06 18:39:47 +01:00
name=dict(),
port=dict(type='int', default=80),
protocol=dict(choices=CLB_PROTOCOLS, default='HTTP'),
2013-11-06 18:39:47 +01:00
state=dict(default='present', choices=['present', 'absent']),
timeout=dict(type='int', default=30),
2013-11-06 18:39:47 +01:00
type=dict(choices=['PUBLIC', 'SERVICENET'], default='PUBLIC'),
vip_id=dict(),
wait=dict(type='bool'),
2013-11-06 18:39:47 +01:00
wait_timeout=dict(default=300),
2013-11-19 19:51:04 +01:00
)
)
module = AnsibleModule(
argument_spec=argument_spec,
2013-12-05 02:26:34 +01:00
required_together=rax_required_together(),
)
if not HAS_PYRAX:
module.fail_json(msg='pyrax is required for this module')
2013-11-06 18:39:47 +01:00
algorithm = module.params.get('algorithm')
meta = module.params.get('meta')
2013-11-06 18:39:47 +01:00
name = module.params.get('name')
port = module.params.get('port')
protocol = module.params.get('protocol')
2013-11-06 18:39:47 +01:00
state = module.params.get('state')
timeout = int(module.params.get('timeout'))
vip_id = module.params.get('vip_id')
2013-11-06 18:39:47 +01:00
vip_type = module.params.get('type')
wait = module.params.get('wait')
wait_timeout = int(module.params.get('wait_timeout'))
2013-11-19 19:51:04 +01:00
setup_rax_module(module, pyrax)
2013-11-06 18:39:47 +01:00
cloud_load_balancer(module, state, name, meta, algorithm, port, protocol,
vip_type, timeout, wait, wait_timeout, vip_id)
2013-11-19 19:51:04 +01:00
# import module snippets
2013-11-06 18:39:47 +01:00
from ansible.module_utils.basic import *
2013-11-19 19:51:04 +01:00
from ansible.module_utils.rax import *
2013-11-19 19:51:04 +01:00
### invoke the module
main()