Skip to content

Commit

Permalink
radosgw_zone: add support zone set
Browse files Browse the repository at this point in the history 
Support zone set from a json doc

Signed-off-by: Seena Fallah <[email protected]>
  • Loading branch information
@clwluvw @guits
clwluvw authored and guits committed Mar 14, 2024
1 parent 2a8b066 commit e219892
Show file tree
Hide file tree
Showing 2 changed files with 109 additions and 19 deletions.
80 changes: 61 additions & 19 deletions library/radosgw_zone.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -124,20 +124,23 @@
RETURN = '''# '''


def container_exec(binary, container_image):
def container_exec(binary, container_image, container_args=[]):
'''
Build the docker CLI to run a command inside a container
'''

container_binary = os.getenv('CEPH_CONTAINER_BINARY')
command_exec = [container_binary,
'run',
'--rm',
'--net=host',
'-v', '/etc/ceph:/etc/ceph:z',
'-v', '/var/lib/ceph/:/var/lib/ceph/:z',
'-v', '/var/log/ceph/:/var/log/ceph/:z',
'--entrypoint=' + binary, container_image]

command_exec = [container_binary, 'run', '--rm', '--net=host']
command_exec.extend(container_args)
command_exec.extend([
'-v', '/etc/ceph:/etc/ceph:z',
'-v', '/var/lib/ceph/:/var/lib/ceph/:z',
'-v', '/var/log/ceph/:/var/log/ceph/:z',
'--entrypoint=' + binary,
container_image,
])

return command_exec


Expand All @@ -154,24 +157,24 @@ def is_containerized():
return container_image


def pre_generate_radosgw_cmd(container_image=None):
def pre_generate_radosgw_cmd(container_image=None, container_args=[]):
'''
Generate radosgw-admin prefix comaand
'''
if container_image:
cmd = container_exec('radosgw-admin', container_image)
cmd = container_exec('radosgw-admin', container_image, container_args)
else:
cmd = ['radosgw-admin']

return cmd


def generate_radosgw_cmd(cluster, args, container_image=None):
def generate_radosgw_cmd(cluster, args, container_image=None, container_args=[]):
'''
Generate 'radosgw' command line to execute
'''

cmd = pre_generate_radosgw_cmd(container_image=container_image)
cmd = pre_generate_radosgw_cmd(container_image=container_image, container_args=container_args) # noqa: E501

base_cmd = [
'--cluster',
Expand Down Expand Up @@ -383,6 +386,37 @@ def remove_zone(module, container_image=None):
return cmd


def set_zone(module, container_image=None):
'''
Set a zone
'''

cluster = module.params.get('cluster')
realm = module.params.get('realm')
zone_doc = module.params.get('zone_doc')

# store the zone_doc in a file
filename = module.tmpdir + 'zone_doc.json'
with open(filename, 'w') as f:
json.dump(zone_doc, f)

container_args = [
'-v', filename + ':' + filename + ':ro'
]
args = [
'set',
'--rgw-realm=' + realm,
'--infile=' + filename,
]

cmd = generate_radosgw_cmd(cluster=cluster,
args=args,
container_image=container_image,
container_args=container_args)

return cmd


def exit_module(module, out, rc, cmd, err, startd, changed=False):
endd = datetime.datetime.now()
delta = endd - startd
Expand All @@ -404,14 +438,15 @@ 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', 'info'], default='present'), # noqa: E501
state=dict(type='str', required=False, choices=['present', 'absent', 'info', 'set'], default='present'), # noqa: E501
realm=dict(type='str', require=True),
zonegroup=dict(type='str', require=True),
endpoints=dict(type='list', require=False, default=[]),
access_key=dict(type='str', required=False, no_log=True),
secret_key=dict(type='str', required=False, no_log=True),
default=dict(type='bool', required=False, default=False),
master=dict(type='bool', required=False, default=False),
zone_doc=dict(type='dict', required=False, default={})
)

module = AnsibleModule(
Expand Down Expand Up @@ -443,8 +478,19 @@ def run_module():
# will return either the image name or None
container_image = is_containerized()

rc, cmd, out, err = exec_commands(module, get_zone(module, container_image=container_image)) # noqa: E501

if state == "set":
zone = json.loads(out) if rc == 0 else {}
zone_doc = module.params.get('zone_doc')
if not zone_doc:
fatal("zone_doc is required when state is set", module)

changed = zone_doc != zone
if changed:
rc, cmd, out, err = exec_commands(module, set_zone(module, container_image=container_image))

if state == "present":
rc, cmd, out, err = exec_commands(module, get_zone(module, container_image=container_image)) # noqa: E501
if rc == 0:
zone = json.loads(out)
_rc, _cmd, _out, _err = exec_commands(module, get_realm(module, container_image=container_image)) # noqa: E501
Expand Down Expand Up @@ -479,17 +525,13 @@ def run_module():
changed = True

elif state == "absent":
rc, cmd, out, err = exec_commands(module, get_zone(module, container_image=container_image)) # noqa: E501
if rc == 0:
rc, cmd, out, err = exec_commands(module, remove_zone(module, container_image=container_image)) # noqa: E501
changed = True
else:
rc = 0
out = "Zone {} doesn't exist".format(name)

elif state == "info":
rc, cmd, out, err = exec_commands(module, get_zone(module, container_image=container_image)) # noqa: E501

exit_module(module=module, out=out, rc=rc, cmd=cmd, err=err, startd=startd, changed=changed) # noqa: E501


Expand Down
48 changes: 48 additions & 0 deletions tests/library/test_radosgw_zone.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -73,6 +73,32 @@ def test_generate_radosgw_cmd(self, image):
])
assert radosgw_zone.generate_radosgw_cmd(fake_cluster, [], image) == expected_cmd

@pytest.mark.parametrize('image', fake_container_image)
@patch.dict(os.environ, {'CEPH_CONTAINER_BINARY': fake_container_binary})
def test_generate_radosgw_cmd_container_args(self, image):
container_args = [
'-v', '/test:/test:ro',
]
expected_cmd = [
fake_container_binary,
'run',
'--rm',
'--net=host',
'-v', '/test:/test:ro',
'-v', '/etc/ceph:/etc/ceph:z',
'-v', '/var/lib/ceph/:/var/lib/ceph/:z',
'-v', '/var/log/ceph/:/var/log/ceph/:z',
'--entrypoint=' + fake_binary,
fake_container_image
]

expected_cmd.extend([
'--cluster',
fake_cluster,
'zone'
])
assert radosgw_zone.generate_radosgw_cmd(fake_cluster, [], image, container_args) == expected_cmd

def test_create_zone(self):
fake_module = MagicMock()
fake_module.params = fake_params
Expand Down Expand Up @@ -163,3 +189,25 @@ def test_remove_zone(self):
]

assert radosgw_zone.remove_zone(fake_module) == expected_cmd

def test_set_zone(self):
fake_module = MagicMock()
fake_module.params = {
'cluster': fake_cluster,
'name': fake_zone,
'realm': fake_realm,
'zonegroup': fake_zonegroup,
'zone_doc': {'id': 'fake_id'},
}

zonefile = fake_module.tmpdir + '/zone.json'

expected_cmd = [
fake_binary,
'--cluster', fake_cluster,
'zone', 'set',
'--rgw-realm=' + fake_realm,
'--infile=' + zonefile,
]

assert radosgw_zone.set_zone(fake_module) == expected_cmd

0 comments on commit e219892

Please sign in to comment.