update sources to v12.1.2

[ceph.git] / ceph / src / pybind / mgr / dashboard / rbd_ls.py

diff --git a/ceph/src/pybind/mgr/dashboard/rbd_ls.py b/ceph/src/pybind/mgr/dashboard/rbd_ls.py
index 510a7777820368a9628b47588ab532219c340ec6..6588766a7dad18962e8de2d5dff28f84b093bc6e 100644 (file)
--- a/ceph/src/pybind/mgr/dashboard/rbd_ls.py
+++ b/ceph/src/pybind/mgr/dashboard/rbd_ls.py
@@ -1,8 +1,30 @@
 
-import rados
 import rbd
+import rados
+from types import OsdMap
 from remote_view_cache import RemoteViewCache
 
+class RbdPoolLs(RemoteViewCache):
+    def _get(self):
+        from mgr_module import ceph_state
+        ctx_capsule = ceph_state.get_context()
+
+        osd_map = self._module.get_sync_object(OsdMap).data
+        osd_pools = [pool['pool_name'] for pool in osd_map['pools']]
+
+        rbd_pools = []
+        for pool in osd_pools:
+            self.log.debug("Constructing IOCtx " + pool)
+            try:
+                ioctx = self._module.rados.open_ioctx(pool)
+                ioctx.stat("rbd_directory")
+                rbd_pools.append(pool)
+            except (rados.PermissionError, rados.ObjectNotFound):
+                self.log.debug("No RBD directory in " + pool)
+            except:
+                self.log.exception("Failed to open pool " + pool)
+
+        return rbd_pools
 
 class RbdLs(RemoteViewCache):
     def __init__(self, module_inst, pool):
@@ -14,16 +36,10 @@ class RbdLs(RemoteViewCache):
         self.rbd = None
 
     def _init(self):
-        from mgr_module import ceph_state
-        ctx_capsule = ceph_state.get_context()
+        self.log.debug("Constructing IOCtx")
+        self.ioctx = self._module.rados.open_ioctx(self.pool)
 
-        self.log.info("Constructing Rados")
-        cluster = rados.Rados(context=ctx_capsule)
-        cluster.connect()
-        self.log.info("Constructing IOCtx")
-        self.ioctx = cluster.open_ioctx("rbd")
-
-        self.log.info("Constructing RBD")
+        self.log.debug("Constructing RBD")
         self.rbd = rbd.RBD()
 
     def _get(self):
@@ -34,5 +50,40 @@ class RbdLs(RemoteViewCache):
             i = rbd.Image(self.ioctx, name)
             stat = i.stat()
             stat['name'] = name
+            features = i.features()
+            stat['features'] = features
+            stat['features_name'] = self._format_bitmask(features)
+
+            try:
+                parent_info = i.parent_info()
+                parent = "{}@{}".format(parent_info[0], parent_info[1])
+                if parent_info[0] != self.pool:
+                    parent = "{}/{}".format(parent_info[0], parent)
+                stat['parent'] = parent
+            except rbd.ImageNotFound:
+                pass
             result.append(stat)
         return result
+
+    def _format_bitmask(self, features):
+        names = ""
+        RBD_FEATURES_NAME_MAPPING = {
+            rbd.RBD_FEATURE_LAYERING: "layering",
+            rbd.RBD_FEATURE_STRIPINGV2: "striping",
+            rbd.RBD_FEATURE_EXCLUSIVE_LOCK: "exclusive-lock",
+            rbd.RBD_FEATURE_OBJECT_MAP: "object-map",
+            rbd.RBD_FEATURE_FAST_DIFF: "fast-diff",
+            rbd.RBD_FEATURE_DEEP_FLATTEN: "deep-flatten",
+            rbd.RBD_FEATURE_JOURNALING: "journaling",
+            rbd.RBD_FEATURE_DATA_POOL: "data-pool",
+        }
+
+        for key in RBD_FEATURES_NAME_MAPPING.keys():
+            if (key & features == 0):
+                continue
+
+            if names:
+                names = names + ", "
+            names = names + RBD_FEATURES_NAME_MAPPING.get(key)
+
+        return names