From 23c729305a6af8f9075e070d382573db18e29ba9 Mon Sep 17 00:00:00 2001
From: Eleanor Cawthon <eleanor.cawthon@inktank.com>
Date: Tue, 5 Jun 2012 15:30:51 -0700
Subject: [PATCH] task/: Added object map benchmarking test

Signed-off-by: Eleanor Cawthon <eleanor.cawthon@inktank.com>
---
 teuthology/task/omapbench.py | 81 ++++++++++++++++++++++++++++++++++++
 1 file changed, 81 insertions(+)
 create mode 100644 teuthology/task/omapbench.py

diff --git a/teuthology/task/omapbench.py b/teuthology/task/omapbench.py
new file mode 100644
index 0000000000000..e01d42ff924a7
--- /dev/null
+++ b/teuthology/task/omapbench.py
@@ -0,0 +1,81 @@
+import contextlib
+import logging
+
+from ..orchestra import run
+
+log = logging.getLogger(__name__)
+
+@contextlib.contextmanager
+def task(ctx, config):
+    """
+    Run omapbench
+
+    The config should be as follows::
+
+		  omapbench:
+		      clients: [client list]
+		      threads: <threads at once>
+		      objects: <number of objects to write>
+		      entries: <number of entries per object map>
+		      keysize: <number of characters per object map key>
+		      valsize: <number of characters per object map val>
+		      increment: <interval to show in histogram (in ms)>
+		      omaptype: <how the omaps should be generated>
+
+    example::
+
+		  tasks:
+		  - ceph:
+		  - omapbench:
+		      clients: [client.0]
+		      threads: 30
+		      objects: 1000
+		      entries: 10
+		      keysize: 10
+		      valsize: 100
+		      increment: 100
+		      omaptype: uniform
+		  - interactive:
+    """
+    log.info('Beginning omapbench...')
+    assert isinstance(config, dict), \
+        "please list clients to run on"
+    omapbench = {}
+    print(str(config.get('increment',-1)))
+    for role in config.get('clients', ['client.0']):
+        assert isinstance(role, basestring)
+        PREFIX = 'client.'
+        assert role.startswith(PREFIX)
+        id_ = role[len(PREFIX):]
+        (remote,) = ctx.cluster.only(role).remotes.iterkeys()
+        proc = remote.run(
+            args=[
+                "/bin/sh", "-c",
+                " ".join(['CEPH_CONF=/tmp/cephtest/ceph.conf',
+                          'LD_LIBRARY_PATH=/tmp/cephtest/binary/usr/local/lib',
+                          '/tmp/cephtest/enable-coredump',
+                          '/tmp/cephtest/binary/usr/local/bin/ceph-coverage',
+                          '/tmp/cephtest/archive/coverage',
+                          '/tmp/cephtest/binary/usr/local/bin/omapbench',
+                          '-k', '/tmp/cephtest/data/{role}.keyring'.format(role=role),
+                          '--name', role[len(PREFIX):],
+                          '-t', str(config.get('threads', 30)),
+                          '-o', str(config.get('objects', 1000)),
+                          '--entries', str(config.get('entries',10)),
+                          '--keysize', str(config.get('keysize',10)),
+                          '--valsize', str(config.get('valsize',1000)),
+                          '--inc', str(config.get('increment',10)),
+                          '--omaptype', str(config.get('omaptype','uniform'))
+                          ]),
+                ],
+            logger=log.getChild('omapbench.{id}'.format(id=id_)),
+            stdin=run.PIPE,
+            wait=False
+            )
+        omapbench[id_] = proc
+
+    try:
+        yield
+    finally:
+        log.info('joining omapbench')
+        run.wait(omapbench.itervalues())
-- 
2.39.5