Merge pull request #4835 from sivel/rax-clb
Add module to create Rackspace Cloud Loadblancers. Fixes #4794
This commit is contained in:
commit
ca63f59ce2
2 changed files with 594 additions and 259 deletions
503
cloud/rax_clb
503
cloud/rax_clb
|
@ -1,4 +1,4 @@
|
||||||
#!/usr/bin/python
|
#!/usr/bin/python -tt
|
||||||
# This file is part of Ansible
|
# This file is part of Ansible
|
||||||
#
|
#
|
||||||
# Ansible is free software: you can redistribute it and/or modify
|
# Ansible is free software: you can redistribute it and/or modify
|
||||||
|
@ -17,334 +17,319 @@
|
||||||
DOCUMENTATION = '''
|
DOCUMENTATION = '''
|
||||||
---
|
---
|
||||||
module: rax_clb
|
module: rax_clb
|
||||||
short_description: add, modify and remove nodes from a Rackspace Cloud Load Balancer
|
short_description: create / delete a load balancer in Rackspace Public Cloud
|
||||||
description:
|
description:
|
||||||
- Adds, modifies and removes nodes from a Rackspace Cloud Load Balancer
|
- creates / deletes a Rackspace Public Cloud load balancer.
|
||||||
version_added: "1.4"
|
version_added: "1.4"
|
||||||
options:
|
options:
|
||||||
address:
|
algorithm:
|
||||||
required: false
|
|
||||||
description:
|
description:
|
||||||
- IP address or domain name of the node
|
- algorithm for the balancer being created
|
||||||
|
choices: ['RANDOM', 'LEAST_CONNECTIONS', 'ROUND_ROBIN', 'WEIGHTED_LEAST_CONNECTIONS', 'WEIGHTED_ROUND_ROBIN']
|
||||||
|
default: LEAST_CONNECTIONS
|
||||||
api_key:
|
api_key:
|
||||||
required: false
|
|
||||||
description:
|
description:
|
||||||
- Rackspace API key (overrides C(credentials))
|
- Rackspace API key (overrides C(credentials))
|
||||||
condition:
|
|
||||||
required: false
|
|
||||||
choices: [ "enabled", "disabled", "draining" ]
|
|
||||||
description:
|
|
||||||
- Condition for the node, which determines its role within the load
|
|
||||||
balancer
|
|
||||||
credentials:
|
credentials:
|
||||||
required: false
|
|
||||||
description:
|
description:
|
||||||
- File to find the Rackspace credentials in (ignored if C(api_key) and
|
- File to find the Rackspace credentials in (ignored if C(api_key) and
|
||||||
C(username) are provided)
|
C(username) are provided)
|
||||||
load_balancer_id:
|
default: null
|
||||||
required: true
|
aliases: ['creds_file']
|
||||||
type: integer
|
meta:
|
||||||
description:
|
description:
|
||||||
- Load balancer id
|
- A hash of metadata to associate with the instance
|
||||||
node_id:
|
default: null
|
||||||
required: false
|
name:
|
||||||
type: integer
|
|
||||||
description:
|
description:
|
||||||
- Node id
|
- Name to give the load balancer
|
||||||
|
default: null
|
||||||
port:
|
port:
|
||||||
required: false
|
|
||||||
type: integer
|
|
||||||
description:
|
description:
|
||||||
- Port number of the load balanced service on the node
|
- Port for the balancer being created
|
||||||
|
default: 80
|
||||||
|
protocol:
|
||||||
|
description:
|
||||||
|
- 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
|
||||||
region:
|
region:
|
||||||
required: false
|
|
||||||
description:
|
description:
|
||||||
- Region to authenticate in
|
- Region to create the load balancer in
|
||||||
|
default: DFW
|
||||||
state:
|
state:
|
||||||
required: false
|
|
||||||
default: "present"
|
|
||||||
choices: [ "present", "absent" ]
|
|
||||||
description:
|
description:
|
||||||
- Indicate desired state of the node
|
- Indicate desired state of the resource
|
||||||
|
choices: ['present', 'absent']
|
||||||
|
default: present
|
||||||
|
timeout:
|
||||||
|
description:
|
||||||
|
- timeout for communication between the balancer and the node
|
||||||
|
default: 30
|
||||||
type:
|
type:
|
||||||
required: false
|
|
||||||
choices: [ "primary", "secondary" ]
|
|
||||||
description:
|
description:
|
||||||
- Type of node
|
- type of interface for the balancer being created
|
||||||
|
choices: ['PUBLIC', 'SERVICENET']
|
||||||
|
default: PUBLIC
|
||||||
username:
|
username:
|
||||||
required: false
|
|
||||||
description:
|
description:
|
||||||
- Rackspace username (overrides C(credentials))
|
- Rackspace username (overrides C(credentials))
|
||||||
virtualenv:
|
|
||||||
required: false
|
|
||||||
description:
|
|
||||||
- Path to a virtualenv that should be activated before doing anything.
|
|
||||||
The virtualenv has to already exist. Useful if installing pyrax
|
|
||||||
globally is not an option.
|
|
||||||
wait:
|
wait:
|
||||||
required: false
|
description:
|
||||||
|
- wait for the balancer to be in state 'running' before returning
|
||||||
default: "no"
|
default: "no"
|
||||||
choices: [ "yes", "no" ]
|
choices: [ "yes", "no" ]
|
||||||
description:
|
|
||||||
- Wait for the load balancer to become active before returning
|
|
||||||
wait_timeout:
|
wait_timeout:
|
||||||
required: false
|
|
||||||
type: integer
|
|
||||||
default: 30
|
|
||||||
description:
|
description:
|
||||||
- How long to wait before giving up and returning an error
|
- how long before wait gives up, in seconds
|
||||||
weight:
|
default: 300
|
||||||
required: false
|
|
||||||
description:
|
|
||||||
- Weight of node
|
|
||||||
requirements: [ "pyrax" ]
|
requirements: [ "pyrax" ]
|
||||||
author: Lukasz Kawczynski
|
author: Christopher H. Laco, Matt Martz
|
||||||
notes:
|
notes:
|
||||||
- "The following environment variables can be used: C(RAX_USERNAME),
|
- The following environment variables can be used, C(RAX_USERNAME),
|
||||||
C(RAX_API_KEY), C(RAX_CREDENTIALS) and C(RAX_REGION)."
|
C(RAX_API_KEY), C(RAX_CREDS_FILE), C(RAX_CREDENTIALS), C(RAX_REGION).
|
||||||
|
- C(RAX_CREDENTIALS) and C(RAX_CREDS_FILE) points to a credentials file
|
||||||
|
appropriate for pyrax. See U(https://github.com/rackspace/pyrax/blob/master/docs/getting_started.md#authenticating)
|
||||||
|
- C(RAX_USERNAME) and C(RAX_API_KEY) obviate the use of a credentials file
|
||||||
|
- C(RAX_REGION) defines a Rackspace Public Cloud region (DFW, ORD, LON, ...)
|
||||||
'''
|
'''
|
||||||
|
|
||||||
EXAMPLES = '''
|
EXAMPLES = '''
|
||||||
# Add a new node to the load balancer
|
- name: Build a Load Balancer
|
||||||
- local_action:
|
gather_facts: False
|
||||||
module: rax_clb
|
hosts: local
|
||||||
load_balancer_id: 71
|
connection: local
|
||||||
address: 10.2.2.3
|
tasks:
|
||||||
port: 80
|
- name: Load Balancer create request
|
||||||
condition: enabled
|
local_action:
|
||||||
type: primary
|
module: rax_clb
|
||||||
wait: yes
|
credentials: ~/.raxpub
|
||||||
credentials: /path/to/credentials
|
name: my-lb
|
||||||
|
port: 8080
|
||||||
# Drain connections from a node
|
protocol: HTTP
|
||||||
- local_action:
|
type: SERVICENET
|
||||||
module: rax_clb
|
timeout: 30
|
||||||
load_balancer_id: 71
|
region: DFW
|
||||||
node_id: 410
|
wait: yes
|
||||||
condition: draining
|
state: present
|
||||||
wait: yes
|
meta:
|
||||||
credentials: /path/to/credentials
|
app: my-cool-app
|
||||||
|
register: my_lb
|
||||||
# Remove a node from the load balancer
|
|
||||||
- local_action:
|
|
||||||
module: rax_clb
|
|
||||||
load_balancer_id: 71
|
|
||||||
node_id: 410
|
|
||||||
state: absent
|
|
||||||
wait: yes
|
|
||||||
credentials: /path/to/credentials
|
|
||||||
'''
|
'''
|
||||||
|
|
||||||
|
import sys
|
||||||
import os
|
import os
|
||||||
|
|
||||||
|
from types import NoneType
|
||||||
|
|
||||||
def _activate_virtualenv(path):
|
try:
|
||||||
path = os.path.expanduser(path)
|
import pyrax
|
||||||
activate_this = os.path.join(path, 'bin', 'activate_this.py')
|
except ImportError:
|
||||||
execfile(activate_this, dict(__file__=activate_this))
|
print("failed=True msg='pyrax required for this module'")
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
NON_CALLABLES = (basestring, bool, dict, int, list, NoneType)
|
||||||
|
ALGORITHMS = ['RANDOM', 'LEAST_CONNECTIONS', 'ROUND_ROBIN',
|
||||||
|
'WEIGHTED_LEAST_CONNECTIONS', 'WEIGHTED_ROUND_ROBIN']
|
||||||
|
PROTOCOLS = ['DNS_TCP', 'DNS_UDP', 'FTP', 'HTTP', 'HTTPS', 'IMAPS', 'IMAPv4',
|
||||||
|
'LDAP', 'LDAPS', 'MYSQL', 'POP3', 'POP3S', 'SMTP', 'TCP',
|
||||||
|
'TCP_CLIENT_FIRST', 'UDP', 'UDP_STREAM', 'SFTP']
|
||||||
|
|
||||||
|
|
||||||
def _get_node(lb, node_id):
|
def cloud_load_balancer(module, state, name, meta, algorithm, port, protocol,
|
||||||
"""Return a node with the given `node_id`"""
|
vip_type, timeout, wait, wait_timeout):
|
||||||
for node in lb.nodes:
|
for arg in (state, name, port, protocol, vip_type):
|
||||||
if node.id == node_id:
|
if not arg:
|
||||||
return node
|
module.fail_json(msg='%s is required for rax_clb' % arg)
|
||||||
return None
|
|
||||||
|
|
||||||
|
if int(timeout) < 30:
|
||||||
|
module.fail_json(msg='"timeout" must be greater than or equal to 30')
|
||||||
|
|
||||||
def _is_primary(node):
|
changed = False
|
||||||
"""Return True if node is primary and enabled"""
|
balancers = []
|
||||||
return (node.type.lower() == 'primary' and
|
|
||||||
node.condition.lower() == 'enabled')
|
|
||||||
|
|
||||||
|
clb = pyrax.cloud_loadbalancers
|
||||||
|
|
||||||
def _get_primary_nodes(lb):
|
for balancer in clb.list():
|
||||||
"""Return a list of primary and enabled nodes"""
|
if name != balancer.name and name != balancer.id:
|
||||||
nodes = []
|
continue
|
||||||
for node in lb.nodes:
|
|
||||||
if _is_primary(node):
|
|
||||||
nodes.append(node)
|
|
||||||
return nodes
|
|
||||||
|
|
||||||
|
balancers.append(balancer)
|
||||||
|
|
||||||
def _node_to_dict(node):
|
if len(balancers) > 1:
|
||||||
"""Return a dictionary containing node details"""
|
module.fail_json(msg='Multiple Load Balancers were matched by name, '
|
||||||
if not node:
|
'try using the Load Balancer ID instead')
|
||||||
return {}
|
|
||||||
return {
|
if state == 'present':
|
||||||
'address': node.address,
|
if isinstance(meta, dict):
|
||||||
'condition': node.condition,
|
metadata = [dict(key=k, value=v) for k, v in meta.items()]
|
||||||
'id': node.id,
|
|
||||||
'port': node.port,
|
if not balancers:
|
||||||
'type': node.type,
|
try:
|
||||||
'weight': node.weight,
|
virtual_ips = [clb.VirtualIP(type=vip_type)]
|
||||||
}
|
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)
|
||||||
|
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
|
||||||
|
|
||||||
|
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
|
||||||
|
print attempts
|
||||||
|
pyrax.utils.wait_for_build(balancer, interval=5, attempts=attempts)
|
||||||
|
|
||||||
|
balancer.get()
|
||||||
|
instance = {}
|
||||||
|
for key, value in vars(balancer).iteritems():
|
||||||
|
if key == 'virtual_ips':
|
||||||
|
virtual_ips = []
|
||||||
|
instance[key] = []
|
||||||
|
for vip in value:
|
||||||
|
vip_dict = {}
|
||||||
|
for vip_key, vip_value in vars(vip).iteritems():
|
||||||
|
if isinstance(vip_value, NON_CALLABLES):
|
||||||
|
vip_dict[vip_key] = vip_value
|
||||||
|
instance[key].append(vip_dict)
|
||||||
|
elif (isinstance(value, NON_CALLABLES) and
|
||||||
|
not key.startswith('_')):
|
||||||
|
instance[key] = value
|
||||||
|
|
||||||
|
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':
|
||||||
|
if balancers:
|
||||||
|
balancer = balancers[0]
|
||||||
|
try:
|
||||||
|
balancer.delete()
|
||||||
|
changed = True
|
||||||
|
except Exception, e:
|
||||||
|
module.fail_json(msg='%s' % e.message)
|
||||||
|
|
||||||
|
instance = {}
|
||||||
|
for key, value in vars(balancer).iteritems():
|
||||||
|
if key == 'virtual_ips':
|
||||||
|
virtual_ips = []
|
||||||
|
instance[key] = []
|
||||||
|
for vip in value:
|
||||||
|
vip_dict = {}
|
||||||
|
for vip_key, vip_value in vars(vip).iteritems():
|
||||||
|
if isinstance(vip_value, NON_CALLABLES):
|
||||||
|
vip_dict[vip_key] = vip_value
|
||||||
|
instance[key].append(vip_dict)
|
||||||
|
elif (isinstance(value, NON_CALLABLES) and
|
||||||
|
not key.startswith('_')):
|
||||||
|
instance[key] = value
|
||||||
|
|
||||||
|
if wait:
|
||||||
|
attempts = wait_timeout / 5
|
||||||
|
pyrax.utils.wait_until(balancer, 'status', ('DELETED'),
|
||||||
|
interval=5, attempts=attempts)
|
||||||
|
else:
|
||||||
|
instance = {}
|
||||||
|
|
||||||
|
module.exit_json(changed=changed, balancer=instance)
|
||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
module = AnsibleModule(
|
module = AnsibleModule(
|
||||||
argument_spec=dict(
|
argument_spec=dict(
|
||||||
address=dict(),
|
algorithm=dict(choices=ALGORITHMS, default='LEAST_CONNECTIONS'),
|
||||||
api_key=dict(),
|
api_key=dict(),
|
||||||
condition=dict(choices=['enabled', 'disabled', 'draining']),
|
credentials=dict(aliases=['creds_file']),
|
||||||
credentials=dict(),
|
meta=dict(type='dict', default={}),
|
||||||
load_balancer_id=dict(required=True, type='int'),
|
name=dict(),
|
||||||
node_id=dict(type='int'),
|
port=dict(type='int', default=80),
|
||||||
port=dict(type='int'),
|
protocol=dict(choices=PROTOCOLS, default='HTTP'),
|
||||||
region=dict(),
|
region=dict(),
|
||||||
state=dict(default='present', choices=['present', 'absent']),
|
state=dict(default='present', choices=['present', 'absent']),
|
||||||
type=dict(choices=['primary', 'secondary']),
|
timeout=dict(type='int', default=30),
|
||||||
|
type=dict(choices=['PUBLIC', 'SERVICENET'], default='PUBLIC'),
|
||||||
username=dict(),
|
username=dict(),
|
||||||
virtualenv=dict(),
|
wait=dict(type='bool'),
|
||||||
wait=dict(default=False, type='bool'),
|
wait_timeout=dict(default=300),
|
||||||
wait_timeout=dict(default=30, type='int'),
|
|
||||||
weight=dict(type='int'),
|
|
||||||
),
|
),
|
||||||
required_together=[
|
|
||||||
['api_key', 'username']
|
|
||||||
],
|
|
||||||
)
|
)
|
||||||
|
|
||||||
address = module.params['address']
|
algorithm = module.params.get('algorithm')
|
||||||
api_key = module.params['api_key']
|
api_key = module.params.get('api_key')
|
||||||
condition = (module.params['condition'] and
|
credentials = module.params.get('credentials')
|
||||||
module.params['condition'].upper())
|
meta = module.params.get('meta')
|
||||||
credentials = module.params['credentials']
|
name = module.params.get('name')
|
||||||
load_balancer_id = module.params['load_balancer_id']
|
port = module.params.get('port')
|
||||||
node_id = module.params['node_id']
|
protocol = module.params.get('protocol')
|
||||||
port = module.params['port']
|
region = module.params.get('region')
|
||||||
region = module.params['region']
|
state = module.params.get('state')
|
||||||
state = module.params['state']
|
timeout = int(module.params.get('timeout'))
|
||||||
typ = module.params['type'] and module.params['type'].upper()
|
username = module.params.get('username')
|
||||||
username = module.params['username']
|
vip_type = module.params.get('type')
|
||||||
virtualenv = module.params['virtualenv']
|
wait = module.params.get('wait')
|
||||||
wait = module.params['wait']
|
wait_timeout = int(module.params.get('wait_timeout'))
|
||||||
wait_timeout = module.params['wait_timeout'] or 1
|
|
||||||
weight = module.params['weight']
|
|
||||||
|
|
||||||
if virtualenv:
|
|
||||||
try:
|
|
||||||
_activate_virtualenv(virtualenv)
|
|
||||||
except IOError, e:
|
|
||||||
module.fail_json(msg='Failed to activate virtualenv %s (%s)' % (
|
|
||||||
virtualenv, e))
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
import pyrax
|
username = username or os.environ.get('RAX_USERNAME')
|
||||||
except ImportError:
|
api_key = api_key or os.environ.get('RAX_API_KEY')
|
||||||
module.fail_json(msg='pyrax is not installed')
|
credentials = (credentials or os.environ.get('RAX_CREDENTIALS') or
|
||||||
|
os.environ.get('RAX_CREDS_FILE'))
|
||||||
username = username or os.environ.get('RAX_USERNAME')
|
region = region or os.environ.get('RAX_REGION')
|
||||||
api_key = api_key or os.environ.get('RAX_API_KEY')
|
except KeyError, e:
|
||||||
credentials = credentials or os.environ.get('RAX_CREDENTIALS')
|
module.fail_json(msg='Unable to load %s' % e.message)
|
||||||
region = region or os.environ.get('RAX_REGION')
|
|
||||||
|
|
||||||
pyrax.set_setting("identity_type", "rackspace")
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
pyrax.set_setting('identity_type', 'rackspace')
|
||||||
if api_key and username:
|
if api_key and username:
|
||||||
pyrax.set_credentials(username, api_key=api_key, region=region)
|
pyrax.set_credentials(username, api_key=api_key, region=region)
|
||||||
elif credentials:
|
elif credentials:
|
||||||
credentials = os.path.expanduser(credentials)
|
credentials = os.path.expanduser(credentials)
|
||||||
pyrax.set_credential_file(credentials, region=region)
|
pyrax.set_credential_file(credentials, region=region)
|
||||||
else:
|
else:
|
||||||
module.fail_json(msg='Credentials not set')
|
raise Exception('No credentials supplied!')
|
||||||
except pyrax.exc.PyraxException, e:
|
except Exception, e:
|
||||||
module.fail_json(msg='%s' % e.message)
|
module.fail_json(msg='%s' % e.message)
|
||||||
|
|
||||||
if not pyrax.cloud_loadbalancers:
|
cloud_load_balancer(module, state, name, meta, algorithm, port, protocol,
|
||||||
module.fail_json(msg='Failed to instantiate load balancer client '
|
vip_type, timeout, wait, wait_timeout)
|
||||||
'(possibly incorrect region)')
|
|
||||||
|
|
||||||
try:
|
from ansible.module_utils.basic import *
|
||||||
lb = pyrax.cloud_loadbalancers.get(load_balancer_id)
|
|
||||||
except pyrax.exc.PyraxException, e:
|
|
||||||
module.fail_json(msg='%s' % e.message)
|
|
||||||
|
|
||||||
if node_id:
|
|
||||||
node = _get_node(lb, node_id)
|
|
||||||
else:
|
|
||||||
node = None
|
|
||||||
|
|
||||||
result = _node_to_dict(node)
|
|
||||||
|
|
||||||
if state == 'absent':
|
|
||||||
if not node: # Removing a non-existent node
|
|
||||||
module.exit_json(changed=False, state=state)
|
|
||||||
|
|
||||||
# The API detects this as well but currently pyrax does not return a
|
|
||||||
# meaningful error message
|
|
||||||
if _is_primary(node) and len(_get_primary_nodes(lb)) == 1:
|
|
||||||
module.fail_json(
|
|
||||||
msg='At least one primary node has to be enabled')
|
|
||||||
|
|
||||||
try:
|
|
||||||
lb.delete_node(node)
|
|
||||||
result = {}
|
|
||||||
except pyrax.exc.NotFound:
|
|
||||||
module.exit_json(changed=False, state=state)
|
|
||||||
except pyrax.exc.PyraxException, e:
|
|
||||||
module.fail_json(msg='%s' % e.message)
|
|
||||||
else: # present
|
|
||||||
if not node:
|
|
||||||
if node_id: # Updating a non-existent node
|
|
||||||
msg = 'Node %d not found' % node_id
|
|
||||||
if lb.nodes:
|
|
||||||
msg += (' (available nodes: %s)' %
|
|
||||||
', '.join([str(x.id) for x in lb.nodes]))
|
|
||||||
module.fail_json(msg=msg)
|
|
||||||
else: # Creating a new node
|
|
||||||
try:
|
|
||||||
node = pyrax.cloudloadbalancers.Node(
|
|
||||||
address=address, port=port, condition=condition,
|
|
||||||
weight=weight, type=typ)
|
|
||||||
resp, body = lb.add_nodes([node])
|
|
||||||
result.update(body['nodes'][0])
|
|
||||||
except pyrax.exc.PyraxException, e:
|
|
||||||
module.fail_json(msg='%s' % e.message)
|
|
||||||
else: # Updating an existing node
|
|
||||||
immutable = {
|
|
||||||
'address': address,
|
|
||||||
'port': port,
|
|
||||||
}
|
|
||||||
|
|
||||||
mutable = {
|
|
||||||
'condition': condition,
|
|
||||||
'type': typ,
|
|
||||||
'weight': weight,
|
|
||||||
}
|
|
||||||
|
|
||||||
for name, value in immutable.items():
|
|
||||||
if value:
|
|
||||||
module.fail_json(
|
|
||||||
msg='Attribute %s cannot be modified' % name)
|
|
||||||
|
|
||||||
for name, value in mutable.items():
|
|
||||||
if value is None or value == getattr(node, name):
|
|
||||||
mutable.pop(name)
|
|
||||||
|
|
||||||
if not mutable:
|
|
||||||
module.exit_json(changed=False, state=state, node=result)
|
|
||||||
|
|
||||||
try:
|
|
||||||
# The diff has to be set explicitly to update node's weight and
|
|
||||||
# type; this should probably be fixed in pyrax
|
|
||||||
lb.update_node(node, diff=mutable)
|
|
||||||
result.update(mutable)
|
|
||||||
except pyrax.exc.PyraxException, e:
|
|
||||||
module.fail_json(msg='%s' % e.message)
|
|
||||||
|
|
||||||
if wait:
|
|
||||||
pyrax.utils.wait_until(lb, "status", "ACTIVE", interval=1,
|
|
||||||
attempts=wait_timeout)
|
|
||||||
if lb.status != 'ACTIVE':
|
|
||||||
module.fail_json(
|
|
||||||
msg='Load balancer not active after %ds (current status: %s)' %
|
|
||||||
(wait_timeout, lb.status.lower()))
|
|
||||||
|
|
||||||
kwargs = {'node': result} if result else {}
|
|
||||||
module.exit_json(changed=True, state=state, **kwargs)
|
|
||||||
|
|
||||||
# this is magic, see lib/ansible/module_common.py
|
|
||||||
#<<INCLUDE_ANSIBLE_MODULE_COMMON>>
|
|
||||||
main()
|
main()
|
||||||
|
|
350
cloud/rax_clb_nodes
Normal file
350
cloud/rax_clb_nodes
Normal file
|
@ -0,0 +1,350 @@
|
||||||
|
#!/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/>.
|
||||||
|
|
||||||
|
DOCUMENTATION = '''
|
||||||
|
---
|
||||||
|
module: rax_clb_nodes
|
||||||
|
short_description: add, modify and remove nodes from a Rackspace Cloud Load Balancer
|
||||||
|
description:
|
||||||
|
- Adds, modifies and removes nodes from a Rackspace Cloud Load Balancer
|
||||||
|
version_added: "1.4"
|
||||||
|
options:
|
||||||
|
address:
|
||||||
|
required: false
|
||||||
|
description:
|
||||||
|
- IP address or domain name of the node
|
||||||
|
api_key:
|
||||||
|
required: false
|
||||||
|
description:
|
||||||
|
- Rackspace API key (overrides C(credentials))
|
||||||
|
condition:
|
||||||
|
required: false
|
||||||
|
choices: [ "enabled", "disabled", "draining" ]
|
||||||
|
description:
|
||||||
|
- Condition for the node, which determines its role within the load
|
||||||
|
balancer
|
||||||
|
credentials:
|
||||||
|
required: false
|
||||||
|
description:
|
||||||
|
- File to find the Rackspace credentials in (ignored if C(api_key) and
|
||||||
|
C(username) are provided)
|
||||||
|
load_balancer_id:
|
||||||
|
required: true
|
||||||
|
type: integer
|
||||||
|
description:
|
||||||
|
- Load balancer id
|
||||||
|
node_id:
|
||||||
|
required: false
|
||||||
|
type: integer
|
||||||
|
description:
|
||||||
|
- Node id
|
||||||
|
port:
|
||||||
|
required: false
|
||||||
|
type: integer
|
||||||
|
description:
|
||||||
|
- Port number of the load balanced service on the node
|
||||||
|
region:
|
||||||
|
required: false
|
||||||
|
description:
|
||||||
|
- Region to authenticate in
|
||||||
|
state:
|
||||||
|
required: false
|
||||||
|
default: "present"
|
||||||
|
choices: [ "present", "absent" ]
|
||||||
|
description:
|
||||||
|
- Indicate desired state of the node
|
||||||
|
type:
|
||||||
|
required: false
|
||||||
|
choices: [ "primary", "secondary" ]
|
||||||
|
description:
|
||||||
|
- Type of node
|
||||||
|
username:
|
||||||
|
required: false
|
||||||
|
description:
|
||||||
|
- Rackspace username (overrides C(credentials))
|
||||||
|
virtualenv:
|
||||||
|
required: false
|
||||||
|
description:
|
||||||
|
- Path to a virtualenv that should be activated before doing anything.
|
||||||
|
The virtualenv has to already exist. Useful if installing pyrax
|
||||||
|
globally is not an option.
|
||||||
|
wait:
|
||||||
|
required: false
|
||||||
|
default: "no"
|
||||||
|
choices: [ "yes", "no" ]
|
||||||
|
description:
|
||||||
|
- Wait for the load balancer to become active before returning
|
||||||
|
wait_timeout:
|
||||||
|
required: false
|
||||||
|
type: integer
|
||||||
|
default: 30
|
||||||
|
description:
|
||||||
|
- How long to wait before giving up and returning an error
|
||||||
|
weight:
|
||||||
|
required: false
|
||||||
|
description:
|
||||||
|
- Weight of node
|
||||||
|
requirements: [ "pyrax" ]
|
||||||
|
author: Lukasz Kawczynski
|
||||||
|
notes:
|
||||||
|
- "The following environment variables can be used: C(RAX_USERNAME),
|
||||||
|
C(RAX_API_KEY), C(RAX_CREDENTIALS) and C(RAX_REGION)."
|
||||||
|
'''
|
||||||
|
|
||||||
|
EXAMPLES = '''
|
||||||
|
# Add a new node to the load balancer
|
||||||
|
- local_action:
|
||||||
|
module: rax_clb_nodes
|
||||||
|
load_balancer_id: 71
|
||||||
|
address: 10.2.2.3
|
||||||
|
port: 80
|
||||||
|
condition: enabled
|
||||||
|
type: primary
|
||||||
|
wait: yes
|
||||||
|
credentials: /path/to/credentials
|
||||||
|
|
||||||
|
# Drain connections from a node
|
||||||
|
- local_action:
|
||||||
|
module: rax_clb_nodes
|
||||||
|
load_balancer_id: 71
|
||||||
|
node_id: 410
|
||||||
|
condition: draining
|
||||||
|
wait: yes
|
||||||
|
credentials: /path/to/credentials
|
||||||
|
|
||||||
|
# Remove a node from the load balancer
|
||||||
|
- local_action:
|
||||||
|
module: rax_clb_nodes
|
||||||
|
load_balancer_id: 71
|
||||||
|
node_id: 410
|
||||||
|
state: absent
|
||||||
|
wait: yes
|
||||||
|
credentials: /path/to/credentials
|
||||||
|
'''
|
||||||
|
|
||||||
|
import os
|
||||||
|
|
||||||
|
|
||||||
|
def _activate_virtualenv(path):
|
||||||
|
path = os.path.expanduser(path)
|
||||||
|
activate_this = os.path.join(path, 'bin', 'activate_this.py')
|
||||||
|
execfile(activate_this, dict(__file__=activate_this))
|
||||||
|
|
||||||
|
|
||||||
|
def _get_node(lb, node_id):
|
||||||
|
"""Return a node with the given `node_id`"""
|
||||||
|
for node in lb.nodes:
|
||||||
|
if node.id == node_id:
|
||||||
|
return node
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
def _is_primary(node):
|
||||||
|
"""Return True if node is primary and enabled"""
|
||||||
|
return (node.type.lower() == 'primary' and
|
||||||
|
node.condition.lower() == 'enabled')
|
||||||
|
|
||||||
|
|
||||||
|
def _get_primary_nodes(lb):
|
||||||
|
"""Return a list of primary and enabled nodes"""
|
||||||
|
nodes = []
|
||||||
|
for node in lb.nodes:
|
||||||
|
if _is_primary(node):
|
||||||
|
nodes.append(node)
|
||||||
|
return nodes
|
||||||
|
|
||||||
|
|
||||||
|
def _node_to_dict(node):
|
||||||
|
"""Return a dictionary containing node details"""
|
||||||
|
if not node:
|
||||||
|
return {}
|
||||||
|
return {
|
||||||
|
'address': node.address,
|
||||||
|
'condition': node.condition,
|
||||||
|
'id': node.id,
|
||||||
|
'port': node.port,
|
||||||
|
'type': node.type,
|
||||||
|
'weight': node.weight,
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
module = AnsibleModule(
|
||||||
|
argument_spec=dict(
|
||||||
|
address=dict(),
|
||||||
|
api_key=dict(),
|
||||||
|
condition=dict(choices=['enabled', 'disabled', 'draining']),
|
||||||
|
credentials=dict(),
|
||||||
|
load_balancer_id=dict(required=True, type='int'),
|
||||||
|
node_id=dict(type='int'),
|
||||||
|
port=dict(type='int'),
|
||||||
|
region=dict(),
|
||||||
|
state=dict(default='present', choices=['present', 'absent']),
|
||||||
|
type=dict(choices=['primary', 'secondary']),
|
||||||
|
username=dict(),
|
||||||
|
virtualenv=dict(),
|
||||||
|
wait=dict(default=False, type='bool'),
|
||||||
|
wait_timeout=dict(default=30, type='int'),
|
||||||
|
weight=dict(type='int'),
|
||||||
|
),
|
||||||
|
required_together=[
|
||||||
|
['api_key', 'username']
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
|
address = module.params['address']
|
||||||
|
api_key = module.params['api_key']
|
||||||
|
condition = (module.params['condition'] and
|
||||||
|
module.params['condition'].upper())
|
||||||
|
credentials = module.params['credentials']
|
||||||
|
load_balancer_id = module.params['load_balancer_id']
|
||||||
|
node_id = module.params['node_id']
|
||||||
|
port = module.params['port']
|
||||||
|
region = module.params['region']
|
||||||
|
state = module.params['state']
|
||||||
|
typ = module.params['type'] and module.params['type'].upper()
|
||||||
|
username = module.params['username']
|
||||||
|
virtualenv = module.params['virtualenv']
|
||||||
|
wait = module.params['wait']
|
||||||
|
wait_timeout = module.params['wait_timeout'] or 1
|
||||||
|
weight = module.params['weight']
|
||||||
|
|
||||||
|
if virtualenv:
|
||||||
|
try:
|
||||||
|
_activate_virtualenv(virtualenv)
|
||||||
|
except IOError, e:
|
||||||
|
module.fail_json(msg='Failed to activate virtualenv %s (%s)' % (
|
||||||
|
virtualenv, e))
|
||||||
|
|
||||||
|
try:
|
||||||
|
import pyrax
|
||||||
|
except ImportError:
|
||||||
|
module.fail_json(msg='pyrax is not installed')
|
||||||
|
|
||||||
|
username = username or os.environ.get('RAX_USERNAME')
|
||||||
|
api_key = api_key or os.environ.get('RAX_API_KEY')
|
||||||
|
credentials = credentials or os.environ.get('RAX_CREDENTIALS')
|
||||||
|
region = region or os.environ.get('RAX_REGION')
|
||||||
|
|
||||||
|
pyrax.set_setting("identity_type", "rackspace")
|
||||||
|
|
||||||
|
try:
|
||||||
|
if api_key and username:
|
||||||
|
pyrax.set_credentials(username, api_key=api_key, region=region)
|
||||||
|
elif credentials:
|
||||||
|
credentials = os.path.expanduser(credentials)
|
||||||
|
pyrax.set_credential_file(credentials, region=region)
|
||||||
|
else:
|
||||||
|
module.fail_json(msg='Credentials not set')
|
||||||
|
except pyrax.exc.PyraxException, e:
|
||||||
|
module.fail_json(msg='%s' % e.message)
|
||||||
|
|
||||||
|
if not pyrax.cloud_loadbalancers:
|
||||||
|
module.fail_json(msg='Failed to instantiate load balancer client '
|
||||||
|
'(possibly incorrect region)')
|
||||||
|
|
||||||
|
try:
|
||||||
|
lb = pyrax.cloud_loadbalancers.get(load_balancer_id)
|
||||||
|
except pyrax.exc.PyraxException, e:
|
||||||
|
module.fail_json(msg='%s' % e.message)
|
||||||
|
|
||||||
|
if node_id:
|
||||||
|
node = _get_node(lb, node_id)
|
||||||
|
else:
|
||||||
|
node = None
|
||||||
|
|
||||||
|
result = _node_to_dict(node)
|
||||||
|
|
||||||
|
if state == 'absent':
|
||||||
|
if not node: # Removing a non-existent node
|
||||||
|
module.exit_json(changed=False, state=state)
|
||||||
|
|
||||||
|
# The API detects this as well but currently pyrax does not return a
|
||||||
|
# meaningful error message
|
||||||
|
if _is_primary(node) and len(_get_primary_nodes(lb)) == 1:
|
||||||
|
module.fail_json(
|
||||||
|
msg='At least one primary node has to be enabled')
|
||||||
|
|
||||||
|
try:
|
||||||
|
lb.delete_node(node)
|
||||||
|
result = {}
|
||||||
|
except pyrax.exc.NotFound:
|
||||||
|
module.exit_json(changed=False, state=state)
|
||||||
|
except pyrax.exc.PyraxException, e:
|
||||||
|
module.fail_json(msg='%s' % e.message)
|
||||||
|
else: # present
|
||||||
|
if not node:
|
||||||
|
if node_id: # Updating a non-existent node
|
||||||
|
msg = 'Node %d not found' % node_id
|
||||||
|
if lb.nodes:
|
||||||
|
msg += (' (available nodes: %s)' %
|
||||||
|
', '.join([str(x.id) for x in lb.nodes]))
|
||||||
|
module.fail_json(msg=msg)
|
||||||
|
else: # Creating a new node
|
||||||
|
try:
|
||||||
|
node = pyrax.cloudloadbalancers.Node(
|
||||||
|
address=address, port=port, condition=condition,
|
||||||
|
weight=weight, type=typ)
|
||||||
|
resp, body = lb.add_nodes([node])
|
||||||
|
result.update(body['nodes'][0])
|
||||||
|
except pyrax.exc.PyraxException, e:
|
||||||
|
module.fail_json(msg='%s' % e.message)
|
||||||
|
else: # Updating an existing node
|
||||||
|
immutable = {
|
||||||
|
'address': address,
|
||||||
|
'port': port,
|
||||||
|
}
|
||||||
|
|
||||||
|
mutable = {
|
||||||
|
'condition': condition,
|
||||||
|
'type': typ,
|
||||||
|
'weight': weight,
|
||||||
|
}
|
||||||
|
|
||||||
|
for name, value in immutable.items():
|
||||||
|
if value:
|
||||||
|
module.fail_json(
|
||||||
|
msg='Attribute %s cannot be modified' % name)
|
||||||
|
|
||||||
|
for name, value in mutable.items():
|
||||||
|
if value is None or value == getattr(node, name):
|
||||||
|
mutable.pop(name)
|
||||||
|
|
||||||
|
if not mutable:
|
||||||
|
module.exit_json(changed=False, state=state, node=result)
|
||||||
|
|
||||||
|
try:
|
||||||
|
# The diff has to be set explicitly to update node's weight and
|
||||||
|
# type; this should probably be fixed in pyrax
|
||||||
|
lb.update_node(node, diff=mutable)
|
||||||
|
result.update(mutable)
|
||||||
|
except pyrax.exc.PyraxException, e:
|
||||||
|
module.fail_json(msg='%s' % e.message)
|
||||||
|
|
||||||
|
if wait:
|
||||||
|
pyrax.utils.wait_until(lb, "status", "ACTIVE", interval=1,
|
||||||
|
attempts=wait_timeout)
|
||||||
|
if lb.status != 'ACTIVE':
|
||||||
|
module.fail_json(
|
||||||
|
msg='Load balancer not active after %ds (current status: %s)' %
|
||||||
|
(wait_timeout, lb.status.lower()))
|
||||||
|
|
||||||
|
kwargs = {'node': result} if result else {}
|
||||||
|
module.exit_json(changed=True, state=state, **kwargs)
|
||||||
|
|
||||||
|
# this is magic, see lib/ansible/module_common.py
|
||||||
|
#<<INCLUDE_ANSIBLE_MODULE_COMMON>>
|
||||||
|
main()
|
Loading…
Reference in a new issue