X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=src%2Fceph%2Fsrc%2Fpybind%2Fmgr%2Frestful%2Fcommon.py;fp=src%2Fceph%2Fsrc%2Fpybind%2Fmgr%2Frestful%2Fcommon.py;h=dc8b39b6e6e3c4dec4363f47810d254efbec8f6c;hb=812ff6ca9fcd3e629e49d4328905f33eee8ca3f5;hp=0000000000000000000000000000000000000000;hpb=15280273faafb77777eab341909a3f495cf248d9;p=stor4nfv.git

diff --git a/src/ceph/src/pybind/mgr/restful/common.py b/src/ceph/src/pybind/mgr/restful/common.py
new file mode 100644
index 0000000..dc8b39b
--- /dev/null
+++ b/src/ceph/src/pybind/mgr/restful/common.py
@@ -0,0 +1,150 @@
+# List of valid osd flags
+OSD_FLAGS = [
+    'pause', 'noup', 'nodown', 'noout', 'noin', 'nobackfill',
+    'norecover', 'noscrub', 'nodeep-scrub',
+]
+
+# Implemented osd commands
+OSD_IMPLEMENTED_COMMANDS = [
+    'scrub', 'deep_scrub', 'repair'
+]
+
+# Valid values for the 'var' argument to 'ceph osd pool set'
+POOL_PROPERTIES_1 = [
+    'size', 'min_size', 'crash_replay_interval', 'pg_num',
+    'crush_rule', 'hashpspool',
+]
+
+POOL_PROPERTIES_2 = [
+    'pgp_num'
+]
+
+POOL_PROPERTIES = POOL_PROPERTIES_1 + POOL_PROPERTIES_2
+
+# Valid values for the 'ceph osd pool set-quota' command
+POOL_QUOTA_PROPERTIES = [
+    ('quota_max_bytes', 'max_bytes'),
+    ('quota_max_objects', 'max_objects'),
+]
+
+POOL_ARGS = POOL_PROPERTIES + map(
+    lambda x: x[0],
+    POOL_QUOTA_PROPERTIES
+)
+
+
+# Transform command to a human readable form
+def humanify_command(command):
+    out = [command['prefix']]
+
+    for arg, val in command.iteritems():
+        if arg != 'prefix':
+            out.append("%s=%s" % (str(arg), str(val)))
+
+    return " ".join(out)
+
+
+def invalid_pool_args(args):
+    invalid = []
+    for arg in args:
+        if arg not in POOL_ARGS:
+            invalid.append(arg)
+
+    return invalid
+
+
+def pool_update_commands(pool_name, args):
+    commands = [[], []]
+
+    # We should increase pgp_num when we are re-setting pg_num
+    if 'pg_num' in args and 'pgp_num' not in args:
+        args['pgp_num'] = args['pg_num']
+
+    # Run the first pool set and quota properties in parallel
+    for var in POOL_PROPERTIES_1:
+        if var in args:
+            commands[0].append({
+                'prefix': 'osd pool set',
+                'pool': pool_name,
+                'var': var,
+                'val': args[var],
+            })
+
+    for (var, field) in POOL_QUOTA_PROPERTIES:
+        if var in args:
+            commands[0].append({
+                'prefix': 'osd pool set-quota',
+                'pool': pool_name,
+                'field': field,
+                'val': str(args[var]),
+            })
+
+    # The second pool set properties need to be run after the first wave
+    for var in POOL_PROPERTIES_2:
+        if var in args:
+            commands[1].append({
+                'prefix': 'osd pool set',
+                'pool': pool_name,
+                'var': var,
+            })
+
+    return commands
+
+
+def crush_rule_osds(nodes, rule):
+    nodes_by_id = dict((n['id'], n) for n in nodes)
+
+    def _gather_leaf_ids(node):
+        if node['id'] >= 0:
+            return set([node['id']])
+
+        result = set()
+        for child_id in node['children']:
+            if child_id >= 0:
+                result.add(child_id)
+            else:
+                result |= _gather_leaf_ids(nodes_by_id[child_id])
+
+        return result
+
+    def _gather_descendent_ids(node, typ):
+        result = set()
+        for child_id in node['children']:
+            child_node = nodes_by_id[child_id]
+            if child_node['type'] == typ:
+                result.add(child_node['id'])
+            elif 'children' in child_node:
+                result |= _gather_descendent_ids(child_node, typ)
+
+        return result
+
+    def _gather_osds(root, steps):
+        if root['id'] >= 0:
+            return set([root['id']])
+
+        osds = set()
+        step = steps[0]
+        if step['op'] == 'choose_firstn':
+            # Choose all descendents of the current node of type 'type'
+            d = _gather_descendent_ids(root, step['type'])
+            for desc_node in [nodes_by_id[i] for i in d]:
+                osds |= _gather_osds(desc_node, steps[1:])
+        elif step['op'] == 'chooseleaf_firstn':
+            # Choose all descendents of the current node of type 'type',
+            # and select all leaves beneath those
+            for desc_node in [nodes_by_id[i] for i in _gather_descendent_ids(root, step['type'])]:
+                # Short circuit another iteration to find the emit
+                # and assume anything we've done a chooseleaf on
+                # is going to be part of the selected set of osds
+                osds |= _gather_leaf_ids(desc_node)
+        elif step['op'] == 'emit':
+            if root['id'] >= 0:
+                osds |= root['id']
+
+        return osds
+
+    osds = set()
+    for i, step in enumerate(rule['steps']):
+        if step['op'] == 'take':
+            osds |= _gather_osds(nodes_by_id[step['item']], rule['steps'][i + 1:])
+    return osds