mirror of
https://github.com/ceph/ceph
synced 2024-12-24 04:14:07 +00:00
418 lines
16 KiB
Python
418 lines
16 KiB
Python
"""
|
|
rgw multisite configuration routines
|
|
"""
|
|
import argparse
|
|
import contextlib
|
|
import logging
|
|
import random
|
|
import string
|
|
from copy import deepcopy
|
|
from util.rgw import rgwadmin, wait_for_radosgw
|
|
from util.rados import create_ec_pool, create_replicated_pool
|
|
from rgw_multi import multisite
|
|
|
|
from teuthology.orchestra import run
|
|
from teuthology import misc
|
|
from teuthology.exceptions import ConfigError
|
|
from teuthology.task import Task
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
class RGWMultisite(Task):
|
|
"""
|
|
Performs rgw multisite configuration to match the given realm definition.
|
|
|
|
- rgw-multisite:
|
|
realm:
|
|
name: test-realm
|
|
is_default: true
|
|
|
|
List one or more zonegroup definitions. These are provided as json
|
|
input to `radosgw-admin zonegroup set`, with the exception of these keys:
|
|
|
|
* 'is_master' is passed on the command line as --master
|
|
* 'is_default' is passed on the command line as --default
|
|
* 'endpoints' given as client names are replaced with actual endpoints
|
|
|
|
zonegroups:
|
|
- name: test-zonegroup
|
|
api_name: test-api
|
|
is_master: true
|
|
is_default: true
|
|
endpoints: [c1.client.0]
|
|
|
|
List each of the zones to be created in this zonegroup.
|
|
|
|
zones:
|
|
- name: test-zone1
|
|
is_master: true
|
|
is_default: true
|
|
endpoints: [c1.client.0]
|
|
- name: test-zone2
|
|
is_default: true
|
|
endpoints: [c2.client.0]
|
|
|
|
A complete example:
|
|
|
|
tasks:
|
|
- install:
|
|
- ceph: {cluster: c1}
|
|
- ceph: {cluster: c2}
|
|
- rgw:
|
|
c1.client.0:
|
|
c2.client.0:
|
|
- rgw-multisite:
|
|
realm:
|
|
name: test-realm
|
|
is_default: true
|
|
zonegroups:
|
|
- name: test-zonegroup
|
|
is_master: true
|
|
is_default: true
|
|
zones:
|
|
- name: test-zone1
|
|
is_master: true
|
|
is_default: true
|
|
endpoints: [c1.client.0]
|
|
- name: test-zone2
|
|
is_default: true
|
|
endpoints: [c2.client.0]
|
|
|
|
"""
|
|
def __init__(self, ctx, config):
|
|
super(RGWMultisite, self).__init__(ctx, config)
|
|
|
|
def setup(self):
|
|
super(RGWMultisite, self).setup()
|
|
|
|
overrides = self.ctx.config.get('overrides', {})
|
|
misc.deep_merge(self.config, overrides.get('rgw-multisite', {}))
|
|
|
|
if not self.ctx.rgw:
|
|
raise ConfigError('rgw-multisite must run after the rgw task')
|
|
role_endpoints = self.ctx.rgw.role_endpoints
|
|
|
|
# construct Clusters and Gateways for each client in the rgw task
|
|
clusters, gateways = extract_clusters_and_gateways(self.ctx,
|
|
role_endpoints)
|
|
|
|
# get the master zone and zonegroup configuration
|
|
mz, mzg = extract_master_zone_zonegroup(self.config['zonegroups'])
|
|
cluster1 = cluster_for_zone(clusters, mz)
|
|
|
|
# create the realm and period on the master zone's cluster
|
|
log.info('creating realm..')
|
|
realm = create_realm(cluster1, self.config['realm'])
|
|
period = realm.current_period
|
|
|
|
creds = gen_credentials()
|
|
|
|
# create the master zonegroup and its master zone
|
|
log.info('creating master zonegroup..')
|
|
master_zonegroup = create_zonegroup(cluster1, gateways, period,
|
|
deepcopy(mzg))
|
|
period.master_zonegroup = master_zonegroup
|
|
|
|
log.info('creating master zone..')
|
|
master_zone = create_zone(self.ctx, cluster1, gateways, creds,
|
|
master_zonegroup, deepcopy(mz))
|
|
master_zonegroup.master_zone = master_zone
|
|
|
|
period.update(master_zone, commit=True)
|
|
restart_zone_gateways(master_zone) # restart with --rgw-zone
|
|
|
|
# create the admin user on the master zone
|
|
log.info('creating admin user..')
|
|
user_args = ['--display-name', 'Realm Admin', '--system']
|
|
user_args += creds.credential_args()
|
|
admin_user = multisite.User('realm-admin')
|
|
admin_user.create(master_zone, user_args)
|
|
|
|
# process 'zonegroups'
|
|
for zg_config in self.config['zonegroups']:
|
|
zones_config = zg_config.pop('zones')
|
|
|
|
zonegroup = None
|
|
for zone_config in zones_config:
|
|
# get the cluster for this zone
|
|
cluster = cluster_for_zone(clusters, zone_config)
|
|
|
|
if cluster != cluster1: # already created on master cluster
|
|
log.info('pulling realm configuration to %s', cluster.name)
|
|
realm.pull(cluster, master_zone.gateways[0], creds)
|
|
|
|
# use the first zone's cluster to create the zonegroup
|
|
if not zonegroup:
|
|
if zg_config['name'] == master_zonegroup.name:
|
|
zonegroup = master_zonegroup
|
|
else:
|
|
log.info('creating zonegroup..')
|
|
zonegroup = create_zonegroup(cluster, gateways,
|
|
period, zg_config)
|
|
|
|
if zone_config['name'] == master_zone.name:
|
|
# master zone was already created
|
|
zone = master_zone
|
|
else:
|
|
# create the zone and commit the period
|
|
log.info('creating zone..')
|
|
zone = create_zone(self.ctx, cluster, gateways, creds,
|
|
zonegroup, zone_config)
|
|
period.update(zone, commit=True)
|
|
|
|
restart_zone_gateways(zone) # restart with --rgw-zone
|
|
|
|
# attach configuration to the ctx for other tasks
|
|
self.ctx.rgw_multisite = argparse.Namespace()
|
|
self.ctx.rgw_multisite.clusters = clusters
|
|
self.ctx.rgw_multisite.gateways = gateways
|
|
self.ctx.rgw_multisite.realm = realm
|
|
self.ctx.rgw_multisite.admin_user = admin_user
|
|
|
|
log.info('rgw multisite configuration completed')
|
|
|
|
def end(self):
|
|
del self.ctx.rgw_multisite
|
|
|
|
class Cluster(multisite.Cluster):
|
|
""" Issues 'radosgw-admin' commands with the rgwadmin() helper """
|
|
def __init__(self, ctx, name, client):
|
|
super(Cluster, self).__init__()
|
|
self.ctx = ctx
|
|
self.name = name
|
|
self.client = client
|
|
|
|
def admin(self, args = None, **kwargs):
|
|
""" radosgw-admin command """
|
|
args = args or []
|
|
args += ['--cluster', self.name]
|
|
args += ['--debug-rgw', '0']
|
|
if kwargs.pop('read_only', False):
|
|
args += ['--rgw-cache-enabled', 'false']
|
|
kwargs['decode'] = False
|
|
check_retcode = kwargs.pop('check_retcode', True)
|
|
r, s = rgwadmin(self.ctx, self.client, args, **kwargs)
|
|
if check_retcode:
|
|
assert r == 0
|
|
return s, r
|
|
|
|
class Gateway(multisite.Gateway):
|
|
""" Controls a radosgw instance using its daemon """
|
|
def __init__(self, role, remote, daemon, *args, **kwargs):
|
|
super(Gateway, self).__init__(*args, **kwargs)
|
|
self.role = role
|
|
self.remote = remote
|
|
self.daemon = daemon
|
|
|
|
def set_zone(self, zone):
|
|
""" set the zone and add its args to the daemon's command line """
|
|
assert self.zone is None, 'zone can only be set once'
|
|
self.zone = zone
|
|
# daemon.restart_with_args() would be perfect for this, except that
|
|
# radosgw args likely include a pipe and redirect. zone arguments at
|
|
# the end won't actually apply to radosgw
|
|
args = self.daemon.command_kwargs.get('args', [])
|
|
try:
|
|
# insert zone args before the first |
|
|
pipe = args.index(run.Raw('|'))
|
|
args = args[0:pipe] + zone.zone_args() + args[pipe:]
|
|
except ValueError, e:
|
|
args += zone.zone_args()
|
|
self.daemon.command_kwargs['args'] = args
|
|
|
|
def start(self, args = None):
|
|
""" (re)start the daemon """
|
|
self.daemon.restart()
|
|
# wait until startup completes
|
|
wait_for_radosgw(self.endpoint())
|
|
|
|
def stop(self):
|
|
""" stop the daemon """
|
|
self.daemon.stop()
|
|
|
|
def extract_clusters_and_gateways(ctx, role_endpoints):
|
|
""" create cluster and gateway instances for all of the radosgw roles """
|
|
clusters = {}
|
|
gateways = {}
|
|
for role, (host, port) in role_endpoints.iteritems():
|
|
cluster_name, daemon_type, client_id = misc.split_role(role)
|
|
# find or create the cluster by name
|
|
cluster = clusters.get(cluster_name)
|
|
if not cluster:
|
|
clusters[cluster_name] = cluster = Cluster(ctx, cluster_name, role)
|
|
# create a gateway for this daemon
|
|
client_with_id = daemon_type + '.' + client_id # match format from rgw.py
|
|
daemon = ctx.daemons.get_daemon('rgw', client_with_id, cluster_name)
|
|
if not daemon:
|
|
raise ConfigError('no daemon for role=%s cluster=%s type=rgw id=%s' % \
|
|
(role, cluster_name, client_id))
|
|
(remote,) = ctx.cluster.only(role).remotes.keys()
|
|
gateways[role] = Gateway(role, remote, daemon, host, port, cluster)
|
|
return clusters, gateways
|
|
|
|
def create_realm(cluster, config):
|
|
""" create a realm from configuration and initialize its first period """
|
|
realm = multisite.Realm(config['name'])
|
|
args = []
|
|
if config.get('is_default', False):
|
|
args += ['--default']
|
|
realm.create(cluster, args)
|
|
realm.current_period = multisite.Period(realm)
|
|
return realm
|
|
|
|
def extract_user_credentials(config):
|
|
""" extract keys from configuration """
|
|
return multisite.Credentials(config['access_key'], config['secret_key'])
|
|
|
|
def extract_master_zone(zonegroup_config):
|
|
""" find and return the master zone definition """
|
|
master = None
|
|
for zone in zonegroup_config['zones']:
|
|
if not zone.get('is_master', False):
|
|
continue
|
|
if master:
|
|
raise ConfigError('zones %s and %s cannot both set \'is_master\'' % \
|
|
(master['name'], zone['name']))
|
|
master = zone
|
|
# continue the loop so we can detect duplicates
|
|
if not master:
|
|
raise ConfigError('one zone must set \'is_master\' in zonegroup %s' % \
|
|
zonegroup_config['name'])
|
|
return master
|
|
|
|
def extract_master_zone_zonegroup(zonegroups_config):
|
|
""" find and return the master zone and zonegroup definitions """
|
|
master_zone, master_zonegroup = (None, None)
|
|
for zonegroup in zonegroups_config:
|
|
# verify that all zonegroups have a master zone set, even if they
|
|
# aren't in the master zonegroup
|
|
zone = extract_master_zone(zonegroup)
|
|
if not zonegroup.get('is_master', False):
|
|
continue
|
|
if master_zonegroup:
|
|
raise ConfigError('zonegroups %s and %s cannot both set \'is_master\'' % \
|
|
(master_zonegroup['name'], zonegroup['name']))
|
|
master_zonegroup = zonegroup
|
|
master_zone = zone
|
|
# continue the loop so we can detect duplicates
|
|
if not master_zonegroup:
|
|
raise ConfigError('one zonegroup must set \'is_master\'')
|
|
return master_zone, master_zonegroup
|
|
|
|
def extract_zone_cluster_name(zone_config):
|
|
""" return the cluster (must be common to all zone endpoints) """
|
|
cluster_name = None
|
|
endpoints = zone_config.get('endpoints')
|
|
if not endpoints:
|
|
raise ConfigError('zone %s missing \'endpoints\' list' % \
|
|
zone_config['name'])
|
|
for role in endpoints:
|
|
name, _, _ = misc.split_role(role)
|
|
if not cluster_name:
|
|
cluster_name = name
|
|
elif cluster_name != name:
|
|
raise ConfigError('all zone %s endpoints must be in the same cluster' % \
|
|
zone_config['name'])
|
|
return cluster_name
|
|
|
|
def cluster_for_zone(clusters, zone_config):
|
|
""" return the cluster entry for the given zone """
|
|
name = extract_zone_cluster_name(zone_config)
|
|
try:
|
|
return clusters[name]
|
|
except KeyError:
|
|
raise ConfigError('no cluster %s found' % name)
|
|
|
|
def gen_access_key():
|
|
return ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(16))
|
|
|
|
def gen_secret():
|
|
return ''.join(random.choice(string.ascii_uppercase + string.ascii_lowercase + string.digits) for _ in range(32))
|
|
|
|
def gen_credentials():
|
|
return multisite.Credentials(gen_access_key(), gen_secret())
|
|
|
|
def extract_gateway_endpoints(gateways, endpoints_config):
|
|
""" return a list of gateway endpoints associated with the given roles """
|
|
endpoints = []
|
|
for role in endpoints_config:
|
|
try:
|
|
# replace role names with their gateway's endpoint
|
|
endpoints.append(gateways[role].endpoint())
|
|
except KeyError:
|
|
raise ConfigError('no radosgw endpoint found for role %s' % role)
|
|
return endpoints
|
|
|
|
def is_default_arg(config):
|
|
return ['--default'] if config.pop('is_default', False) else []
|
|
|
|
def is_master_arg(config):
|
|
return ['--master'] if config.pop('is_master', False) else []
|
|
|
|
def create_zonegroup(cluster, gateways, period, config):
|
|
""" pass the zonegroup configuration to `zonegroup set` """
|
|
config.pop('zones', None) # remove 'zones' from input to `zonegroup set`
|
|
endpoints = config.get('endpoints')
|
|
if endpoints:
|
|
# replace client names with their gateway endpoints
|
|
config['endpoints'] = extract_gateway_endpoints(gateways, endpoints)
|
|
zonegroup = multisite.ZoneGroup(config['name'], period)
|
|
# `zonegroup set` needs --default on command line, and 'is_master' in json
|
|
args = is_default_arg(config)
|
|
zonegroup.set(cluster, config, args)
|
|
period.zonegroups.append(zonegroup)
|
|
return zonegroup
|
|
|
|
def create_zone(ctx, cluster, gateways, creds, zonegroup, config):
|
|
""" create a zone with the given configuration """
|
|
zone = multisite.Zone(config['name'], zonegroup, cluster)
|
|
|
|
# collect Gateways for the zone's endpoints
|
|
endpoints = config.get('endpoints')
|
|
if not endpoints:
|
|
raise ConfigError('no \'endpoints\' for zone %s' % config['name'])
|
|
zone.gateways = [gateways[role] for role in endpoints]
|
|
for gateway in zone.gateways:
|
|
gateway.set_zone(zone)
|
|
|
|
# format the gateway endpoints
|
|
endpoints = [g.endpoint() for g in zone.gateways]
|
|
|
|
args = is_default_arg(config)
|
|
args += is_master_arg(config)
|
|
args += creds.credential_args()
|
|
if len(endpoints):
|
|
args += ['--endpoints', ','.join(endpoints)]
|
|
zone.create(cluster, args)
|
|
zonegroup.zones.append(zone)
|
|
|
|
create_zone_pools(ctx, zone)
|
|
if ctx.rgw.compression_type:
|
|
configure_zone_compression(zone, ctx.rgw.compression_type)
|
|
return zone
|
|
|
|
def create_zone_pools(ctx, zone):
|
|
""" Create the data_pool for each placement type """
|
|
gateway = zone.gateways[0]
|
|
cluster = zone.cluster
|
|
for pool_config in zone.data.get('placement_pools', []):
|
|
pool_name = pool_config['val']['data_pool']
|
|
if ctx.rgw.ec_data_pool:
|
|
create_ec_pool(gateway.remote, pool_name, zone.name, 64,
|
|
ctx.rgw.erasure_code_profile, cluster.name)
|
|
else:
|
|
create_replicated_pool(gateway.remote, pool_name, 64, cluster.name)
|
|
|
|
def configure_zone_compression(zone, compression):
|
|
""" Set compression type in the zone's default-placement """
|
|
zone.json_command(zone.cluster, 'placement', ['modify',
|
|
'--placement-id', 'default-placement',
|
|
'--compression', compression
|
|
])
|
|
|
|
def restart_zone_gateways(zone):
|
|
zone.stop()
|
|
zone.start()
|
|
|
|
task = RGWMultisite
|