Add a new unique_name param to the digital_ocean module for idempotence.

As it stands now, it is difficult to write idempotent tasks for digital
ocean droplets.  Digital ocean assigns new nodes a random id when they
are provisioned and that id is the only key that can be used to identify
it in subsequent runs of that play.

The workflow previously involved manual intervention:

- write a play defining a new node with no specified id
- run it, collect the randomly assigned id by hand
- modify the play to add the id by hand so future runs don't create
  duplicate nodes
- perform future re-runs that check if the node exists (by its id)
  - if it does exist then do nothing.
  - if it does not exist, then create it and return a *new random id*
  - collect the new random id by hand, modify the playbook file, and
    start all over.

Its a huge pain.

The modifications in this commit allow you to use the 'hostname' as a
primary key for idempotence with digital ocean.  By default, digital
ocean will let you create as many hosts with the same hostname as you
like.  Here, we provide an option to constrain the user to using only
unique hostnames.

The workflow will now look like:

- write a play defining a new node with a specified hostname and
  "unique_name: true""
- run it, create the new node and move on.
- re-run it, notice that a node with that hostname is already created
  and move on.
This commit is contained in:
Ralph Bean 2013-09-18 16:07:22 -04:00
parent 6ecf41530b
commit a9a2a96218

View file

@ -45,6 +45,11 @@ options:
name: name:
description: description:
- String, this is the name of the droplet - must be formatted by hostname rules, or the name of a SSH key. - String, this is the name of the droplet - must be formatted by hostname rules, or the name of a SSH key.
unique_name:
description:
- Bool, require unique hostnames. By default, digital ocean allows multiple hosts with the same name. Setting this to "yes" allows only one host per name. Useful for idempotence.
default: "no"
choices: [ "yes", "no" ]
size_id: size_id:
description: description:
- Numeric, this is the id of the size you would like the droplet created at. - Numeric, this is the id of the size you would like the droplet created at.
@ -211,13 +216,22 @@ class Droplet(JsonfyMixIn):
return droplet return droplet
@classmethod @classmethod
def find(cls, id): def find(cls, id=None, name=None):
if not id: if not id and not name:
return False return False
droplets = cls.list_all() droplets = cls.list_all()
# Check first by id. digital ocean requires that it be unique
for droplet in droplets: for droplet in droplets:
if droplet.id == id: if droplet.id == id:
return droplet return droplet
# Failing that, check by hostname.
for droplet in droplets:
if droplet.name == name:
return droplet
return False return False
@classmethod @classmethod
@ -281,7 +295,17 @@ def core(module):
if command == 'droplet': if command == 'droplet':
Droplet.setup(client_id, api_key) Droplet.setup(client_id, api_key)
if state in ('active', 'present'): if state in ('active', 'present'):
droplet = Droplet.find(module.params['id'])
# First, try to find a droplet by id.
droplet = Droplet.find(id=module.params['id'])
# If we couldn't find the droplet and the user is allowing unique
# hostnames, then check to see if a droplet with the specified
# hostname already exists.
if not droplet and module.params['unique_name']:
droplet = Droplet.find(name=getkeyordie('name'))
# If both of those attempts failed, then create a new droplet.
if not droplet: if not droplet:
droplet = Droplet.add( droplet = Droplet.add(
name=getkeyordie('name'), name=getkeyordie('name'),
@ -290,18 +314,30 @@ def core(module):
region_id=getkeyordie('region_id'), region_id=getkeyordie('region_id'),
ssh_key_ids=module.params['ssh_key_ids'] ssh_key_ids=module.params['ssh_key_ids']
) )
if droplet.is_powered_on(): if droplet.is_powered_on():
changed = False changed = False
droplet.ensure_powered_on( droplet.ensure_powered_on(
wait=getkeyordie('wait'), wait=getkeyordie('wait'),
wait_timeout=getkeyordie('wait_timeout') wait_timeout=getkeyordie('wait_timeout')
) )
module.exit_json(changed=changed, droplet=droplet.to_json()) module.exit_json(changed=changed, droplet=droplet.to_json())
elif state in ('absent', 'deleted'): elif state in ('absent', 'deleted'):
droplet = Droplet.find(getkeyordie('id')) # First, try to find a droplet by id.
droplet = Droplet.find(id=getkeyordie('id'))
# If we couldn't find the droplet and the user is allowing unique
# hostnames, then check to see if a droplet with the specified
# hostname already exists.
if not droplet and module.params['unique_name']:
droplet = Droplet.find(name=getkeyordie('name'))
if not droplet: if not droplet:
module.exit_json(changed=False, msg='The droplet is not found.') module.exit_json(changed=False, msg='The droplet is not found.')
event_json = droplet.destroy() event_json = droplet.destroy()
module.exit_json(changed=True, event_id=event_json['event_id']) module.exit_json(changed=True, event_id=event_json['event_id'])
@ -336,6 +372,7 @@ def main():
region_id = dict(type='int'), region_id = dict(type='int'),
ssh_key_ids = dict(default=''), ssh_key_ids = dict(default=''),
id = dict(aliases=['droplet_id'], type='int'), id = dict(aliases=['droplet_id'], type='int'),
unique_name = dict(type='bool', choices=BOOLEANS, default='no'),
wait = dict(type='bool', choices=BOOLEANS, default='yes'), wait = dict(type='bool', choices=BOOLEANS, default='yes'),
wait_timeout = dict(default=300, type='int'), wait_timeout = dict(default=300, type='int'),
ssh_pub_key = dict(type='str'), ssh_pub_key = dict(type='str'),