# Copyright 2020, Red Hat, Inc. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. from __future__ import absolute_import, division, print_function __metaclass__ = type from ansible.module_utils.basic import AnsibleModule try: from ansible.module_utils.ca_common import is_containerized, \ generate_cmd, \ exec_command, \ exit_module except ImportError: from module_utils.ca_common import is_containerized, \ generate_cmd, \ exec_command, \ exit_module import datetime import json ANSIBLE_METADATA = { 'metadata_version': '1.1', 'status': ['preview'], 'supported_by': 'community' } DOCUMENTATION = ''' --- module: ceph_ec_profile short_description: Manage Ceph Erasure Code profile version_added: "2.8" description: - Manage Ceph Erasure Code profile options: cluster: description: - The ceph cluster name. required: false default: ceph name: description: - name of the profile. required: true state: description: If 'present' is used, the module creates a profile. If 'absent' is used, the module will delete the profile. required: false choices: ['present', 'absent', 'info'] default: present stripe_unit: description: - The amount of data in a data chunk, per stripe. required: false k: description: - Number of data-chunks the object will be split in required: true m: description: - Compute coding chunks for each object and store them on different OSDs. required: true crush_device_class: description: - Restrict placement to devices of a specific class (hdd/ssd) required: false author: - Guillaume Abrioux ''' EXAMPLES = ''' - name: create an erasure code profile ceph_ec_profile: name: foo k: 4 m: 2 - name: delete an erassure code profile ceph_ec_profile: name: foo state: absent ''' RETURN = '''# ''' def get_profile(name, cluster='ceph', container_image=None): ''' Get existing profile ''' args = ['get', name, '--format=json'] cmd = generate_cmd(sub_cmd=['osd', 'erasure-code-profile'], args=args, cluster=cluster, container_image=container_image) return cmd def create_profile(name, user_profile, force, cluster='ceph', container_image=None): # noqa: E501 ''' Create a profile ''' args = ['set', name] for key, value in user_profile.items(): args.append('{}={}'.format(key, value)) if force: args.append('--force') cmd = generate_cmd(sub_cmd=['osd', 'erasure-code-profile'], args=args, cluster=cluster, container_image=container_image) return cmd def delete_profile(name, cluster='ceph', container_image=None): ''' Delete a profile ''' args = ['rm', name] cmd = generate_cmd(sub_cmd=['osd', 'erasure-code-profile'], args=args, cluster=cluster, container_image=container_image) return cmd def parse_user_profile(module): profile_keys = ['plugin', 'k', 'm', 'd', 'l', 'c', 'stripe_unit', 'scalar_mds', 'technique', 'crush-root', 'crush-device-class', 'crush-failure-domain'] profile = {} for key in profile_keys: ansible_lookup_key = key.replace('-', '_') value = module.params.get(ansible_lookup_key) if value: profile[key] = value return profile def run_module(): module_args = dict( cluster=dict(type='str', required=False, default='ceph'), name=dict(type='str', required=True), state=dict(type='str', required=False, choices=['present', 'absent'], default='present'), stripe_unit=dict(type='str', required=False), plugin=dict(type='str', required=False, default='jerasure'), k=dict(type='int', required=False), m=dict(type='int', required=False), d=dict(type='int', required=False), l=dict(type='int', required=False), c=dict(type='int', required=False), scalar_mds=dict(type='str', required=False), technique=dict(type='str', required=False), crush_root=dict(type='str', required=False), crush_failure_domain=dict(type='str', required=False), crush_device_class=dict(type='str', required=False), force=dict(type='bool', required=False, default=False), ) module = AnsibleModule( argument_spec=module_args, supports_check_mode=True, required_if=[['state', 'present', ['k', 'm']]], ) # Gather module parameters in variables name = module.params.get('name') cluster = module.params.get('cluster') state = module.params.get('state') force = module.params.get('force') user_profile = parse_user_profile(module) if module.check_mode: module.exit_json( changed=False, stdout='', stderr='', rc=0, start='', end='', delta='', ) startd = datetime.datetime.now() diff = dict(before="", after="") changed = False # will return either the image name or None container_image = is_containerized() if state == "present": rc, cmd, out, err = exec_command(module, get_profile(name, cluster, container_image=container_image)) # noqa: E501 current_profile = {} if rc == 0: current_profile = json.loads(out) changed = current_profile != user_profile if changed: diff['before'] = json.dumps(current_profile) diff['after'] = json.dumps(user_profile) rc, cmd, out, err = exec_command(module, create_profile(name, user_profile, force, cluster, container_image=container_image), # noqa: E501 check_rc=True) elif state == "absent": rc, cmd, out, err = exec_command(module, delete_profile(name, cluster, container_image=container_image)) # noqa: E501 if not err: out = 'Profile {} removed.'.format(name) changed = True else: rc = 0 out = "Skipping, the profile {} doesn't exist".format(name) exit_module(module=module, out=out, rc=rc, cmd=cmd, err=err, startd=startd, changed=changed, diff=diff) # noqa: E501 def main(): run_module() if __name__ == '__main__': main()