From 47ba1711d70412409d9036b05eae452246f60d43 Mon Sep 17 00:00:00 2001
From: Josh Durgin <josh.durgin@inktank.com>
Date: Mon, 24 Mar 2014 18:19:41 -0700
Subject: [PATCH] rgw: add option to use erasure coding instead of replication

Signed-off-by: Josh Durgin <josh.durgin@inktank.com>
---
 teuthology/task/rgw.py        | 36 ++++++++++++++++++++++++++++++++---
 teuthology/task_util/rados.py | 16 ++++++++++++++++
 2 files changed, 49 insertions(+), 3 deletions(-)

diff --git a/teuthology/task/rgw.py b/teuthology/task/rgw.py
index 12700f60a2a71..de818ed91bc41 100644
--- a/teuthology/task/rgw.py
+++ b/teuthology/task/rgw.py
@@ -13,7 +13,7 @@ from ..orchestra import run
 from teuthology import misc as teuthology
 from teuthology import contextutil
 from teuthology.task_util.rgw import rgwadmin
-from teuthology.task_util.rados import rados
+from teuthology.task_util.rados import rados, create_ec_pool, create_replicated_pool
 
 log = logging.getLogger(__name__)
 
@@ -462,6 +462,23 @@ def configure_users(ctx, config, everywhere=False):
 
     yield
 
+@contextlib.contextmanager
+def create_nonregion_pools(ctx, config, regions):
+    """Create replicated or erasure coded data pools for rgw."""
+    if regions:
+        yield
+        return
+
+    log.info('creating data pools')
+    for client in config.keys():
+        (remote,) = ctx.cluster.only(client).remotes.iterkeys()
+        data_pool = '.rgw.buckets'
+        if ctx.rgw.ec_data_pool:
+            create_ec_pool(remote, data_pool, client, 64)
+        else:
+            create_replicated_pool(remote, data_pool, 64)
+    yield
+
 @contextlib.contextmanager
 def configure_regions_and_zones(ctx, config, regions, role_endpoints):
     """
@@ -515,12 +532,18 @@ def configure_regions_and_zones(ctx, config, regions, role_endpoints):
             rados(ctx, mon,
                   cmd=['-p', zone_info['domain_root'],
                        'rm', 'zone_info.default'])
+
             (remote,) = ctx.cluster.only(role).remotes.keys()
             for pool_info in zone_info['placement_pools']:
                 remote.run(args=['ceph', 'osd', 'pool', 'create',
                                  pool_info['val']['index_pool'], '64', '64'])
-                remote.run(args=['ceph', 'osd', 'pool', 'create',
-                                 pool_info['val']['data_pool'], '64', '64'])
+                if ctx.get('ec-data-pool'):
+                    create_ec_pool(remote, pool_info['val']['data_pool'],
+                                   zone, 64)
+                else:
+                    create_replicated_pool(remote, pool_info['val']['data_pool'],
+                                           64)
+
             rgwadmin(ctx, client,
                      cmd=['-n', client, 'zone', 'set', '--rgw-zone', zone],
                      stdin=StringIO(json.dumps(dict(zone_info.items() + user_info.items()))),
@@ -626,6 +649,7 @@ def task(ctx, config):
                 rgw region root pool: .rgw.rroot.bar
                 rgw zone root pool: .rgw.zroot.bar-secondary 
         - rgw:
+            ec-data-pool: true
             regions:
               foo:
                 api name: api_name # default: region name
@@ -673,6 +697,11 @@ def task(ctx, config):
     # stash the region info for later, since it was deleted from the config structure
     ctx.rgw.regions = regions
 
+    ctx.rgw.ec_data_pool = False
+    if 'ec-data-pool' in config:
+        ctx.rgw.ec_data_pool = bool(config['ec-data-pool'])
+        del config['ec-data-pool']
+
     with contextutil.nested(
         lambda: create_dirs(ctx=ctx, config=config),
         lambda: configure_regions_and_zones(
@@ -686,6 +715,7 @@ def task(ctx, config):
             config=config,
             everywhere=bool(regions),
             ),
+        lambda: create_nonregion_pools(ctx=ctx, config=config, regions=regions),
         lambda: ship_config(ctx=ctx, config=config,
                             role_endpoints=role_endpoints),
         lambda: start_rgw(ctx=ctx, config=config),
diff --git a/teuthology/task_util/rados.py b/teuthology/task_util/rados.py
index 1ce4cefd71e25..12aa667a62b20 100644
--- a/teuthology/task_util/rados.py
+++ b/teuthology/task_util/rados.py
@@ -23,3 +23,19 @@ def rados(ctx, remote, cmd, wait=True, check_status=False):
         return proc.exitstatus
     else:
         return proc
+
+def create_ec_pool(remote, name, profile_name, pgnum, m=1, k=2):
+    remote.run(args=[
+        'ceph', 'osd', 'erasure-code-profile', 'set',
+        profile_name, 'm=' + str(m), 'k=' + str(k),
+        'ruleset-failure-domain=osd',
+        ])
+    remote.run(args=[
+        'ceph', 'osd', 'pool', 'create', name,
+        str(pgnum), str(pgnum), 'erasure', profile_name,
+        ])
+
+def create_replicated_pool(remote, name, pgnum):
+    remote.run(args=[
+        'ceph', 'osd', 'pool', 'create', name, str(pgnum), str(pgnum),
+        ])
-- 
2.39.5