Merge pull request #2043 from chrishoffman/rabbitmq

Rabbitmq Module Updates
reviewable/pr18780/r1
Michael DeHaan 12 years ago
commit df9aa8a02e

@ -54,22 +54,23 @@ examples:
class RabbitMqPlugins(object):
def __init__(self, module):
self.module = module
self._rabbitmq_plugins = module.get_bin_path('rabbitmq-plugins', True)
def _exec(self, args):
cmd = ["rabbitmq-plugins"]
def _exec(self, args, run_in_check_mode=False):
if not self.module.check_mode or (self.module.check_mode and run_in_check_mode):
cmd = [self._rabbitmq_plugins]
rc, out, err = self.module.run_command(cmd + args, check_rc=True)
return out.splitlines()
return list()
def get_all(self):
return self._exec(["list", "-E", "-m"])
return self._exec(['list', '-E', '-m'], True)
def enable(self, name):
if not self.module.check_mode:
self._exec(["enable", name])
self._exec(['enable', name])
def disable(self, name):
if not self.module.check_mode:
self._exec(["disable", name])
self._exec(['disable', name])
def main():
arg_spec = dict(

@ -82,7 +82,7 @@ options:
default: present
choices: [present, absent]
examples:
- code: rabbitmq_user user=joe password=changeme vhost="/" configure_priv=".*" read_priv=".*" write_priv=".*" state=present
- code: rabbitmq_user user=joe password=changeme vhost=/ configure_priv=.* read_priv=.* write_priv=.* state=present
description: Add user to server and assign full access control
'''
@ -92,7 +92,7 @@ class RabbitMqUser(object):
self.username = username
self.password = password
if tags is None:
self.tags = []
self.tags = list()
else:
self.tags = tags.split(',')
@ -106,14 +106,17 @@ class RabbitMqUser(object):
self._tags = None
self._permissions = None
self._rabbitmqctl = module.get_bin_path('rabbitmqctl', True)
def _exec(self, args):
cmd = ["rabbitmqctl", "-q"]
def _exec(self, args, run_in_check_mode=False):
if not self.module.check_mode or (self.module.check_mode and run_in_check_mode):
cmd = [self._rabbitmqctl, '-q']
rc, out, err = self.module.run_command(cmd + args, check_rc=True)
return out.splitlines()
return list()
def get(self):
users = self._exec(["list_users"])
users = self._exec(['list_users'], True)
for user_tag in users:
user, tags = user_tag.split('\t')
@ -125,16 +128,14 @@ class RabbitMqUser(object):
if tags != '':
self._tags = tags.split(',')
else:
self._tags = []
self._tags = list()
self._permissions = self._get_permissions()
return True
return False
def _get_permissions(self):
perms_out = self._exec(["list_user_permissions", self.username])
perms_out = self._exec(['list_user_permissions', self.username], True)
for perm in perms_out:
vhost, configure_priv, write_priv, read_priv = perm.split('\t')
@ -144,21 +145,16 @@ class RabbitMqUser(object):
return dict()
def add(self):
if not self.module.check_mode:
self._exec(["add_user", self.username, self.password])
self._exec(['add_user', self.username, self.password])
def delete(self):
if not self.module.check_mode:
self._exec(["delete_user", self.username])
self._exec(['delete_user', self.username])
def set_tags(self):
if not self.module.check_mode:
self._exec(["set_user_tags", self.username] + self.tags)
self._exec(['set_user_tags', self.username] + self.tags)
def set_permissions(self):
if not self.module.check_mode:
cmd = ["set_permissions"]
cmd = ['set_permissions']
cmd.append('-p')
cmd.append(self.permissions['vhost'])
cmd.append(self.username)
@ -168,18 +164,7 @@ class RabbitMqUser(object):
self._exec(cmd)
def has_tags_modifications(self):
if (not self._tags and len(self.tags) > 0) or (not self.tags and len(self._tags) > 0):
return True
else:
for tag in self._tags:
if tag not in self.tags:
return True
for tag in self.tags:
if tag not in self._tags:
return True
return False
return set(self.tags) != set(self._tags)
def has_permissions_modifications(self):
return self._permissions != self.permissions
@ -238,7 +223,7 @@ def main():
rabbitmq_user.set_permissions()
changed = True
module.exit_json(changed=changed)
module.exit_json(changed=changed, user=username, state=state)
# this is magic, see lib/ansible/module_common.py
#<<INCLUDE_ANSIBLE_MODULE_COMMON>>

@ -31,62 +31,105 @@ options:
- The name of the vhost to manage
required: true
default: null
aliases: [vhost]
tracing:
description:
Enable/disable tracing for a vhost
default: no
choices: [yes, no]
aliases: [trace]
state:
description:
- The state of vhost
required: true
default: null
choices: [ "present", "absent" ]
default: present
choices: [present, absent]
examples:
- code: "rabbitmq_vhost: name=/test state=present"
- code: 'rabbitmq_vhost: name=/test state=present'
description: Ensure that the vhost /test exists.
author: Matt Cordial
'''
class RabbitMqVhost(object):
def __init__(self, module, name, tracing):
self.module = module
self.name = name
self.tracing = tracing
def main():
arg_spec = dict(
name=dict(required=True),
state=dict(required=False, choices=['present', 'absent'])
)
self._tracing = False
self._rabbitmqctl = module.get_bin_path('rabbitmqctl', True)
module = AnsibleModule(argument_spec=arg_spec)
def _exec(self, args, run_in_check_mode=False):
if not self.module.check_mode or (self.module.check_mode and run_in_check_mode):
cmd = [self._rabbitmqctl, '-q']
rc, out, err = self.module.run_command(cmd + args, check_rc=True)
return out.splitlines()
return list()
name = module.params['name']
state = module.params['state']
def get(self):
vhosts = self._exec(['list_vhosts', 'name', 'tracing'], True)
RABBITMQCTL = module.get_bin_path('rabbitmqctl', True)
for vhost in vhosts:
name, tracing = vhost.split('\t')
if name == self.name:
self._tracing = self.module.boolean(tracing)
return True
return False
present = False
rc, out, err = module.run_command('%s list_vhosts' % RABBITMQCTL)
for line in out.splitlines():
if line.strip() == name:
present = True
break
def add(self):
return self._exec(['add_vhost', self.name])
if state == 'present' and present:
module.exit_json(changed=False, name=name, state=state)
def delete(self):
return self._exec(['delete_vhost', self.name])
if state == 'present' and not present:
rc, out, err = module.run_command('%s add_vhost %s' % (RABBITMQCTL, name))
if '...done' in out:
module.exit_json(changed=True, name=name, state=state)
def set_tracing(self):
if self.tracing != self._tracing:
if self.tracing:
self._enable_tracing()
else:
module.fail_json(msg=out, name=name, state=state)
self._disable_tracing()
return True
return False
def _enable_tracing(self):
return self._exec(['trace_on', '-p', self.name])
if state == 'absent' and not present:
module.exit_json(changed=False, name=name, state=state)
def _disable_tracing(self):
return self._exec(['trace_off', '-p', self.name])
if state == 'absent' and present:
rc, out, err = module.run_command('%s delete_vhost %s' % (RABBITMQCTL, name))
if '...done' in out:
module.exit_json(changed=True, name=name, state=state)
def main():
arg_spec = dict(
name=dict(required=True, aliases=['vhost']),
tracing=dict(default='off', choices=BOOLEANS, aliases=['trace']),
state=dict(default='present', choices=['present', 'absent'])
)
module = AnsibleModule(
argument_spec=arg_spec,
supports_check_mode=True
)
name = module.params['name']
tracing = module.boolean(module.params['tracing'])
state = module.params['state']
rabbitmq_vhost = RabbitMqVhost(module, name, tracing)
changed = False
if rabbitmq_vhost.get():
if state == 'absent':
rabbitmq_vhost.delete()
changed = True
else:
module.fail_json(msg=out, name=name, state=state)
if rabbitmq_vhost.set_tracing():
changed = True
elif state == 'present':
rabbitmq_vhost.add()
rabbitmq_vhost.set_tracing()
changed = True
module.exit_json(changed=False, name=name, state=state)
module.exit_json(changed=changed, name=name, state=state)
# this is magic, see lib/ansible/module_common.py
#<<INCLUDE_ANSIBLE_MODULE_COMMON>>
main()

Loading…
Cancel
Save