1 files changed, 87 insertions, 0 deletions
diff --git a/src/pybind/mgr/dashboard/services/cluster.py b/src/pybind/mgr/dashboard/services/cluster.py
new file mode 100644
index 000000000..9caaf1963
--- /dev/null
+++ b/src/pybind/mgr/dashboard/services/cluster.py
@@ -0,0 +1,87 @@
+# -*- coding: utf-8 -*-
+from enum import Enum
+from typing import NamedTuple
+
+from .. import mgr
+
+
+class ClusterCapacity(NamedTuple):
+    total_avail_bytes: int
+    total_bytes: int
+    total_used_raw_bytes: int
+    total_objects: int
+    total_pool_bytes_used: int
+    average_object_size: int
+
+
+class ClusterModel:
+
+    class Status(Enum):
+        INSTALLED = 0
+        POST_INSTALLED = 1
+
+    status: Status
+
+    def __init__(self, status=Status.POST_INSTALLED.name):
+        """
+        :param status: The status of the cluster. Assume that the cluster
+            is already functional by default.
+        :type status: str
+        """
+        self.status = self.Status[status]
+
+    def dict(self):
+        return {'status': self.status.name}
+
+    def to_db(self):
+        mgr.set_store('cluster/status', self.status.name)
+
+    @classmethod
+    def from_db(cls):
+        """
+        Get the stored cluster status from the configuration key/value store.
+        If the status is not set, assume it is already fully functional.
+        """
+        return cls(status=mgr.get_store('cluster/status', cls.Status.POST_INSTALLED.name))
+
+    @classmethod
+    def get_capacity(cls) -> ClusterCapacity:
+        df = mgr.get('df')
+        total_pool_bytes_used = 0
+        average_object_size = 0
+        total_data_pool_objects = 0
+        total_data_pool_bytes_used = 0
+        rgw_pools_data = cls.get_rgw_pools()
+
+        for pool in df['pools']:
+            pool_name = str(pool['name'])
+            if pool_name in rgw_pools_data:
+                if pool_name.endswith('.data'):
+                    objects = pool['stats']['objects']
+                    pool_bytes_used = pool['stats']['bytes_used']
+                    total_pool_bytes_used += pool_bytes_used
+                    total_data_pool_objects += objects
+                    replica = rgw_pools_data[pool_name]
+                    total_data_pool_bytes_used += pool_bytes_used / replica
+
+        average_object_size = total_data_pool_bytes_used / total_data_pool_objects if total_data_pool_objects != 0 else 0  # noqa E501  #pylint: disable=line-too-long
+
+        return ClusterCapacity(
+            total_avail_bytes=df['stats']['total_avail_bytes'],
+            total_bytes=df['stats']['total_bytes'],
+            total_used_raw_bytes=df['stats']['total_used_raw_bytes'],
+            total_objects=total_data_pool_objects,
+            total_pool_bytes_used=total_pool_bytes_used,
+            average_object_size=average_object_size
+        )._asdict()
+
+    @classmethod
+    def get_rgw_pools(cls):
+        rgw_pool_size = {}
+
+        osd_map = mgr.get('osd_map')
+        for pool in osd_map['pools']:
+            if 'rgw' in pool.get('application_metadata', {}):
+                name = pool['pool_name']
+                rgw_pool_size[name] = pool['size']
+        return rgw_pool_size